diff --git a/config.json b/config.json new file mode 100644 index 0000000..f1cc05d --- /dev/null +++ b/config.json @@ -0,0 +1,41 @@ +[ + { + "task_name": "MacBook Air M1", + "enabled": true, + "keyword": "macbook air m1", + "max_pages": 5, + "personal_only": true, + "min_price": "3000", + "max_price": "5000", + "cron": "3 12 * * *", + "ai_prompt_base_file": "prompts/base_prompt.txt", + "ai_prompt_criteria_file": "prompts/macbook_criteria.txt", + "is_running": false + }, + { + "task_name": "DJI Mini 3 Pro", + "enabled": true, + "keyword": "mini 3 pro", + "max_pages": 3, + "personal_only": true, + "ai_prompt_base_file": "prompts/base_prompt.txt", + "ai_prompt_criteria_file": "prompts/dji_mini3_criteria.txt", + "min_price": "3500", + "max_price": "4500", + "cron": "", + "is_running": false + }, + { + "task_name": "婴儿车", + "enabled": false, + "keyword": "babycare 婴儿车", + "max_pages": 3, + "personal_only": true, + "min_price": "500", + "max_price": "1000", + "cron": "", + "ai_prompt_base_file": "prompts/base_prompt.txt", + "ai_prompt_criteria_file": "prompts/babycare_婴儿车_criteria.txt", + "is_running": false + } +] \ No newline at end of file diff --git a/src/prompt_utils.py b/src/prompt_utils.py index 10fe7b6..be36f98 100644 --- a/src/prompt_utils.py +++ b/src/prompt_utils.py @@ -65,6 +65,11 @@ async def generate_criteria(user_description: str, reference_file_path: str) -> ) generated_text = response.choices[0].message.content print("AI已成功生成内容。") + + # 处理content可能为None的情况 + if generated_text is None: + raise RuntimeError("AI返回的内容为空,请检查模型配置或重试。") + return generated_text.strip() except Exception as e: print(f"调用 OpenAI API 时出错: {e}")