unitlab 2.3.40__py3-none-any.whl → 2.3.42__py3-none-any.whl

unitlab/main.py

@@ -122,7 +122,7 @@ def dataset_download(
 def run_agent(
     api_key: API_KEY,
     device_id: Annotated[str, typer.Option(help="Device ID")] = None,
-    base_domain: Annotated[str, typer.Option(help="Base domain for tunnels")] = "1scan.uz",
+    base_domain: Annotated[str, typer.Option(help="Base domain for tunnels")] = "api-dev.unitlab.ai",
   
 ):
 

unitlab/persistent_tunnel.py

@@ -14,7 +14,7 @@ from fastapi import FastAPI
 import uvicorn
 import threading
 import psutil
-
+import secrets
 
 api = FastAPI()
 
@@ -38,7 +38,7 @@ class PersistentTunnel:
         
         # Clean device ID for subdomain
         if device_id:
-            self.device_id = device_id.replace('-', '').replace('_', '').replace('.', '').lower()[:20]
+            self.device_id = device_id.replace('_', '').replace('.', '').lower()[:30]
         else:
             import uuid
             self.device_id = str(uuid.uuid4())[:8]
@@ -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))
@@ -150,26 +151,26 @@ class PersistentTunnel:
             return False
         
         # First, check if SSH DNS record exists and delete it
-        print("🔍 Checking for existing SSH DNS record: {}.{}".format(self.ssh_subdomain, self.domain))
-        list_url = "{}?name={}.{}".format(url, self.ssh_subdomain, self.domain)
-        list_response = requests.get(list_url, headers=headers)
-        
-        if list_response.status_code == 200:
-            records = list_response.json().get("result", [])
-            print("Found {} existing DNS records".format(len(records)))
-            print('this is new version')
-            for record in records:
-                if record["name"] == "{}.{}".format(self.ssh_subdomain, self.domain):
-                    record_id = record["id"]
-                    print("🗑️  Deleting old SSH DNS record: {}".format(record_id))
-                    delete_url = "{}/{}".format(url, record_id)
-                    delete_response = requests.delete(delete_url, headers=headers)
-                    if delete_response.status_code in [200, 204]:
-                        print("✅ Deleted old SSH DNS record")
-                    else:
-                        print("⚠️  Could not delete old SSH DNS record: {}".format(delete_response.text[:200]))
-        else:
-            print("⚠️  Could not list DNS records: {}".format(list_response.text[:200]))
+        # print("🔍 Checking for existing SSH DNS record: {}.{}".format(self.ssh_subdomain, self.domain))
+        # list_url = "{}?name={}.{}".format(url, self.ssh_subdomain, self.domain)
+        # list_response = requests.get(list_url, headers=headers)
+        
+        # if list_response.status_code == 200:
+        #     records = list_response.json().get("result", [])
+        #     print("Found {} existing DNS records".format(len(records)))
+        #     print('this is new version')
+        #     for record in records:
+        #         if record["name"] == "{}.{}".format(self.ssh_subdomain, self.domain):
+        #             record_id = record["id"]
+        #             print("🗑️  Deleting old SSH DNS record: {}".format(record_id))
+        #             delete_url = "{}/{}".format(url, record_id)
+        #             delete_response = requests.delete(delete_url, headers=headers)
+        #             if delete_response.status_code in [200, 204]:
+        #                 print("✅ Deleted old SSH DNS record")
+        #             else:
+        #                 print("⚠️  Could not delete old SSH DNS record: {}".format(delete_response.text[:200]))
+        # else:
+        #     print("⚠️  Could not list DNS records: {}".format(list_response.text[:200]))
         
         # Wait a moment for DNS deletion to propagate
         time.sleep(2)
@@ -262,15 +263,20 @@ class PersistentTunnel:
             f.write("credentials-file: {}\n\n".format(cred_file))
             f.write("ingress:\n")
 
-            # SSH service on dedicated subdomain (s{deviceid}.unitlab-ai.com)
+                  # SSH service on dedicated subdomain (s{deviceid}.unitlab-ai.com)
             f.write("  - hostname: {}.{}\n".format(self.ssh_subdomain, self.domain))
             f.write("    service: ssh://localhost:22\n")
 
+      
+
+
             # API (more specific path goes first)
             f.write("  - hostname: {}.{}\n".format(self.subdomain, self.domain))
             f.write("    path: /api-agent/*\n")
             f.write("    service: http://localhost:8001\n")
 
+      
+
             # Jupyter (general hostname for HTTP)
             f.write("  - hostname: {}.{}\n".format(self.subdomain, self.domain))
             f.write("    service: http://localhost:8888\n")
@@ -440,7 +446,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 +496,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.dist-info → unitlab-2.3.42.dist-info}/METADATA

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

unitlab-2.3.42.dist-info/RECORD

@@ -0,0 +1,13 @@
+unitlab/__init__.py,sha256=Wtk5kQ_MTlxtd3mxJIn2qHVK5URrVcasMMPjD3BtrVM,214
+unitlab/__main__.py,sha256=6Hs2PV7EYc5Tid4g4OtcLXhqVHiNYTGzSBdoOnW2HXA,29
+unitlab/client.py,sha256=wqREtDuYc5ixeloPEGm0hp1sdUtB59sB1bIJjBcO1y0,25983
+unitlab/exceptions.py,sha256=68Tr6LreEzjQ3Vns8HAaWdtewtkNUJOvPazbf6NSnXU,950
+unitlab/main.py,sha256=QA1xv1D-sA4Igga2knulJ99g66aaKFyjQCR2pyyo70w,4431
+unitlab/persistent_tunnel.py,sha256=pBdmHrgXS-7CgQjq429W2DzflW8OfkeW9qQwu4r6z4g,22640
+unitlab/utils.py,sha256=9gPRu-d6pbhSoVdll1GXe4eoz_uFYOSbYArFDQdlUZs,1922
+unitlab-2.3.42.dist-info/licenses/LICENSE.md,sha256=Gn7RRvByorAcAaM-WbyUpsgi5ED1-bKFFshbWfYYz2Y,1069
+unitlab-2.3.42.dist-info/METADATA,sha256=fl1snGGNMX5N7sKFRIx1KT0tGGniTy_dM1iZ7tVtXX0,1046
+unitlab-2.3.42.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+unitlab-2.3.42.dist-info/entry_points.txt,sha256=ig-PjKEqSCj3UTdyANgEi4tsAU84DyXdaOJ02NHX4bY,45
+unitlab-2.3.42.dist-info/top_level.txt,sha256=Al4ZlTYE3fTJK2o6YLCDMH5_DjuQkffRBMxgmWbKaqQ,8
+unitlab-2.3.42.dist-info/RECORD,,

unitlab-2.3.40.dist-info/RECORD

@@ -1,13 +0,0 @@
-unitlab/__init__.py,sha256=Wtk5kQ_MTlxtd3mxJIn2qHVK5URrVcasMMPjD3BtrVM,214
-unitlab/__main__.py,sha256=6Hs2PV7EYc5Tid4g4OtcLXhqVHiNYTGzSBdoOnW2HXA,29
-unitlab/client.py,sha256=wqREtDuYc5ixeloPEGm0hp1sdUtB59sB1bIJjBcO1y0,25983
-unitlab/exceptions.py,sha256=68Tr6LreEzjQ3Vns8HAaWdtewtkNUJOvPazbf6NSnXU,950
-unitlab/main.py,sha256=RbuCUaExvOEoA33T9O0hnr94R-L70RXyMjQshxRWR-o,4421
-unitlab/persistent_tunnel.py,sha256=NkXoTK3yn2mVN3hHoUqSQI-8ph_yfBU_YgEFumynyvo,22313
-unitlab/utils.py,sha256=9gPRu-d6pbhSoVdll1GXe4eoz_uFYOSbYArFDQdlUZs,1922
-unitlab-2.3.40.dist-info/licenses/LICENSE.md,sha256=Gn7RRvByorAcAaM-WbyUpsgi5ED1-bKFFshbWfYYz2Y,1069
-unitlab-2.3.40.dist-info/METADATA,sha256=Wy1EHazXknuZ_jJYBjcKB_iXOvP3X4Rtt0MEUeeSKwc,1046
-unitlab-2.3.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-unitlab-2.3.40.dist-info/entry_points.txt,sha256=ig-PjKEqSCj3UTdyANgEi4tsAU84DyXdaOJ02NHX4bY,45
-unitlab-2.3.40.dist-info/top_level.txt,sha256=Al4ZlTYE3fTJK2o6YLCDMH5_DjuQkffRBMxgmWbKaqQ,8
-unitlab-2.3.40.dist-info/RECORD,,