remove check_tool_choice because "any" works for all providers

Author: cosminacho

src/uipath_langchain/agent/react/llm_node.py

@@ -60,7 +60,6 @@ def create_llm_node(
     """
     bindable_tools = list(tools) if tools else []
     payload_handler = get_payload_handler(model)
-    tool_choice_required_value = payload_handler.get_required_tool_choice()
 
     async def llm_node(state: StateT):
         messages: list[AnyMessage] = state.messages
@@ -78,16 +77,15 @@ async def llm_node(state: StateT):
         static_schema_tools = _apply_tool_argument_properties(
             bindable_tools, state, input_schema
         )
-        base_llm = model.bind_tools(static_schema_tools)
 
         if (
             not is_conversational
             and bindable_tools
             and consecutive_thinking_messages >= thinking_messages_limit
         ):
-            llm = base_llm.bind(tool_choice=tool_choice_required_value)
+            llm = model.bind_tools(static_schema_tools, tool_choice="any")
         else:
-            llm = base_llm
+            llm = model.bind_tools(static_schema_tools)
 
         response = await llm.ainvoke(messages)
         if not isinstance(response, AIMessage):

src/uipath_langchain/chat/handlers/base.py

@@ -1,7 +1,6 @@
 """Abstract base class for model payload handlers."""
 
 from abc import ABC, abstractmethod
-from typing import Any
 
 from langchain_core.messages import AIMessage
 
@@ -12,17 +11,6 @@ class ModelPayloadHandler(ABC):
     Each handler provides provider-specific parameter values for LLM operations.
     """
 
-    @abstractmethod
-    def get_required_tool_choice(self) -> str | dict[str, Any]:
-        """Get the tool_choice value that enforces tool usage.
-
-        Returns:
-            Provider-specific value to force tool usage:
-            - "required" for OpenAI-compatible models
-            - "any" for Bedrock Converse and Vertex models (string format)
-            - {"type": "any"} for Bedrock Invoke API (dict format required)
-        """
-
     @abstractmethod
     def check_stop_reason(self, response: AIMessage) -> None:
         """Check response stop reason and raise exception for faulty terminations.

src/uipath_langchain/chat/handlers/bedrock_converse.py

@@ -1,7 +1,5 @@
 """Bedrock Converse payload handler."""
 
-from typing import Any
-
 from langchain_core.messages import AIMessage
 from uipath.runtime.errors import UiPathErrorCode
 
@@ -42,10 +40,6 @@
 class BedrockConversePayloadHandler(ModelPayloadHandler):
     """Payload handler for AWS Bedrock Converse API."""
 
-    def get_required_tool_choice(self) -> str | dict[str, Any]:
-        """Get tool_choice value for Bedrock Converse API."""
-        return "any"
-
     def check_stop_reason(self, response: AIMessage) -> None:
         """Check Bedrock Converse stopReason and raise exception for faulty terminations.
 

src/uipath_langchain/chat/handlers/bedrock_invoke.py

@@ -1,7 +1,5 @@
 """Bedrock Invoke payload handler."""
 
-from typing import Any
-
 from langchain_core.messages import AIMessage
 from uipath.runtime.errors import UiPathErrorCode
 
@@ -37,10 +35,6 @@
 class BedrockInvokePayloadHandler(ModelPayloadHandler):
     """Payload handler for AWS Bedrock Invoke API."""
 
-    def get_required_tool_choice(self) -> str | dict[str, Any]:
-        """Get tool_choice value for Bedrock Invoke API."""
-        return {"type": "any"}
-
     def check_stop_reason(self, response: AIMessage) -> None:
         """Check Bedrock Invoke stop_reason and raise exception for faulty terminations.
 

src/uipath_langchain/chat/handlers/openai_completions.py

@@ -1,7 +1,5 @@
 """OpenAI Chat Completions payload handler."""
 
-from typing import Any
-
 from langchain_core.messages import AIMessage
 from uipath.runtime.errors import UiPathErrorCode
 
@@ -31,10 +29,6 @@
 class OpenAICompletionsPayloadHandler(ModelPayloadHandler):
     """Payload handler for OpenAI Chat Completions API."""
 
-    def get_required_tool_choice(self) -> str | dict[str, Any]:
-        """Get tool_choice value for OpenAI Completions API."""
-        return "required"
-
     def check_stop_reason(self, response: AIMessage) -> None:
         """Check OpenAI finish_reason and raise exception for faulty terminations.
 

src/uipath_langchain/chat/handlers/openai_responses.py

@@ -1,7 +1,5 @@
 """OpenAI payload handlers."""
 
-from typing import Any
-
 from langchain_core.messages import AIMessage
 from uipath.runtime.errors import UiPathErrorCode
 
@@ -35,10 +33,6 @@
 class OpenAIResponsesPayloadHandler(ModelPayloadHandler):
     """Payload handler for OpenAI Responses API."""
 
-    def get_required_tool_choice(self) -> str | dict[str, Any]:
-        """Get tool_choice value for OpenAI Responses API."""
-        return "required"
-
     def check_stop_reason(self, response: AIMessage) -> None:
         """Check OpenAI Responses API status and raise exception for faulty terminations.
 

src/uipath_langchain/chat/handlers/vertex_gemini.py

@@ -1,7 +1,5 @@
 """Vertex Gemini payload handler."""
 
-from typing import Any
-
 from langchain_core.messages import AIMessage
 from uipath.runtime.errors import UiPathErrorCode
 
@@ -115,10 +113,6 @@
 class VertexGeminiPayloadHandler(ModelPayloadHandler):
     """Payload handler for Google Vertex AI Gemini API."""
 
-    def get_required_tool_choice(self) -> str | dict[str, Any]:
-        """Get tool_choice value for Vertex Gemini API."""
-        return "any"
-
     def check_stop_reason(self, response: AIMessage) -> None:
         """Check Vertex Gemini finishReason and raise exception for faulty terminations.