cloudbrain-client 1.1.1__tar.gz → 1.1.2__tar.gz

{cloudbrain_client-1.1.1 → cloudbrain_client-1.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cloudbrain-client
-Version: 1.1.1
+Version: 1.1.2
 Summary: CloudBrain Client - AI collaboration and communication system with AI-to-AI collaboration support
 Author: CloudBrain Team
 License: MIT

{cloudbrain_client-1.1.1 → cloudbrain_client-1.1.2}/cloudbrain_client/ai_websocket_client.py

@@ -90,6 +90,15 @@ class AIWebSocketClient:
     async def handle_message(self, data: dict):
         """Handle incoming message"""
         message_type = data.get('type')
+        request_id = data.get('request_id')
+        
+        # Check if this is a response to a request
+        if request_id and request_id in self.message_handlers:
+            handler = self.message_handlers[request_id]
+            if isinstance(handler, asyncio.Future):
+                handler.set_result(data)
+                del self.message_handlers[request_id]
+            return
         
         if message_type == 'new_message':
             await self.handle_new_message(data)
@@ -226,6 +235,47 @@ class AIWebSocketClient:
         
         await self.ws.send(json.dumps(message))
     
+    async def send_request(self, request_type: str, data: dict = None) -> dict:
+        """
+        Send a custom request and wait for response
+        
+        Args:
+            request_type: Type of request
+            data: Optional data dictionary
+        
+        Returns:
+            Response dictionary from server
+        """
+        if not self.connected:
+            return {"error": "Not connected"}
+        
+        # Create a unique request ID
+        request_id = f"req_{request_type}_{id(self)}"
+        
+        # Create a future to wait for response
+        response_future = asyncio.Future()
+        
+        # Store the future in message handlers
+        self.message_handlers[request_id] = response_future
+        
+        # Send the request
+        message = {
+            'type': request_type,
+            'request_id': request_id,
+            **(data or {})
+        }
+        
+        await self.ws.send(json.dumps(message))
+        
+        # Wait for response with timeout
+        try:
+            response = await asyncio.wait_for(response_future, timeout=10.0)
+            return response
+        except asyncio.TimeoutError:
+            return {"error": "Request timeout"}
+        except Exception as e:
+            return {"error": str(e)}
+    
     async def close(self):
         """Close connection"""
         if self.ws:

{cloudbrain_client-1.1.1 → cloudbrain_client-1.1.2}/cloudbrain_client/cloudbrain_collaboration_helper.py

@@ -22,12 +22,30 @@ from cloudbrain_client.ai_websocket_client import AIWebSocketClient
 class CloudBrainCollaborator:
     """Helper class for AI agents to collaborate through CloudBrain"""
     
-    def __init__(self, ai_id: int, server_url: str = 'ws://127.0.0.1:8766'):
+    def __init__(self, ai_id: int, server_url: str = 'ws://127.0.0.1:8766', db_path: str = None):
         self.ai_id = ai_id
         self.server_url = server_url
         self.client = None
         self.connected = False
         self.ai_name = None
+        # Use provided db_path or default to server/ai_db/cloudbrain.db relative to current directory
+        if db_path:
+            self.db_path = Path(db_path)
+        else:
+            # Try to find the database in common locations
+            possible_paths = [
+                Path.cwd() / "server" / "ai_db" / "cloudbrain.db",
+                Path(__file__).parent.parent.parent.parent / "server" / "ai_db" / "cloudbrain.db",
+                Path.home() / "gits" / "hub" / "cloudbrain" / "server" / "ai_db" / "cloudbrain.db",
+            ]
+            self.db_path = None
+            for path in possible_paths:
+                if path.exists():
+                    self.db_path = path
+                    break
+            if self.db_path is None:
+                # Default to the first option even if it doesn't exist yet
+                self.db_path = Path.cwd() / "server" / "ai_db" / "cloudbrain.db"
         
     async def connect(self):
         """Connect to CloudBrain server"""
@@ -59,7 +77,7 @@ class CloudBrainCollaborator:
             return []
         
         try:
-            conn = sqlite3.connect(Path(__file__).parent / "server" / "ai_db" / "cloudbrain.db")
+            conn = sqlite3.connect(self.db_path)
             conn.row_factory = sqlite3.Row
             cursor = conn.cursor()
             
@@ -251,13 +269,13 @@ class CloudBrainCollaborationHelper:
     4. Track - Monitor collaboration progress
     """
     
-    def __init__(self, ai_id: int, ai_name: str = "", server_url: str = 'ws://127.0.0.1:8766'):
+    def __init__(self, ai_id: int, ai_name: str = "", server_url: str = 'ws://127.0.0.1:8766', db_path: str = None):
         self.ai_id = ai_id
         self.ai_name = ai_name
         self.server_url = server_url
         self.client = None
         self.connected = False
-        self._collaborator = CloudBrainCollaborator(ai_id, server_url)
+        self._collaborator = CloudBrainCollaborator(ai_id, server_url, db_path)
         
     async def connect(self):
         """Connect to CloudBrain server"""
@@ -319,7 +337,7 @@ class CloudBrainCollaborationHelper:
             return {"error": "Not connected"}
         
         try:
-            conn = sqlite3.connect(Path(__file__).parent / "server" / "ai_db" / "cloudbrain.db")
+            conn = sqlite3.connect(self._collaborator.db_path)
             conn.row_factory = sqlite3.Row
             cursor = conn.cursor()
             
@@ -357,6 +375,27 @@ class CloudBrainCollaborationHelper:
         except Exception as e:
             print(f"❌ Error getting collaboration progress: {e}")
             return {"error": str(e)}
+    
+    async def _send_request(self, request_type: str, data: dict) -> dict:
+        """
+        Send a custom request to the CloudBrain server
+        
+        Args:
+            request_type: Type of request (e.g., 'brain_save_state', 'brain_load_state')
+            data: Dictionary of request data
+        
+        Returns:
+            Response dictionary from server
+        """
+        if not self.connected or not self._collaborator.client:
+            return {"error": "Not connected to server"}
+        
+        try:
+            response = await self._collaborator.client.send_request(request_type, data)
+            return response
+        except Exception as e:
+            print(f"❌ Error sending request: {e}")
+            return {"error": str(e)}
 
 
 async def integrate_cloudbrain_to_tasks(ai_id: int, tasks: List[Dict[str, Any]]) -> bool:

{cloudbrain_client-1.1.1 → cloudbrain_client-1.1.2}/cloudbrain_client.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cloudbrain-client
-Version: 1.1.1
+Version: 1.1.2
 Summary: CloudBrain Client - AI collaboration and communication system with AI-to-AI collaboration support
 Author: CloudBrain Team
 License: MIT

{cloudbrain_client-1.1.1 → cloudbrain_client-1.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "cloudbrain-client"
-version = "1.1.1"
+version = "1.1.2"
 description = "CloudBrain Client - AI collaboration and communication system with AI-to-AI collaboration support"
 readme = "README.md"
 requires-python = ">=3.8"