Quellcode durchsuchen

fix/moonshot-function-call (#3339)

Yeuoly vor 1 Jahr
Ursprung
Commit
18b855140d
1 geänderte Dateien mit 1 neuen und 8 gelöschten Zeilen
  1. 1 8
      api/core/model_runtime/model_providers/moonshot/llm/llm.py

+ 1 - 8
api/core/model_runtime/model_providers/moonshot/llm/llm.py

@@ -132,15 +132,13 @@ class MoonshotLargeLanguageModel(OAIAPICompatLargeLanguageModel):
                         "id": function_call.id,
                         "type": function_call.type,
                         "function": {
-                            "name": f"functions.{function_call.function.name}",
+                            "name": function_call.function.name,
                             "arguments": function_call.function.arguments
                         }
                     })
         elif isinstance(message, ToolPromptMessage):
             message = cast(ToolPromptMessage, message)
             message_dict = {"role": "tool", "content": message.content, "tool_call_id": message.tool_call_id}
-            if not message.name.startswith("functions."):
-                message.name = f"functions.{message.name}"
         elif isinstance(message, SystemPromptMessage):
             message = cast(SystemPromptMessage, message)
             message_dict = {"role": "system", "content": message.content}
@@ -238,11 +236,6 @@ class MoonshotLargeLanguageModel(OAIAPICompatLargeLanguageModel):
                 if new_tool_call.type:
                     tool_call.type = new_tool_call.type
                 if new_tool_call.function.name:
-                    # remove the functions. prefix
-                    if new_tool_call.function.name.startswith('functions.'):
-                        parts = new_tool_call.function.name.split('functions.')
-                        if len(parts) > 1:
-                            new_tool_call.function.name = parts[1]
                     tool_call.function.name = new_tool_call.function.name
                 if new_tool_call.function.arguments:
                     tool_call.function.arguments += new_tool_call.function.arguments