vectorvein 0.2.30__tar.gz → 0.2.32__tar.gz

{vectorvein-0.2.30 → vectorvein-0.2.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vectorvein
-Version: 0.2.30
+Version: 0.2.32
 Summary: VectorVein Python SDK
 Author-Email: Anderson <andersonby@163.com>
 License: MIT

{vectorvein-0.2.30 → vectorvein-0.2.32}/pyproject.toml

@@ -17,7 +17,7 @@ description = "VectorVein Python SDK"
 name = "vectorvein"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "0.2.30"
+version = "0.2.32"
 
 [project.license]
 text = "MIT"

{vectorvein-0.2.30 → vectorvein-0.2.32}/src/vectorvein/chat_clients/utils.py

@@ -559,6 +559,7 @@ def format_workflow_messages(message: dict, content: str, backend: BackendType):
 def transform_from_openai_message(message: ChatCompletionMessageParam, backend: BackendType):
     role = message.get("role", "user")
     content = message.get("content", "")
+    tool_calls = message.get("tool_calls", [])
 
     if backend == BackendType.Anthropic:
         if role == "tool":
@@ -582,9 +583,32 @@ def transform_from_openai_message(message: ChatCompletionMessageParam, backend:
                         formatted_content.append(format_image_message(item["image_url"]["url"], backend))
                     else:
                         formatted_content.append(item)
+            if tool_calls:
+                for tool_call in tool_calls:
+                    formatted_content.append(
+                        {
+                            "type": "tool_use",
+                            "id": tool_call["id"],
+                            "name": tool_call["function"]["name"],
+                            "input": json.loads(tool_call["function"]["arguments"]),
+                        }
+                    )
             return {"role": role, "content": formatted_content}
         else:
-            return {"role": role, "content": content}
+            if tool_calls:
+                formatted_content = [{"type": "text", "text": content}]
+                for tool_call in tool_calls:
+                    formatted_content.append(
+                        {
+                            "type": "tool_use",
+                            "id": tool_call["id"],
+                            "name": tool_call["function"]["name"],
+                            "input": json.loads(tool_call["function"]["arguments"]),
+                        }
+                    )
+                return {"role": role, "content": formatted_content}
+            else:
+                return {"role": role, "content": content}
     else:
         return message  # 对于其他后端，保持原样