latitude-sdk 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

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/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.dist-info → latitude_sdk-1.0.1.dist-info}/METADATA

@@ -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.dist-info → latitude_sdk-1.0.1.dist-info}/RECORD

@@ -11,10 +11,10 @@ latitude_sdk/sdk/errors.py,sha256=9GlGdDE8LGy3dE2Ry_BipBg-tDbQx7LWXJfSnTJSSBE,17
 latitude_sdk/sdk/evaluations.py,sha256=fDGtAWjdPG9OuKLit6u-jufVleC1EnshRplK6RN8iyg,2277
 latitude_sdk/sdk/latitude.py,sha256=fdOWrAs0k3puLIhyU_zwvLsMyV_f1RxDWpV65_Euenc,2928
 latitude_sdk/sdk/logs.py,sha256=CyHkRJvPl_p7wTSvR9bgxEI5akS0Tjc9FeQRb2C2vMg,1997
-latitude_sdk/sdk/prompts.py,sha256=w3A-dhE_V1ueYjc8LBRXmQn5XZVCnsMdxs7Rau7P_E0,15818
-latitude_sdk/sdk/types.py,sha256=D6wKkt5WSC0Ih6QT3QFSaU9YoxURKw8GKM7aH_uQ2-Q,6892
+latitude_sdk/sdk/prompts.py,sha256=HXvJVqcn446VI2nlIT2lCuOrNypNqE0x1Sy0l77MDFM,16756
+latitude_sdk/sdk/types.py,sha256=v8K1zzJqPexos08U4cnsHJ3LrkVRlbgMNvwtKuXoxuQ,6922
 latitude_sdk/util/__init__.py,sha256=alIDGBnxWH4JvP-UW-7N99seBBi0r1GV1h8f1ERFBec,21
 latitude_sdk/util/utils.py,sha256=hMOmF-u1QaDgOwXN6ME6n4TaQ70yZKLvijDUqNCMwXI,2844
-latitude_sdk-1.0.0.dist-info/METADATA,sha256=nK2qNPBfe6nAZLSVYkMfaU-ub9Y4flAUXyKc6OMRr2k,2060
-latitude_sdk-1.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-latitude_sdk-1.0.0.dist-info/RECORD,,
+latitude_sdk-1.0.1.dist-info/METADATA,sha256=6eLckfer4mDy1JY6ZzWNJcdh0zk_RmLJ7H2aFEoQfCg,2060
+latitude_sdk-1.0.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+latitude_sdk-1.0.1.dist-info/RECORD,,