fix: default error_on_no_tool_call to false in llm wrappers

Author: radu-mocanu

packages/uipath-llamaindex/pyproject.toml

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

packages/uipath-llamaindex/src/uipath_llamaindex/llms/_openai.py

@@ -1,7 +1,9 @@
 import os
-from typing import Any
+from typing import Any, List
 
 import httpx
+from llama_index.core.base.llms.types import ChatResponse
+from llama_index.core.tools import ToolSelection
 from llama_index.llms.azure_openai import AzureOpenAI  # type: ignore
 from uipath._utils._ssl_context import get_httpx_client_kwargs
 from uipath.utils import EndpointManager
@@ -87,3 +89,13 @@ def __init__(
         }
         final_kwargs = {**defaults, **kwargs}
         super().__init__(**final_kwargs)
+
+    def get_tool_calls_from_response(
+        self,
+        response: ChatResponse,
+        error_on_no_tool_call: bool = False,
+        **kwargs: Any,
+    ) -> List[ToolSelection]:
+        return super().get_tool_calls_from_response(
+            response, error_on_no_tool_call=error_on_no_tool_call, **kwargs
+        )

packages/uipath-llamaindex/src/uipath_llamaindex/llms/bedrock.py

@@ -1,6 +1,6 @@
 import logging
 import os
-from typing import Any, Optional, Sequence
+from typing import Any, List, Optional, Sequence
 
 from uipath.utils import EndpointManager
 
@@ -49,6 +49,7 @@ def _check_bedrock_dependencies() -> None:
     CompletionResponse,
     CompletionResponseAsyncGen,
 )
+from llama_index.core.tools import ToolSelection  # noqa: E402
 from llama_index.core.llms.callbacks import (  # noqa: E402
     llm_chat_callback,
     llm_completion_callback,
@@ -184,6 +185,16 @@ def __init__(
             **kwargs,
         )
 
+    def get_tool_calls_from_response(
+        self,
+        response: ChatResponse,
+        error_on_no_tool_call: bool = False,
+        **kwargs: Any,
+    ) -> List[ToolSelection]:
+        return super().get_tool_calls_from_response(
+            response, error_on_no_tool_call=error_on_no_tool_call, **kwargs
+        )
+
 
 class UiPathChatBedrock(Bedrock):
     def __init__(

packages/uipath-llamaindex/src/uipath_llamaindex/llms/vertex.py

@@ -1,6 +1,6 @@
 import logging
 import os
-from typing import Any, Generator, Optional, Sequence
+from typing import Any, Generator, List, Optional, Sequence
 
 import httpx
 from llama_index.core.callbacks import CallbackManager
@@ -47,6 +47,7 @@ def _check_vertex_dependencies() -> None:
     CompletionResponseAsyncGen,
     CompletionResponseGen,
 )
+from llama_index.core.tools import ToolSelection  # noqa: E402
 from llama_index.core.bridge.pydantic import PrivateAttr  # noqa: E402
 from llama_index.core.llms.callbacks import (  # noqa: E402
     llm_chat_callback,
@@ -410,3 +411,13 @@ async def gen() -> ChatResponseAsyncGen:
             )
 
         return gen()
+
+    def get_tool_calls_from_response(
+        self,
+        response: ChatResponse,
+        error_on_no_tool_call: bool = False,
+        **kwargs: Any,
+    ) -> List[ToolSelection]:
+        return super().get_tool_calls_from_response(
+            response, error_on_no_tool_call=error_on_no_tool_call, **kwargs
+        )