添加AI代理模块

2026-01-07 11:56:53 +08:00
parent c634cb2daa
commit d51c65a580
12 changed files with 600 additions and 0 deletions
--- a/api/agents/prompts/init.py
+++ b/api/agents/prompts/init.py
@@ -0,0 +1,18 @@
+"""
+提示词模块
+"""
+from .conversation_prompts import ConversationStage, get_system_prompt as get_conversation_prompt, get_questions_for_stage, INTERVIEW_QUESTIONS
+from .memory_prompts import get_system_prompt as get_memory_prompt, get_chapter_classification_prompt, get_text_rewrite_prompt, CHAPTER_CATEGORIES, CHAPTER_ORDER
+
+__all__ = [
+    "ConversationStage",
+    "get_conversation_prompt",
+    "get_questions_for_stage",
+    "INTERVIEW_QUESTIONS",
+    "get_memory_prompt",
+    "get_chapter_classification_prompt",
+    "get_text_rewrite_prompt",
+    "CHAPTER_CATEGORIES",
+    "CHAPTER_ORDER",
+]
+