flowforge-sdk 0.2.3__tar.gz → 0.2.4__tar.gz

{flowforge_sdk-0.2.3 → flowforge_sdk-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowforge-sdk
-Version: 0.2.3
+Version: 0.2.4
 Summary: Python SDK for FlowForge - AI workflow orchestration
 Project-URL: Homepage, https://github.com/flowforge/flowforge
 Project-URL: Documentation, https://flowforge.dev/docs

{flowforge_sdk-0.2.3 → flowforge_sdk-0.2.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flowforge-sdk"
-version = "0.2.3"
+version = "0.2.4"
 description = "Python SDK for FlowForge - AI workflow orchestration"
 readme = "README.md"
 requires-python = ">=3.11"

{flowforge_sdk-0.2.3 → flowforge_sdk-0.2.4}/src/flowforge/steps.py

@@ -564,9 +564,32 @@ class StepManager:
                 "content": ai_response.get("content", ""),
             }
 
+            # Normalize tool calls to OpenAI format for litellm compatibility.
+            # Server stores flat format: {"id", "name", "arguments"}
+            # LiteLLM expects OpenAI format: {"id", "type", "function": {"name", "arguments"}}
+            raw_tool_calls = ai_response.get("tool_calls", [])
+            normalized_tool_calls = []
+            for tc in raw_tool_calls:
+                if "function" in tc:
+                    # Already in OpenAI format
+                    normalized_tool_calls.append(tc)
+                else:
+                    # Convert flat format to OpenAI format
+                    args = tc.get("arguments", {})
+                    if not isinstance(args, str):
+                        args = json.dumps(args)
+                    normalized_tool_calls.append({
+                        "id": tc.get("id", ""),
+                        "type": "function",
+                        "function": {
+                            "name": tc.get("name", ""),
+                            "arguments": args,
+                        },
+                    })
+
             # Include tool calls in assistant message if present
-            if ai_response.get("tool_calls"):
-                assistant_message["tool_calls"] = ai_response["tool_calls"]
+            if normalized_tool_calls:
+                assistant_message["tool_calls"] = normalized_tool_calls
 
             state.messages.append(assistant_message)
 
@@ -574,12 +597,12 @@ class StepManager:
             finish_reason = ai_response.get("finish_reason", "stop")
 
             # If no tool calls, we're done
-            if not ai_response.get("tool_calls") or finish_reason == "stop":
+            if not normalized_tool_calls or finish_reason == "stop":
                 state.status = "completed"
                 break
 
             # Execute each tool call
-            for tool_call in ai_response.get("tool_calls", []):
+            for tool_call in normalized_tool_calls:
                 tool_call_id = tool_call.get("id", f"tool-{state.tool_calls_count}")
                 tool_name = tool_call.get("function", {}).get("name", "")
                 tool_args = tool_call.get("function", {}).get("arguments", {})