feat: 데이터 저장 경로를 model_Dev/data로 업데이트
This commit is contained in:
@@ -9,6 +9,7 @@ def main():
|
||||
parser.add_argument('--topics', nargs='+', help='수집할 주제 목록', default=None)
|
||||
parser.add_argument('--config', default='./config.json', help='설정 파일 경로')
|
||||
parser.add_argument('--auto-topics', action='store_true', help='AI가 스스로 주제를 선정하여 조사')
|
||||
parser.add_argument('--save-path', help='데이터 저장 경로 (예: /content/drive/MyDrive/MyFolder)', default=None)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
@@ -22,7 +23,18 @@ def main():
|
||||
print("모델 다운로드 실패. 프로그램을 종료합니다.")
|
||||
sys.exit(1)
|
||||
|
||||
# 2. AI 에이전트 초기화
|
||||
# 2. 설정 파일 로드 및 수정
|
||||
with open(args.config, 'r') as f:
|
||||
config = json.load(f)
|
||||
|
||||
if args.save_path:
|
||||
config['data_storage']['drive_mount_path'] = args.save_path
|
||||
# 수정된 config 저장
|
||||
with open(args.config, 'w') as f:
|
||||
json.dump(config, f, indent=2)
|
||||
print(f"저장 경로 설정됨: {args.save_path}")
|
||||
|
||||
# 3. AI 에이전트 초기화
|
||||
print("AI 에이전트 초기화 중...")
|
||||
agent = AIAgent(args.config)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user