lmnr 0.2.2__py3-none-any.whl → 0.2.3__py3-none-any.whl

lmnr/sdk/remote_debugger.py

@@ -1,4 +1,4 @@
-from typing import Callable
+from typing import Callable, Optional
 from websockets.sync.client import connect
 from lmnr.types import NodeInput, RegisterDebuggerRequest, SDKError, ToolCall
 import uuid
@@ -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,8 +28,10 @@ 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)
         with connect(
             self.url,
@@ -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,16 @@ class RemoteDebugger:
                         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:
+    def _format_session_id(self) -> str:
         return \
 f"""
 ========================================
 Debugger Session ID:
-{session_id}
+{self.session_id}
 ========================================
 """
     

{lmnr-0.2.2.dist-info → lmnr-0.2.3.dist-info}/METADATA

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

{lmnr-0.2.2.dist-info → lmnr-0.2.3.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=SsKVBnw1rKHsOp-kdSfi87QMUMjmAf3y4Xi3tFM0GOI,3094
+lmnr/sdk/remote_debugger.py,sha256=eXbfnl0RoBcGCViajIFsY5W28HQIf791HTffbWBNJY4,3156
 lmnr/types.py,sha256=f_46qjFK_jJziNFEzN-ip8bgPHxkohib2xGesNOdyxs,1526
-lmnr-0.2.2.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
-lmnr-0.2.2.dist-info/METADATA,sha256=atAJKnS9lga-I89PGbtuIOJGc3HuRPHY377Xa4RIldA,2258
-lmnr-0.2.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-lmnr-0.2.2.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
-lmnr-0.2.2.dist-info/RECORD,,
+lmnr-0.2.3.dist-info/LICENSE,sha256=67b_wJHVV1CBaWkrKFWU1wyqTPSdzH77Ls-59631COg,10411
+lmnr-0.2.3.dist-info/METADATA,sha256=GdJL6YTZ3VnbuQvkz9kPN5QYrC4wBFdkpwq_Er79P28,2258
+lmnr-0.2.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+lmnr-0.2.3.dist-info/entry_points.txt,sha256=Qg7ZRax4k-rcQsZ26XRYQ8YFSBiyY2PNxYfq4a6PYXI,41
+lmnr-0.2.3.dist-info/RECORD,,