修改了部分调用

This commit is contained in:
jinye_huang 2025-04-22 16:30:48 +08:00
parent e3d466353d
commit 8b6d68b2b0
2 changed files with 11 additions and 1 deletions

View File

@ -259,7 +259,17 @@ def prepare_topic_generation(
# 创建AI Agent (still create agent here for the topic generation phase)
try:
print("Initializing AI Agent for topic generation...")
ai_agent = AI_Agent(config["api_url"], config["model"], config["api_key"])
# --- Read timeout/retry from config ---
request_timeout = config.get("request_timeout", 30) # Default 30 seconds
max_retries = config.get("max_retries", 3) # Default 3 retries
# --- Pass values to AI_Agent ---
ai_agent = AI_Agent(
config["api_url"],
config["model"],
config["api_key"],
timeout=request_timeout,
max_retries=max_retries
)
except Exception as e:
print(f"Error initializing AI Agent for topic generation: {e}")
traceback.print_exc()