unitlab 2.3.40__tar.gz → 2.3.41__tar.gz

{unitlab-2.3.40/src/unitlab.egg-info → unitlab-2.3.41}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unitlab
-Version: 2.3.40
+Version: 2.3.41
 Home-page: https://github.com/teamunitlab/unitlab-sdk
 Author: Unitlab Inc.
 Author-email: team@unitlab.ai

{unitlab-2.3.40 → unitlab-2.3.41}/setup.py

@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
 
 setup(
     name="unitlab",
-    version="2.3.40",
+    version="2.3.41",
     license="MIT",
     author="Unitlab Inc.",
     author_email="team@unitlab.ai",

{unitlab-2.3.40 → unitlab-2.3.41}/src/unitlab/persistent_tunnel.py

@@ -14,7 +14,7 @@ from fastapi import FastAPI
 import uvicorn
 import threading
 import psutil
-
+import secrets
 
 api = FastAPI()
 
@@ -67,23 +67,23 @@ class PersistentTunnel:
             "Content-Type": "application/json"                                                                                          
         } 
     
-    def get_or_create_tunnel(self):
-        """Always create a new tunnel with unique name to avoid conflicts"""
-        # Generate unique tunnel name to avoid conflicts
-        import uuid
-        unique_suffix = str(uuid.uuid4())[:8]
-        self.tunnel_name = "agent-{}-{}".format(self.device_id, unique_suffix)
-        print("🔧 Creating tunnel: {}...".format(self.tunnel_name))
-        
-        # Always create new tunnel
-        return self.create_new_tunnel()
-    
+    # def get_or_create_tunnel(self):
+    #     """Always create a new tunnel with unique name to avoid conflicts"""
+    #     # Generate unique tunnel name to avoid conflicts
+    #     import uuid
+    #     unique_suffix = str(uuid.uuid4())[:8]
+    #     self.tunnel_name = "agent-{}-{}".format(self.device_id, unique_suffix)
+    #     print("🔧 Creating tunnel: {}...".format(self.tunnel_name))
+        
+    #     # Always create new tunnel
+    #     return self.create_new_tunnel()
+
+
     def create_new_tunnel(self):
-        """Create a brand new tunnel"""
+        """Create a new tunnel via Cloudflare API"""
         print("🔧 Creating new tunnel: {}...".format(self.tunnel_name))
         
         # Generate random tunnel secret (32 bytes)
-        import secrets
         tunnel_secret = base64.b64encode(secrets.token_bytes(32)).decode()
         
         url = "https://api.cloudflare.com/client/v4/accounts/{}/cfd_tunnel".format(self.cf_account_id)
@@ -107,12 +107,13 @@ class PersistentTunnel:
                 "TunnelID": self.tunnel_id
             }
             
-            # Save credentials to file with tunnel name (not ID) for consistency
+            # Save credentials to file
             cred_file = "/tmp/tunnel-{}.json".format(self.tunnel_name)
             with open(cred_file, 'w') as f:
                 json.dump(self.tunnel_credentials, f)
             
             print("✅ Tunnel created: {}".format(self.tunnel_id))
+            print("✅ Credentials saved to: {}".format(cred_file))
             return cred_file
         else:
             print("❌ Failed to create tunnel: {}".format(response.text))
@@ -440,7 +441,7 @@ class PersistentTunnel:
             # API credentials are hardcoded, so we're ready to go
             
             # 1. Get existing or create new tunnel via API
-            cred_file = self.get_or_create_tunnel()
+            cred_file = self.create_new_tunnel()
         
             
             # 2. Create DNS record
@@ -490,6 +491,12 @@ class PersistentTunnel:
             self.jupyter_process.terminate()
         if self.tunnel_process:
             self.tunnel_process.terminate()
+            try:
+                self.tunnel_process.wait(timeout=5)
+            except subprocess.TimeoutExpired:
+                self.tunnel_process.kill()
+                self.tunnel_process.wait()
+            print("✅ Tunnel stopped")
         
     #     # Optionally delete tunnel when stopping
     #     if self.tunnel_id:

{unitlab-2.3.40 → unitlab-2.3.41/src/unitlab.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unitlab
-Version: 2.3.40
+Version: 2.3.41
 Home-page: https://github.com/teamunitlab/unitlab-sdk
 Author: Unitlab Inc.
 Author-email: team@unitlab.ai