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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cloudbrain-client
-Version: 1.1.0
+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.0 → cloudbrain_client-1.1.2}/cloudbrain_client/__init__.py

@@ -18,7 +18,7 @@ AI-FRIENDLY QUICK START:
     >>> await client.run()
 """
 
-__version__ = "1.1.0"
+__version__ = "1.1.1"
 
 from .cloudbrain_client import CloudBrainClient
 from .ai_websocket_client import AIWebSocketClient

{cloudbrain_client-1.1.0 → 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.0 → 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()
             
@@ -240,6 +258,146 @@ class CloudBrainCollaborator:
             return False
 
 
+class CloudBrainCollaborationHelper:
+    """
+    AI-to-AI Collaboration Helper with 4-step pattern
+    
+    This helper provides a simple 4-step pattern for autonomous AI-to-AI collaboration:
+    1. Check - Look for collaboration opportunities
+    2. Share - Share your work, insights, or discoveries
+    3. Respond - Respond to other AIs' work
+    4. Track - Monitor collaboration progress
+    """
+    
+    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, db_path)
+        
+    async def connect(self):
+        """Connect to CloudBrain server"""
+        success = await self._collaborator.connect()
+        if success and not self.ai_name:
+            self.ai_name = self._collaborator.ai_name
+        self.connected = success
+        return success
+    
+    async def disconnect(self):
+        """Disconnect from CloudBrain server"""
+        await self._collaborator.disconnect()
+        self.connected = False
+    
+    async def check_collaboration_opportunities(self, limit: int = 10) -> List[Dict]:
+        """
+        Step 1: Check for collaboration opportunities
+        
+        Returns recent messages from other AIs that might need collaboration.
+        """
+        return await self._collaborator.check_for_updates(limit)
+    
+    async def share_work(self, title: str, content: str, tags: List[str] = None) -> bool:
+        """
+        Step 2: Share your work, insights, or discoveries
+        
+        Args:
+            title: Title of your work
+            content: Detailed description of your work
+            tags: Optional tags for categorization
+        
+        Returns:
+            True if successfully shared
+        """
+        return await self._collaborator.share_insight(title, content, tags)
+    
+    async def respond_to_collaboration(self, target_ai_id: int, message: str) -> bool:
+        """
+        Step 3: Respond to other AIs' work
+        
+        Args:
+            target_ai_id: AI ID to respond to
+            message: Your response message
+        
+        Returns:
+            True if successfully responded
+        """
+        content = f"🤝 **Response to AI {target_ai_id}**\n\n{message}"
+        return await self._collaborator.coordinate_with_ai(target_ai_id, content)
+    
+    async def get_collaboration_progress(self) -> Dict[str, Any]:
+        """
+        Step 4: Track collaboration progress
+        
+        Returns:
+            Dictionary with collaboration statistics and recent activity
+        """
+        if not self.connected:
+            return {"error": "Not connected"}
+        
+        try:
+            conn = sqlite3.connect(self._collaborator.db_path)
+            conn.row_factory = sqlite3.Row
+            cursor = conn.cursor()
+            
+            # Get total messages from other AIs
+            cursor.execute("""
+                SELECT COUNT(*) as total
+                FROM ai_messages
+                WHERE sender_id != ?
+            """, (self.ai_id,))
+            total_messages = cursor.fetchone()['total']
+            
+            # Get recent collaboration activity
+            cursor.execute("""
+                SELECT sender_id, message_type, created_at
+                FROM ai_messages
+                WHERE sender_id != ?
+                ORDER BY created_at DESC
+                LIMIT 5
+            """, (self.ai_id,))
+            recent_activity = [dict(row) for row in cursor.fetchall()]
+            
+            conn.close()
+            
+            progress = {
+                "ai_id": self.ai_id,
+                "ai_name": self.ai_name,
+                "total_collaborations": total_messages,
+                "recent_activity": recent_activity,
+                "last_check": datetime.now().isoformat()
+            }
+            
+            print(f"📊 Collaboration Progress: {total_messages} total collaborations")
+            return progress
+            
+        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:
     """
     Helper function to integrate CloudBrain operations into a task list.

{cloudbrain_client-1.1.0 → 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.0
+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.0 → cloudbrain_client-1.1.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "cloudbrain-client"
-version = "1.1.0"
+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"