Fix: Version update to match the new argument parallel_tool_calls instead of openai_parallel_tool_calls (#305)

Author: cosminacho

pyproject.toml

@@ -6,7 +6,7 @@ readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11,<3.14"
 dependencies = [
   "uipath==2.9.5",
-  "uipath-langchain[vertex,bedrock]==0.6.4",
+  "uipath-langchain[vertex,bedrock]==0.7.0",
   "langgraph>=1.0.0,<2.0.0",
   "azure-monitor-opentelemetry-exporter>=1.0.0b46,<2.0.0",
   "opentelemetry-instrumentation-httpx>=0.60b0",

src/uipath_agents/agent_graph_builder/graph.py

@@ -17,11 +17,12 @@
 from uipath_langchain.agent.tools import create_tools_from_resources
 from uipath_langchain.agent.tools.mcp import create_mcp_tools_from_agent
 
+from uipath_agents.agent_graph_builder.version import supports_parallel_tool_calls
+
 from .config import AgentExecutionType, get_thinking_messages_limit
 from .llm_utils import create_llm
 from .message_utils import create_message_factory
 from .session_info_debug_state import SessionInfoDebugStateFactory
-from .version import supports_openai_parallel_tool_calls
 
 AGENT_MAX_ITERATIONS_DEFAULT = 25
 
@@ -81,8 +82,8 @@ async def build_agent_graph(
             agent_definition.settings.model
         ),
         is_conversational=agent_definition.is_conversational,
-        enable_openai_parallel_tool_calls=supports_openai_parallel_tool_calls(
-            agent_definition.version
+        parallel_tool_calls=supports_parallel_tool_calls(
+            agent_definition.version, agent_definition.settings.model
         ),
     )
 

src/uipath_agents/agent_graph_builder/version.py

@@ -5,9 +5,12 @@
 _PARALLEL_TOOL_CALLS_MIN_VERSION = Version("1.1.0")
 
 
-def supports_openai_parallel_tool_calls(version: str) -> bool:
-    """Check if the agent version supports parallel tool calls."""
+def supports_parallel_tool_calls(version: str, model_name: str) -> bool:
+    """Check if the agent version and model name supports parallel tool calls."""
     try:
-        return Version(version) >= _PARALLEL_TOOL_CALLS_MIN_VERSION
+        return (
+            Version(version) >= _PARALLEL_TOOL_CALLS_MIN_VERSION
+            or "gpt" not in model_name.lower()
+        )
     except InvalidVersion:
         return True

tests/unit/agent_graph_builder/test_graph.py

@@ -14,9 +14,7 @@
 
 from uipath_agents.agent_graph_builder import build_agent_graph
 from uipath_agents.agent_graph_builder.config import AgentExecutionType
-from uipath_agents.agent_graph_builder.version import (
-    supports_openai_parallel_tool_calls,
-)
+from uipath_agents.agent_graph_builder.version import supports_parallel_tool_calls
 
 
 def create_test_agent_definition(**overrides: Any) -> LowCodeAgentDefinition:
@@ -786,28 +784,40 @@ async def test_conversational_agent_uses_conversational_message_factory(self):
 
 
 class TestSupportsParallelToolCalls:
-    """Test _supports_parallel_tool_calls version check."""
+    """Test supports_parallel_tool_calls version and model name checks."""
 
     def test_version_1_0_0_is_false(self) -> None:
-        assert supports_openai_parallel_tool_calls("1.0.0") is False
+        assert supports_parallel_tool_calls("1.0.0", "gpt-4") is False
 
     def test_version_1_0_9_is_false(self) -> None:
-        assert supports_openai_parallel_tool_calls("1.0.9") is False
+        assert supports_parallel_tool_calls("1.0.9", "gpt-4") is False
 
     def test_version_1_1_0_is_true(self) -> None:
-        assert supports_openai_parallel_tool_calls("1.1.0") is True
+        assert supports_parallel_tool_calls("1.1.0", "gpt-4") is True
 
     def test_version_1_2_0_is_true(self) -> None:
-        assert supports_openai_parallel_tool_calls("1.2.0") is True
+        assert supports_parallel_tool_calls("1.2.0", "gpt-4") is True
 
     def test_version_2_0_0_is_true(self) -> None:
-        assert supports_openai_parallel_tool_calls("2.0.0") is True
+        assert supports_parallel_tool_calls("2.0.0", "gpt-4") is True
+
+    def test_invalid_version_returns_true(self) -> None:
+        assert supports_parallel_tool_calls("invalid", "gpt-4") is True
+
+    def test_empty_version_returns_true(self) -> None:
+        assert supports_parallel_tool_calls("", "gpt-4") is True
+
+    def test_gpt_model_with_sufficient_version_is_true(self) -> None:
+        assert supports_parallel_tool_calls("1.1.0", "gpt-4o") is True
+
+    def test_gpt_model_name_case_insensitive(self) -> None:
+        assert supports_parallel_tool_calls("1.1.0", "GPT-4o-mini") is True
 
-    def test_invalid_version_is_false(self) -> None:
-        assert supports_openai_parallel_tool_calls("invalid") is True
+    def test_non_gpt_model_with_sufficient_version_is_true(self) -> None:
+        assert supports_parallel_tool_calls("1.1.0", "claude-3-sonnet") is True
 
-    def test_empty_string_is_false(self) -> None:
-        assert supports_openai_parallel_tool_calls("") is True
+    def test_non_gpt_model_old_version_is_true(self) -> None:
+        assert supports_parallel_tool_calls("1.0.0", "claude-3-opus") is True
 
 
 @pytest.mark.asyncio
@@ -847,7 +857,7 @@ async def test_old_version_disables_parallel_tool_calls(self) -> None:
             await build_agent_graph(agent_def)
 
             config = mock_create.call_args.kwargs["config"]
-            assert config.enable_openai_parallel_tool_calls is False
+            assert config.parallel_tool_calls is False
 
     async def test_new_version_enables_parallel_tool_calls(self) -> None:
         agent_def = create_test_agent_definition(version="1.1.0")
@@ -873,4 +883,4 @@ async def test_new_version_enables_parallel_tool_calls(self) -> None:
             await build_agent_graph(agent_def)
 
             config = mock_create.call_args.kwargs["config"]
-            assert config.enable_openai_parallel_tool_calls is True
+            assert config.parallel_tool_calls is True