latitude-sdk 1.0.0__tar.gz → 1.0.1__tar.gz

{latitude_sdk-1.0.0 → latitude_sdk-1.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: latitude-sdk
-Version: 1.0.0
+Version: 1.0.1
 Summary: Latitude SDK for Python
 Project-URL: repository, https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python
 Project-URL: homepage, https://github.com/latitude-dev/latitude-llm/tree/main/packages/sdks/python#readme

{latitude_sdk-1.0.0 → latitude_sdk-1.0.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "latitude-sdk"
-version = "1.0.0"
+version = "1.0.1"
 description = "Latitude SDK for Python"
 authors = [{ name = "Latitude Data SL", email = "hello@latitude.so" }]
 maintainers = [{ name = "Latitude Data SL", email = "hello@latitude.so" }]

{latitude_sdk-1.0.0 → latitude_sdk-1.0.1}/src/latitude_sdk/sdk/prompts.py

@@ -204,9 +204,26 @@ class Prompts:
         # NOTE: FinishedEvent not in on_event
         return FinishedEvent(uuid=uuid, conversation=conversation, response=response)
 
-    def _pause_tool_execution(self) -> ToolResult:
+    @staticmethod
+    def _pause_tool_execution() -> Any:
         raise OnToolCallPaused()
 
+    @staticmethod
+    async def _wrap_tool_handler(
+        handler: OnToolCall, arguments: Dict[str, Any], details: OnToolCallDetails
+    ) -> ToolResult:
+        tool_result: Dict[str, Any] = {"id": details.id, "name": details.name}
+
+        try:
+            result = await handler(arguments, details)
+
+            return ToolResult(**tool_result, result=result)
+        except Exception as exception:
+            if isinstance(exception, OnToolCallPaused):
+                raise exception
+
+            return ToolResult(**tool_result, result=str(exception), is_error=True)
+
     async def _handle_tool_calls(
         self, result: FinishedEvent, options: Union[RunPromptOptions, ChatPromptOptions]
     ) -> Optional[FinishedEvent]:
@@ -230,15 +247,22 @@ class Prompts:
                     response=f"Tool {tool_call.name} not supplied",
                 )
 
-        details = OnToolCallDetails(
-            conversation_uuid=result.uuid,
-            messages=result.conversation,
-            pause_execution=self._pause_tool_execution,
-            requested_tool_calls=result.response.tool_calls,
-        )
-
         tool_results = await asyncio.gather(
-            *[options.tools[tool_call.name](tool_call, details) for tool_call in result.response.tool_calls],
+            *[
+                self._wrap_tool_handler(
+                    options.tools[tool_call.name],
+                    tool_call.arguments,
+                    OnToolCallDetails(
+                        id=tool_call.id,
+                        name=tool_call.name,
+                        conversation_uuid=result.uuid,
+                        messages=result.conversation,
+                        pause_execution=self._pause_tool_execution,
+                        requested_tool_calls=result.response.tool_calls,
+                    ),
+                )
+                for tool_call in result.response.tool_calls
+            ],
             return_exceptions=False,
         )
 
@@ -403,6 +427,7 @@ class Prompts:
     def _adapt_prompt_config(self, config: Dict[str, Any], adapter: Adapter) -> Dict[str, Any]:
         adapted_config: Dict[str, Any] = {}
 
+        # NOTE: Should we delete attributes not supported by the provider?
         for attr, value in config.items():
             if attr in _PROMPT_ATTR_TO_ADAPTER_ATTR and adapter in _PROMPT_ATTR_TO_ADAPTER_ATTR[attr][1]:
                 adapted_config[_PROMPT_ATTR_TO_ADAPTER_ATTR[attr][0]] = value

{latitude_sdk-1.0.0 → latitude_sdk-1.0.1}/src/latitude_sdk/sdk/types.py

@@ -216,6 +216,8 @@ class StreamCallbacks(Model):
 
 
 class OnToolCallDetails(Model):
+    id: str
+    name: str
     conversation_uuid: str
     messages: List[Message]
     pause_execution: Callable[[], ToolResult]
@@ -224,7 +226,7 @@ class OnToolCallDetails(Model):
 
 @runtime_checkable
 class OnToolCall(Protocol):
-    async def __call__(self, call: ToolCall, details: OnToolCallDetails) -> ToolResult: ...
+    async def __call__(self, arguments: Dict[str, Any], details: OnToolCallDetails) -> Any: ...
 
 
 @runtime_checkable

{latitude_sdk-1.0.0 → latitude_sdk-1.0.1}/tests/prompts/chat_test.py

@@ -50,7 +50,11 @@ class TestChatPromptSync(TestCase):
         on_event_mock = Mock()
         on_finished_mock = Mock()
         on_error_mock = Mock()
-        actual_tool_mock = AsyncMock(side_effect=fixtures.CONVERSATION_TOOL_RESULTS)
+        actual_tool_mock = AsyncMock(
+            side_effect=[
+                r.result if not r.is_error else Exception(r.result) for r in fixtures.CONVERSATION_TOOL_RESULTS
+            ]
+        )
         other_tool_mock = AsyncMock()
         conversation_uuid = "conversation-uuid"
         messages = self.create_conversation(4)
@@ -101,8 +105,10 @@ class TestChatPromptSync(TestCase):
             self.assertEqual(
                 actual_tool_mock.call_args_list[index][0],
                 (
-                    fixtures.CONVERSATION_TOOL_CALLS[index],
+                    fixtures.CONVERSATION_TOOL_CALLS[index].arguments,
                     OnToolCallDetails.model_construct(
+                        id=fixtures.CONVERSATION_TOOL_CALLS[index].id,
+                        name=fixtures.CONVERSATION_TOOL_CALLS[index].name,
                         conversation_uuid=fixtures.CONVERSATION_FINISHED_EVENT.uuid,
                         messages=fixtures.CONVERSATION_FINISHED_EVENT.conversation,
                         pause_execution=mock.ANY,
@@ -159,8 +165,10 @@ class TestChatPromptSync(TestCase):
             self.assertEqual(
                 actual_tool_mock.call_args_list[index][0],
                 (
-                    fixtures.CONVERSATION_TOOL_CALLS[index],
+                    fixtures.CONVERSATION_TOOL_CALLS[index].arguments,
                     OnToolCallDetails.model_construct(
+                        id=fixtures.CONVERSATION_TOOL_CALLS[index].id,
+                        name=fixtures.CONVERSATION_TOOL_CALLS[index].name,
                         conversation_uuid=fixtures.CONVERSATION_FINISHED_EVENT.uuid,
                         messages=fixtures.CONVERSATION_FINISHED_EVENT.conversation,
                         pause_execution=mock.ANY,
@@ -321,7 +329,11 @@ class TestChatPromptStream(TestCase):
         on_event_mock = Mock()
         on_finished_mock = Mock()
         on_error_mock = Mock()
-        actual_tool_mock = AsyncMock(side_effect=fixtures.CONVERSATION_TOOL_RESULTS)
+        actual_tool_mock = AsyncMock(
+            side_effect=[
+                r.result if not r.is_error else Exception(r.result) for r in fixtures.CONVERSATION_TOOL_RESULTS
+            ]
+        )
         other_tool_mock = AsyncMock()
         conversation_uuid = "conversation-uuid"
         messages = self.create_conversation(4)
@@ -379,8 +391,10 @@ class TestChatPromptStream(TestCase):
             self.assertEqual(
                 actual_tool_mock.call_args_list[index][0],
                 (
-                    fixtures.CONVERSATION_TOOL_CALLS[index],
+                    fixtures.CONVERSATION_TOOL_CALLS[index].arguments,
                     OnToolCallDetails.model_construct(
+                        id=fixtures.CONVERSATION_TOOL_CALLS[index].id,
+                        name=fixtures.CONVERSATION_TOOL_CALLS[index].name,
                         conversation_uuid=fixtures.CONVERSATION_FINISHED_EVENT.uuid,
                         messages=fixtures.CONVERSATION_FINISHED_EVENT.conversation,
                         pause_execution=mock.ANY,
@@ -439,8 +453,10 @@ class TestChatPromptStream(TestCase):
             self.assertEqual(
                 actual_tool_mock.call_args_list[index][0],
                 (
-                    fixtures.CONVERSATION_TOOL_CALLS[index],
+                    fixtures.CONVERSATION_TOOL_CALLS[index].arguments,
                     OnToolCallDetails.model_construct(
+                        id=fixtures.CONVERSATION_TOOL_CALLS[index].id,
+                        name=fixtures.CONVERSATION_TOOL_CALLS[index].name,
                         conversation_uuid=fixtures.CONVERSATION_FINISHED_EVENT.uuid,
                         messages=fixtures.CONVERSATION_FINISHED_EVENT.conversation,
                         pause_execution=mock.ANY,

{latitude_sdk-1.0.0 → latitude_sdk-1.0.1}/tests/prompts/run_test.py

@@ -131,7 +131,11 @@ class TestRunPromptSync(TestCase):
         on_event_mock = Mock()
         on_finished_mock = Mock()
         on_error_mock = Mock()
-        actual_tool_mock = AsyncMock(side_effect=fixtures.CONVERSATION_TOOL_RESULTS)
+        actual_tool_mock = AsyncMock(
+            side_effect=[
+                r.result if not r.is_error else Exception(r.result) for r in fixtures.CONVERSATION_TOOL_RESULTS
+            ]
+        )
         other_tool_mock = AsyncMock()
         path = "prompt-path"
         options = RunPromptOptions(
@@ -188,8 +192,10 @@ class TestRunPromptSync(TestCase):
             self.assertEqual(
                 actual_tool_mock.call_args_list[index][0],
                 (
-                    fixtures.CONVERSATION_TOOL_CALLS[index],
+                    fixtures.CONVERSATION_TOOL_CALLS[index].arguments,
                     OnToolCallDetails.model_construct(
+                        id=fixtures.CONVERSATION_TOOL_CALLS[index].id,
+                        name=fixtures.CONVERSATION_TOOL_CALLS[index].name,
                         conversation_uuid=fixtures.CONVERSATION_FINISHED_EVENT.uuid,
                         messages=fixtures.CONVERSATION_FINISHED_EVENT.conversation,
                         pause_execution=mock.ANY,
@@ -251,8 +257,10 @@ class TestRunPromptSync(TestCase):
             self.assertEqual(
                 actual_tool_mock.call_args_list[index][0],
                 (
-                    fixtures.CONVERSATION_TOOL_CALLS[index],
+                    fixtures.CONVERSATION_TOOL_CALLS[index].arguments,
                     OnToolCallDetails.model_construct(
+                        id=fixtures.CONVERSATION_TOOL_CALLS[index].id,
+                        name=fixtures.CONVERSATION_TOOL_CALLS[index].name,
                         conversation_uuid=fixtures.CONVERSATION_FINISHED_EVENT.uuid,
                         messages=fixtures.CONVERSATION_FINISHED_EVENT.conversation,
                         pause_execution=mock.ANY,
@@ -508,7 +516,11 @@ class TestRunPromptStream(TestCase):
         on_event_mock = Mock()
         on_finished_mock = Mock()
         on_error_mock = Mock()
-        actual_tool_mock = AsyncMock(side_effect=fixtures.CONVERSATION_TOOL_RESULTS)
+        actual_tool_mock = AsyncMock(
+            side_effect=[
+                r.result if not r.is_error else Exception(r.result) for r in fixtures.CONVERSATION_TOOL_RESULTS
+            ]
+        )
         other_tool_mock = AsyncMock()
         path = "prompt-path"
         options = RunPromptOptions(
@@ -572,8 +584,10 @@ class TestRunPromptStream(TestCase):
             self.assertEqual(
                 actual_tool_mock.call_args_list[index][0],
                 (
-                    fixtures.CONVERSATION_TOOL_CALLS[index],
+                    fixtures.CONVERSATION_TOOL_CALLS[index].arguments,
                     OnToolCallDetails.model_construct(
+                        id=fixtures.CONVERSATION_TOOL_CALLS[index].id,
+                        name=fixtures.CONVERSATION_TOOL_CALLS[index].name,
                         conversation_uuid=fixtures.CONVERSATION_FINISHED_EVENT.uuid,
                         messages=fixtures.CONVERSATION_FINISHED_EVENT.conversation,
                         pause_execution=mock.ANY,
@@ -637,8 +651,10 @@ class TestRunPromptStream(TestCase):
             self.assertEqual(
                 actual_tool_mock.call_args_list[index][0],
                 (
-                    fixtures.CONVERSATION_TOOL_CALLS[index],
+                    fixtures.CONVERSATION_TOOL_CALLS[index].arguments,
                     OnToolCallDetails.model_construct(
+                        id=fixtures.CONVERSATION_TOOL_CALLS[index].id,
+                        name=fixtures.CONVERSATION_TOOL_CALLS[index].name,
                         conversation_uuid=fixtures.CONVERSATION_FINISHED_EVENT.uuid,
                         messages=fixtures.CONVERSATION_FINISHED_EVENT.conversation,
                         pause_execution=mock.ANY,

{latitude_sdk-1.0.0 → latitude_sdk-1.0.1}/tests/utils/fixtures.py

@@ -769,13 +769,12 @@ CONVERSATION_TOOL_RESULTS = [
     ToolResult(
         id="toolu_01ARatRfRidTDshkg1UuQhW2",
         name="calculator",
-        result="true",
-        is_error=False,
+        result=True,
     ),
     ToolResult(
         id="toolu_B0398l23AOdTDshkg1UuQhZ3",
         name="calculator",
-        result={"error": "ERRROR_INVALID_EXPRESSION", "message": "Expression is invalid"},
+        result="Expression is invalid",
         is_error=True,
     ),
 ]
@@ -786,8 +785,7 @@ CONVERSATION_TOOL_RESULTS_MESSAGES: List[Message] = [
             ToolResultContent(
                 id="toolu_01ARatRfRidTDshkg1UuQhW2",
                 name="calculator",
-                result="true",
-                is_error=False,
+                result=True,
             ),
         ],
     ),
@@ -796,7 +794,7 @@ CONVERSATION_TOOL_RESULTS_MESSAGES: List[Message] = [
             ToolResultContent(
                 id="toolu_B0398l23AOdTDshkg1UuQhZ3",
                 name="calculator",
-                result={"error": "ERRROR_INVALID_EXPRESSION", "message": "Expression is invalid"},
+                result="Expression is invalid",
                 is_error=True,
             ),
         ],