lmnr 0.2.4__py3-none-any.whl → 0.2.5__py3-none-any.whl

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/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

{lmnr-0.2.4.dist-info → lmnr-0.2.5.dist-info}/METADATA

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

@@ -19,10 +19,10 @@ lmnr/cli/{{cookiecutter.lmnr_pipelines_dir_name}}/pipelines/{{cookiecutter.pipel
 lmnr/cli/{{cookiecutter.lmnr_pipelines_dir_name}}/pipelines/{{cookiecutter.pipeline_dir_name}}/{{cookiecutter.pipeline_dir_name}}.py,sha256=WG-ZMofPpGXCx5jdWVry3_XBzcKjqn8ZycFSiWEOBPg,2858
 lmnr/cli/{{cookiecutter.lmnr_pipelines_dir_name}}/types.py,sha256=iWuflMV7TiaBPs6-B-BlrovvWpZgHGGHK0v8rSqER7A,997
 lmnr/sdk/endpoint.py,sha256=tT6-w-mwbh4BAwnj5G0pCVE_Sz8EUzZmpBtacm_T2pE,6359
-lmnr/sdk/remote_debugger.py,sha256=lAL7t0DXADLMTLsh6AN3rqVYf_A-aeG_mMbyU-G2wTk,3299
-lmnr/types.py,sha256=Pi6R5qMmN4fiwZR0TzP4rxfSP6rFv4iouZ_OKnJsmhA,1685
-lmnr-0.2.4.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.2.4.dist-info/METADATA,sha256=BUDTL9WKd18EieHeb47or0gwjGd1dkSz0nJsvx6AFiw,5494
-lmnr-0.2.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-lmnr-0.2.4.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
-lmnr-0.2.4.dist-info/RECORD,,
+lmnr/sdk/remote_debugger.py,sha256=pNSmmk-KAAOUygKQAeTB1DbB3vvlpHL8JLwPBpBhBwA,3892
+lmnr/types.py,sha256=Y41GGiCVemBUCTKedCwDhxjdsoO8uENr3LRwfdp9MI0,1771
+lmnr-0.2.5.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.2.5.dist-info/METADATA,sha256=BgYsaYsxprvRZ5-RxZk8DQdSTKmF51hOvU27hg_0wn4,5492
+lmnr-0.2.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+lmnr-0.2.5.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
+lmnr-0.2.5.dist-info/RECORD,,