ttsd-colabcli 1.0.3 → 1.0.4

package/core/daemon.py

@@ -106,7 +106,19 @@ async def launch_worker(email: str, public_url: str):
     
     try:
         from app.colab_cli.runtime import ColabRuntime
-        _worker_bytes = open(os.path.join(os.path.dirname(__file__), "../colab/worker.py"), "rb").read()
+        # Try local first (satellite_node/daemon.py), fallback to packaged structure
+        possible_paths = [
+            os.path.join(os.path.dirname(__file__), "../colab/worker.py"),
+            os.path.join(os.path.dirname(__file__), "colab", "worker.py")
+        ]
+        _worker_bytes = None
+        for p in possible_paths:
+            if os.path.exists(p):
+                _worker_bytes = open(p, "rb").read()
+                break
+        if not _worker_bytes:
+            raise FileNotFoundError(f"Could not find worker.py in {possible_paths}")
+
         _worker_b64 = base64.b64encode(_worker_bytes).decode()
         
         _rt = ColabRuntime(url, token, session_name=name)
@@ -133,7 +145,18 @@ print("DEPLOYED PID=" + str(proc.pid) + " ALIVE=" + str(ret is None), flush=True
         await loop.run_in_executor(None, lambda: _rt.execute_code(_deploy_code, timeout=300))
         logger.info("Worker deployed on Colab runtime for %s", email)
     except Exception as e:
-        logger.error("Worker deploy failed: %s", e)
+        logger.error("Worker deploy failed: %s, cleaning up Colab assignment", e)
+        # Rollback: unassign the runtime so quota isn't wasted
+        try:
+            from app.colab_cli.common import kill_process
+            if pid:
+                kill_process(pid)
+        except Exception:
+            pass
+        try:
+            await loop.run_in_executor(None, _client.unassign, endpoint)
+        except Exception:
+            pass
         await report_status(email, "FAILED", error=str(e))
         return
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ttsd-colabcli",
-  "version": "1.0.3",
+  "version": "1.0.4",
   "description": "Satellite Node CLI for TTS Dubbing Distributed Architecture",
   "main": "cli.js",
   "bin": {