lmnr 0.2.2__tar.gz → 0.2.3.1__tar.gz

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

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

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

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

{lmnr-0.2.2 → lmnr-0.2.3.1}/src/lmnr/sdk/remote_debugger.py

@@ -1,6 +1,6 @@
-from typing import Callable
+from typing import Callable, Optional
 from websockets.sync.client import connect
-from lmnr.types import NodeInput, RegisterDebuggerRequest, SDKError, ToolCall
+from lmnr.types import DeregisterDebuggerRequest, NodeInput, RegisterDebuggerRequest, SDKError, ToolCall
 import uuid
 import json
 from threading import Thread
@@ -14,9 +14,11 @@ class RemoteDebugger:
         self.stop_flag = False
         self.session = None
     
-    def start(self):
+    def start(self) -> Optional[str]:
         self.stop_flag = False
+        self.session = self._generate_session_id()
         self.thread.start()
+        return self.session
     
     def stop(self):
         self.stop_flag = True
@@ -26,9 +28,11 @@ class RemoteDebugger:
         # in case the user wants to start the debugger again
         self.thread = Thread(target=self._run)
     
+    def get_session_id(self) -> str:
+        return self.session
+    
     def _run(self):
-        self.session = self._generate_session_id()
-        request = RegisterDebuggerRequest(debugger_session_id=self.session)
+        request = RegisterDebuggerRequest(debuggerSessionId=self.session)
         with connect(
             self.url,
             additional_headers={
@@ -36,6 +40,7 @@ class RemoteDebugger:
             }
         ) as websocket:
             websocket.send(request.model_dump_json())
+            print(self._format_session_id())
             while not self.stop_flag:
                 try:
                     # blocks the thread until a message is received or a timeout (3 seconds) occurs
@@ -66,20 +71,17 @@ class RemoteDebugger:
                         pass
                     response = tool(**arguments)
                     websocket.send(json.dumps(response))
-    
-
-    def get_session_id(self) -> str:
-        return self._format_session_id(self.session)
+            websocket.send(DeregisterDebuggerRequest(debuggerSessionId=self.session, deregister=True).model_dump_json())
 
     def _generate_session_id(self) -> str:
         return uuid.uuid4().urn[9:]
 
-    def _format_session_id(self, session_id: str) -> str:
+    def _format_session_id(self) -> str:
         return \
 f"""
 ========================================
 Debugger Session ID:
-{session_id}
+{self.session}
 ========================================
 """
     

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

@@ -51,5 +51,10 @@ class ToolCall(pydantic.BaseModel):
     type: Optional[str]
     function: ToolCallRequest
 
+# TODO: allow snake_case and manually convert to camelCase
 class RegisterDebuggerRequest(pydantic.BaseModel):
-    debugger_session_id: str
+    debuggerSessionId: str
+
+class DeregisterDebuggerRequest(pydantic.BaseModel):
+    debuggerSessionId: str
+    deregister: bool