refactor: AI对话基本完成

2026-07-25 22:06:06 +00:00 · 2024-01-27 21:50:40 +08:00
parent 850a4eeb7c
commit f10af09bd2
14 changed files with 308 additions and 46 deletions
@@ -21,14 +21,9 @@ class ScenarioPromptLoader(loader.PromptLoader):
                file_json = json.loads(file_str)
                messages = []
                for msg in file_json["prompt"]:
-                    role = llm_entities.MessageRole.SYSTEM
+                    role = 'system'
                    if "role" in msg:
-                        if msg["role"] == "user":
-                            role = llm_entities.MessageRole.USER
-                        elif msg["role"] == "system":
-                            role = llm_entities.MessageRole.SYSTEM
-                        elif msg["role"] == "function":
-                            role = llm_entities.MessageRole.FUNCTION
+                        role = msg['role']
                    messages.append(
                        llm_entities.Message(
                            role=role,