lmnr 0.2.1__tar.gz → 0.2.2__tar.gz

{lmnr-0.2.1 → lmnr-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lmnr
-Version: 0.2.1
+Version: 0.2.2
 Summary: Python SDK for Laminar AI
 License: Apache-2.0
 Author: lmnr.ai

{lmnr-0.2.1 → lmnr-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lmnr"
-version = "0.2.1"
+version = "0.2.2"
 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.1"
+version = "0.2.2"
 description = "Python SDK for Laminar AI"
 authors = ["lmnr.ai"]
 readme = "README.md"

{lmnr-0.2.1 → lmnr-0.2.2}/src/lmnr/__init__.py

@@ -4,4 +4,5 @@ from .types import (
     EndpointRunError,
     EndpointRunResponse,
     NodeInput
-)
+)
+from .sdk.remote_debugger import RemoteDebugger as LaminarRemoteDebugger

lmnr-0.2.2/src/lmnr/sdk/remote_debugger.py

@@ -0,0 +1,85 @@
+from typing import Callable
+from websockets.sync.client import connect
+from lmnr.types import NodeInput, RegisterDebuggerRequest, SDKError, ToolCall
+import uuid
+import json
+from threading import Thread
+
+class RemoteDebugger:
+    def __init__(self, project_api_key: str, tools: list[Callable[..., NodeInput]] = []):
+        self.project_api_key = project_api_key
+        self.url = 'wss://api.lmnr.ai/v2/endpoint/ws'
+        self.tools = tools
+        self.thread = Thread(target=self._run)
+        self.stop_flag = False
+        self.session = None
+    
+    def start(self):
+        self.stop_flag = False
+        self.thread.start()
+    
+    def stop(self):
+        self.stop_flag = True
+        self.thread.join()
+        self.session = None
+        # python allows running threads only once, so we need to create a new thread
+        # in case the user wants to start the debugger again
+        self.thread = Thread(target=self._run)
+    
+    def _run(self):
+        self.session = self._generate_session_id()
+        request = RegisterDebuggerRequest(debugger_session_id=self.session)
+        with connect(
+            self.url,
+            additional_headers={
+                'Authorization': f'Bearer {self.project_api_key}'
+            }
+        ) as websocket:
+            websocket.send(request.model_dump_json())
+            while not self.stop_flag:
+                try:
+                    # blocks the thread until a message is received or a timeout (3 seconds) occurs
+                    message = websocket.recv(3)
+                except TimeoutError:
+                    continue
+                try:
+                    tool_call = ToolCall.model_validate_json(message)
+                except:
+                    raise SDKError(f'Invalid message received:\n{message}')
+                matching_tools = [
+                    tool for tool in self.tools
+                    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])}'
+                    )
+                tool = matching_tools[0]
+                if tool.__name__ == tool_call.function.name:
+                    # default the arguments to an empty dictionary
+                    arguments = {}
+                    try:
+                        arguments = json.loads(tool_call.function.arguments)
+                    except:
+                        pass
+                    response = tool(**arguments)
+                    websocket.send(json.dumps(response))
+    
+
+    def get_session_id(self) -> str:
+        return self._format_session_id(self.session)
+
+    def _generate_session_id(self) -> str:
+        return uuid.uuid4().urn[9:]
+
+    def _format_session_id(self, session_id: str) -> str:
+        return \
+f"""
+========================================
+Debugger Session ID:
+{session_id}
+========================================
+"""
+    

{lmnr-0.2.1 → lmnr-0.2.2}/src/lmnr/types.py

@@ -49,4 +49,7 @@ class ToolCallRequest(pydantic.BaseModel):
 class ToolCall(pydantic.BaseModel):
     id: Optional[str]
     type: Optional[str]
-    function: ToolCallRequest
+    function: ToolCallRequest
+
+class RegisterDebuggerRequest(pydantic.BaseModel):
+    debugger_session_id: str