lmnr 0.2.4__tar.gz → 0.2.5__tar.gz

{lmnr-0.2.4 → lmnr-0.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lmnr
-Version: 0.2.4
+Version: 0.2.5
 Summary: Python SDK for Laminar AI
 License: Apache-2.0
 Author: lmnr.ai
@@ -121,9 +121,8 @@ Example:
 from lmnr import LaminarRemoteDebugger, NodeInput
 
 # adding **kwargs is safer, in case an LLM produces more arguments than needed
-def my_tool(arg1: string, arg2: string, **kwargs) -> NodeInput {
+def my_tool(arg1: string, arg2: string, **kwargs) -> NodeInput:
     return f'{arg1}&{arg2}'
-}
 
 debugger = LaminarRemoteDebugger('<YOUR_PROJECT_API_KEY>', [my_tool])
 session_id = debugger.start()  # the session id will also be printed to console
@@ -184,3 +183,4 @@ print(f"RESULT:\n{res}")
 ## PROJECT_API_KEY
 
 Read more [here](https://docs.lmnr.ai/api-reference/introduction#authentication) on how to get `PROJECT_API_KEY`.
+

{lmnr-0.2.4 → lmnr-0.2.5}/README.md

@@ -98,9 +98,8 @@ Example:
 from lmnr import LaminarRemoteDebugger, NodeInput
 
 # adding **kwargs is safer, in case an LLM produces more arguments than needed
-def my_tool(arg1: string, arg2: string, **kwargs) -> NodeInput {
+def my_tool(arg1: string, arg2: string, **kwargs) -> NodeInput:
     return f'{arg1}&{arg2}'
-}
 
 debugger = LaminarRemoteDebugger('<YOUR_PROJECT_API_KEY>', [my_tool])
 session_id = debugger.start()  # the session id will also be printed to console
@@ -160,4 +159,4 @@ print(f"RESULT:\n{res}")
 
 ## PROJECT_API_KEY
 
-Read more [here](https://docs.lmnr.ai/api-reference/introduction#authentication) on how to get `PROJECT_API_KEY`.
+Read more [here](https://docs.lmnr.ai/api-reference/introduction#authentication) on how to get `PROJECT_API_KEY`.

{lmnr-0.2.4 → lmnr-0.2.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lmnr"
-version = "0.2.4"
+version = "0.2.5"
 description = "Python SDK for Laminar AI"
 authors = [
   { name = "lmnr.ai", email = "founders@lmnr.ai" }
@@ -11,7 +11,7 @@ license = "Apache-2.0"
 
 [tool.poetry]
 name = "lmnr"
-version = "0.2.4"
+version = "0.2.5"
 description = "Python SDK for Laminar AI"
 authors = ["lmnr.ai"]
 readme = "README.md"

{lmnr-0.2.4 → lmnr-0.2.5}/src/lmnr/sdk/remote_debugger.py

@@ -1,6 +1,7 @@
 from typing import Callable, Optional
 from websockets.sync.client import connect
-from lmnr.types import DeregisterDebuggerRequest, NodeInput, RegisterDebuggerRequest, SDKError, ToolCall
+import websockets
+from lmnr.types import DeregisterDebuggerRequest, NodeInput, RegisterDebuggerRequest, SDKError, ToolCall, ToolCallError
 import uuid
 import json
 from threading import Thread
@@ -47,6 +48,9 @@ class RemoteDebugger:
                     message = websocket.recv(3)
                 except TimeoutError:
                     continue
+                except websockets.exceptions.ConnectionClosedError:
+                    print("Connection closed. Please restart the debugger.")
+                    return
                 try:
                     tool_call = ToolCall.model_validate_json(message)
                 except:
@@ -56,11 +60,11 @@ class RemoteDebugger:
                     if tool.__name__ == tool_call.function.name
                 ]
                 if not matching_tools:
-                    raise SDKError(
-                        f'Tool {tool_call.function.name} not found.'
-                        ' Registered tools: '
-                        f'{", ".join([tool.__name__ for tool in self.tools])}'
-                    )
+                    error_message = f'Tool {tool_call.function.name} not found.' +\
+                    f' Registered tools: {", ".join([tool.__name__ for tool in self.tools])}'
+                    e = ToolCallError(error=error_message)
+                    websocket.send(e.model_dump_json())
+                    continue
                 tool = matching_tools[0]
                 if tool.__name__ == tool_call.function.name:
                     # default the arguments to an empty dictionary
@@ -69,8 +73,13 @@ class RemoteDebugger:
                         arguments = json.loads(tool_call.function.arguments)
                     except:
                         pass
-                    response = tool(**arguments)
-                    websocket.send(json.dumps(response))
+                    try:
+                        response = tool(**arguments)  # of type NodeInput
+                        websocket.send(json.dumps(response))
+                    except Exception as e:
+                        error_message = f'Error occurred while running tool {tool.__name__}: {e}'
+                        e = ToolCallError(error=error_message)
+                        websocket.send(e.model_dump_json())
             websocket.send(DeregisterDebuggerRequest(debuggerSessionId=self.session, deregister=True).model_dump_json())
 
     def _generate_session_id(self) -> str:

{lmnr-0.2.4 → lmnr-0.2.5}/src/lmnr/types.py

@@ -39,8 +39,8 @@ class EndpointRunError(Exception):
             return super().__str__()
         
 class SDKError(Exception):
-    def __init__(self, error_mesasge: str):
-        super().__init__(error_mesasge)
+    def __init__(self, error_message: str):
+        super().__init__(error_message)
 
 class ToolCallRequest(pydantic.BaseModel):
     name: str
@@ -51,6 +51,10 @@ class ToolCall(pydantic.BaseModel):
     type: Optional[str]
     function: ToolCallRequest
 
+ToolCallResponse = NodeInput
+class ToolCallError(pydantic.BaseModel):
+    error: str
+
 # TODO: allow snake_case and manually convert to camelCase
 class RegisterDebuggerRequest(pydantic.BaseModel):
     debuggerSessionId: str