Update abstra-lib

Author: abstra-bot

abstra_internals/cloud_api/__init__.py

@@ -1,7 +1,7 @@
 import json
 from threading import Thread
 from time import sleep
-from typing import Any, Optional
+from typing import Any, Callable, Optional
 
 import requests
 import simple_websocket
@@ -131,7 +131,7 @@ class SessionPathMessage(BaseModel):
     sessionPath: str
 
 
-def connect_tunnel():
+def connect_tunnel(on_public_url_update: Optional[Callable[[], None]]):
     url = f"{CLOUD_API_ENDPOINT}/tunnel/connect".replace("https://", "wss://").replace(
         "http://", "ws://"
     )
@@ -194,6 +194,10 @@ def loop():
                         f"{CLOUD_API_ENDPOINT}/tunnel/forward/{session.sessionPath}"
                     )
                     Settings.set_public_url(public_url)
+
+                    if on_public_url_update:
+                        on_public_url_update()
+
                     print(
                         f"Hooks can also be fired from {Fore.GREEN} {public_url}/_hooks/:hook-path{Fore.RESET}"
                     )

abstra_internals/controllers/workflows.py

@@ -1,6 +1,7 @@
 from typing import Dict, List, Optional, TypedDict
 
 from abstra_internals.controllers.main import UnknownNodeTypeError
+from abstra_internals.controllers.service.roles.client import RoleClientController
 from abstra_internals.repositories.factory import Repositories
 from abstra_internals.repositories.project.project import (
     AgentStage,
@@ -282,4 +283,6 @@ def update_workflow(self, workflow_state_dto: Dict):
 
         ProjectRepository.save(project)
 
+        RoleClientController(self.repos).sync_connection_pool()
+
         return self._make_workflow_dto(project)

abstra_internals/interface/cli/editor.py

@@ -79,6 +79,6 @@ def editor(headless: bool):
     if not headless:
         browser_open_editor()
 
-    connect_tunnel()
+    connect_tunnel(on_public_url_update=role_client_controller.sync_connection_pool)
 
     server.serve_forever()

abstra_internals/server/routes/workflows.py

@@ -29,6 +29,7 @@ def _update_workflow():
             if payload is None:
                 raise Exception("No payload found")
             new_state = controller.update_workflow(payload)
+
             return new_state
         except Exception as e:
             AbstraLogger.capture_exception(e)