Browse Source

Update api_server: temporary support for avatar input

StrayWarrior 2 days ago
parent
commit
33e8355f5b
1 changed files with 7 additions and 0 deletions
  1. 7 0
      api_server.py

+ 7 - 0
api_server.py

@@ -253,6 +253,13 @@ def run_chat_prompt(req_data):
     messages = [system_prompt]
     if req_data['scene'] == 'custom_debugging':
         messages.extend(compose_openai_chat_messages_no_time(dialogue_history))
+        if '头像' in system_prompt['content']:
+            messages.extend({
+                "role": 'user',
+                "content": [
+                    {"type": "image_url", "image_url": {"url": user_profile['avatar']}}
+                ]
+            })
     else:
         messages.extend(DialogueManager.compose_chat_messages_openai_compatible(dialogue_history, current_time_str))
     return run_openai_chat(messages, model_name, temperature=1, top_p=0.7, max_tokens=1024)