Переглянути джерело

Update api_server: rename compose_dialogue to format_dialogue_history

StrayWarrior 3 тижнів тому
батько
коміт
2bd5e7c9f7

+ 2 - 2
pqai_agent_server/api_server.py

@@ -190,8 +190,8 @@ def format_data_for_prompt():
         elif format_type == 'dialogue':
             if not isinstance(content, list):
                 return wrap_response(400, msg='dialogue should be a list')
-            from pqai_agent_server.utils.prompt_util import compose_dialogue
-            response = compose_dialogue(content)
+            from pqai_agent_server.utils.prompt_util import format_dialogue_history
+            response = format_dialogue_history(content)
         else:
             return wrap_response(400, msg='Invalid format_type')
         return wrap_response(200, data=response)

+ 1 - 1
pqai_agent_server/utils/prompt_util.py

@@ -178,7 +178,7 @@ def run_response_type_prompt(req_data):
     return run_openai_chat(messages, model_name, temperature=0.2, max_tokens=128)
 
 
-def compose_dialogue(dialogue: List[Dict]) -> str:
+def format_dialogue_history(dialogue: List[Dict]) -> str:
     role_map = {'user': '用户', 'assistant': '客服'}
     messages = []
     for msg in dialogue: