jinye_huang/TravelContentCreator
Template
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

修改了部分调用

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
utils/__pycache__/tweet_generator.cpython-312.pyc
View File

Binary file not shown.

12
utils/tweet_generator.py
View File

@ -259,7 +259,17 @@ def prepare_topic_generation(
# 创建AI Agent (still create agent here for the topic generation phase) # 创建AI Agent (still create agent here for the topic generation phase)
try: try:
print("Initializing AI Agent for topic generation...") 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: except Exception as e:
print(f"Error initializing AI Agent for topic generation: {e}") print(f"Error initializing AI Agent for topic generation: {e}")
traceback.print_exc() traceback.print_exc()