feat: chat model factory defaults + omit strict_mode=False from payload (#808)

Co-authored-by: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: cosminacho

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "uipath-langchain"
-version = "0.10.4"
+version = "0.10.5"
 description = "Python SDK that enables developers to build and deploy LangGraph agents to the UiPath Cloud Platform"
 readme = { file = "README.md", content-type = "text/markdown" }
 requires-python = ">=3.11"

src/uipath_langchain/chat/_legacy/chat_model_factory.py

@@ -40,7 +40,7 @@ def _create_openai_llm(
     model: str,
     api_flavor: APIFlavor,
     temperature: float | None,
-    max_tokens: int,
+    max_tokens: int | None,
     agenthub_config: str,
     byo_connection_id: str | None = None,
     **kwargs: Any,
@@ -87,7 +87,7 @@ def _create_bedrock_llm(
     model: str,
     api_flavor: APIFlavor,
     temperature: float | None,
-    max_tokens: int,
+    max_tokens: int | None,
     agenthub_config: str,
     byo_connection_id: str | None = None,
     **kwargs: Any,
@@ -244,8 +244,8 @@ def _get_model_info(
 
 def get_chat_model(
     model: str,
-    temperature: float,
-    max_tokens: int,
+    temperature: float | None,
+    max_tokens: int | None,
     agenthub_config: str,
     byo_connection_id: str | None = None,
     **kwargs: Any,

src/uipath_langchain/chat/chat_model_factory.py

@@ -25,6 +25,9 @@
 
 _UNSET: Final[Any] = object()
 DEFAULT_TIMEOUT_SECONDS: Final[float] = 300.0
+DEFAULT_MAX_TOKENS: Final[int] = 1000
+DEFAULT_TEMPERATURE: Final[float] = 0.0
+DEFAULT_MAX_RETRIES: Final[int] = 3
 
 
 def get_chat_model(
@@ -36,10 +39,10 @@ def get_chat_model(
     vendor_type: VendorType | str | None = None,
     api_flavor: ApiFlavor | str | None = None,
     custom_class: type[UiPathBaseChatModel] | None = None,
-    temperature: float | None = _UNSET,
-    max_tokens: int | None = _UNSET,
+    temperature: float | None = DEFAULT_TEMPERATURE,
+    max_tokens: int | None = DEFAULT_MAX_TOKENS,
     timeout: float | None = DEFAULT_TIMEOUT_SECONDS,
-    max_retries: int | None = _UNSET,
+    max_retries: int | None = DEFAULT_MAX_RETRIES,
     callbacks: Callbacks = _UNSET,
     # Legacy-only arguments
     agenthub_config: str | None = None,
@@ -58,10 +61,14 @@ def get_chat_model(
             Converse). Auto-detected when omitted.
         custom_class: Custom ``UiPathBaseChatModel`` subclass to instantiate
             instead of the auto-detected one.
-        temperature: Sampling temperature. Forwarded only when explicitly set.
-        max_tokens: Maximum output tokens. Forwarded only when explicitly set.
+        temperature: Sampling temperature. Defaults to 0.0. Pass ``None`` to
+            omit the parameter when the underlying client supports it.
+        max_tokens: Maximum output tokens. Defaults to 1000 to match the
+            historical default from ``UiPathRequestMixin``. Pass ``None`` to
+            forward an explicit unset value (lets the underlying client apply
+            its own default or use no limit).
         timeout: Request timeout in seconds. Defaults to 300 seconds.
-        max_retries: Max retry count. Forwarded only when explicitly set.
+        max_retries: Max retry count. Defaults to 3.
         callbacks: LangChain callbacks (handlers or a manager) attached to the
             returned chat model. Accepts ``list[BaseCallbackHandler]`` or a
             ``BaseCallbackManager``. Forwarded only when explicitly set.
@@ -131,8 +138,8 @@ def _legacy_chat_model(
 
     return _legacy_get_chat_model(
         model,
-        temperature if temperature is not _UNSET and temperature is not None else 0.0,
-        max_tokens if max_tokens is not _UNSET and max_tokens is not None else 0,
+        temperature,
+        max_tokens,
         agenthub_config,
         byo_connection_id,
         **kwargs,

src/uipath_langchain/chat/handlers/anthropic.py

@@ -48,8 +48,8 @@ def get_tool_binding_kwargs(
         kwargs: dict[str, Any] = {"tool_choice": tool_choice}
         if parallel_tool_calls is not None:
             kwargs["parallel_tool_calls"] = parallel_tool_calls
-        if strict_mode is not None:
-            kwargs["strict"] = strict_mode
+        if strict_mode is True:
+            kwargs["strict"] = True
         return kwargs
 
     def check_stop_reason(self, response: AIMessage) -> None:

src/uipath_langchain/chat/handlers/fireworks.py

@@ -18,8 +18,8 @@ def get_tool_binding_kwargs(
         strict_mode: bool | None = None,
     ) -> dict[str, Any]:
         kwargs: dict[str, Any] = {"tool_choice": tool_choice}
-        if strict_mode is not None:
-            kwargs["strict"] = strict_mode
+        if strict_mode is True:
+            kwargs["strict"] = True
         return kwargs
 
     def check_stop_reason(self, response: Any) -> None:

src/uipath_langchain/chat/handlers/openai.py

@@ -73,8 +73,8 @@ def get_tool_binding_kwargs(
         kwargs: dict[str, Any] = {"tool_choice": tool_choice}
         if parallel_tool_calls is not None:
             kwargs["parallel_tool_calls"] = parallel_tool_calls
-        if strict_mode is not None:
-            kwargs["strict"] = strict_mode
+        if strict_mode is True:
+            kwargs["strict"] = True
         return kwargs
 
     def check_stop_reason(self, response: AIMessage) -> None:

tests/chat/handlers/test_tool_binding_kwargs.py

@@ -103,10 +103,11 @@ def test_strict_mode_true(self):
         assert result["strict"] is True
 
     def test_strict_mode_false(self):
+        """strict_mode=False must be omitted from the request payload."""
         result = self.handler.get_tool_binding_kwargs(
             tools=self.tools, tool_choice="auto", strict_mode=False
         )
-        assert result["strict"] is False
+        assert "strict" not in result
 
     def test_all_keys_present(self):
         result = self.handler.get_tool_binding_kwargs(
@@ -161,17 +162,18 @@ def test_strict_mode_true(self):
         assert result["strict"] is True
 
     def test_strict_mode_false(self):
+        """strict_mode=False must be omitted from the request payload."""
         result = self.handler.get_tool_binding_kwargs(
             tools=self.tools, tool_choice="auto", strict_mode=False
         )
-        assert result["strict"] is False
+        assert "strict" not in result
 
     def test_all_keys_present(self):
         result = self.handler.get_tool_binding_kwargs(
             tools=self.tools,
             tool_choice="any",
             parallel_tool_calls=True,
-            strict_mode=False,
+            strict_mode=True,
         )
         assert set(result.keys()) == {"tool_choice", "parallel_tool_calls", "strict"}