feat: 데이터 저장 경로를 model_Dev/data로 업데이트

This commit is contained in:
2025-08-28 10:26:29 +09:00
parent fdf330143f
commit bed6ce22f5
3 changed files with 28 additions and 10 deletions

View File

@@ -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)