cloudbrain-client 1.1.2__py3-none-any.whl → 1.1.4__py3-none-any.whl

cloudbrain_client/cloudbrain_collaboration_helper.py

@@ -51,7 +51,7 @@ class CloudBrainCollaborator:
         """Connect to CloudBrain server"""
         try:
             self.client = AIWebSocketClient(self.ai_id, self.server_url)
-            await self.client.connect(start_message_loop=False)
+            await self.client.connect(start_message_loop=True)
             self.connected = True
             self.ai_name = self.client.ai_name
             print(f"✅ Connected to CloudBrain as {self.ai_name} (AI {self.ai_id})")
@@ -275,18 +275,36 @@ class CloudBrainCollaborationHelper:
         self.server_url = server_url
         self.client = None
         self.connected = False
+        self._message_loop_task = None
         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
+        try:
+            self.client = AIWebSocketClient(self.ai_id, self.server_url)
+            await self.client.connect(start_message_loop=False)
+            self.connected = True
+            self.ai_name = self.client.ai_name
+            print(f"✅ Connected to CloudBrain as {self.ai_name} (AI {self.ai_id})")
+            
+            # Start message loop in background
+            self._message_loop_task = asyncio.create_task(self.client.message_loop())
+            
+            return True
+        except Exception as e:
+            print(f"❌ Connection error: {e}")
+            return False
     
     async def disconnect(self):
         """Disconnect from CloudBrain server"""
+        # Cancel message loop task
+        if self._message_loop_task:
+            self._message_loop_task.cancel()
+            try:
+                await self._message_loop_task
+            except asyncio.CancelledError:
+                pass
+        
         await self._collaborator.disconnect()
         self.connected = False
     
@@ -396,6 +414,76 @@ class CloudBrainCollaborationHelper:
         except Exception as e:
             print(f"❌ Error sending request: {e}")
             return {"error": str(e)}
+    
+    async def get_documentation(self, doc_id: int = None, title: str = None, category: str = None) -> Optional[Dict]:
+        """
+        Get documentation from CloudBrain
+        
+        Args:
+            doc_id: Documentation ID
+            title: Documentation title
+            category: Documentation category (gets most recent in category)
+        
+        Returns:
+            Documentation dictionary or None
+        """
+        data = {}
+        if doc_id:
+            data['doc_id'] = doc_id
+        elif title:
+            data['title'] = title
+        elif category:
+            data['category'] = category
+        
+        response = await self._send_request('documentation_get', data)
+        
+        if response and response.get('type') == 'documentation':
+            return response.get('documentation')
+        
+        return None
+    
+    async def list_documentation(self, category: str = None, limit: int = 50) -> List[Dict]:
+        """
+        List available documentation
+        
+        Args:
+            category: Filter by category
+            limit: Maximum number of results
+        
+        Returns:
+            List of documentation summaries
+        """
+        data = {'limit': limit}
+        if category:
+            data['category'] = category
+        
+        response = await self._send_request('documentation_list', data)
+        
+        if response and response.get('type') == 'documentation_list':
+            return response.get('documents', [])
+        
+        return []
+    
+    async def search_documentation(self, query: str, limit: int = 20) -> List[Dict]:
+        """
+        Search documentation using full-text search
+        
+        Args:
+            query: Search query
+            limit: Maximum number of results
+        
+        Returns:
+            List of matching documents with snippets
+        """
+        response = await self._send_request('documentation_search', {
+            'query': query,
+            'limit': limit
+        })
+        
+        if response and response.get('type') == 'documentation_search_results':
+            return response.get('results', [])
+        
+        return []
 
 
 async def integrate_cloudbrain_to_tasks(ai_id: int, tasks: List[Dict[str, Any]]) -> bool:

{cloudbrain_client-1.1.2.dist-info → cloudbrain_client-1.1.4.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: cloudbrain-client
-Version: 1.1.2
-Summary: CloudBrain Client - AI collaboration and communication system with AI-to-AI collaboration support
+Version: 1.1.4
+Summary: CloudBrain Client - AI collaboration and communication system with AI-to-AI collaboration support and documentation access
 Author: CloudBrain Team
 License: MIT
 Project-URL: Homepage, https://github.com/cloudbrain-project/cloudbrain

{cloudbrain_client-1.1.2.dist-info → cloudbrain_client-1.1.4.dist-info}/RECORD

@@ -2,11 +2,11 @@ cloudbrain_client/__init__.py,sha256=-0LOV9olyyutc_zO-7YH4vSmq2fe1YiGS0HUT69i0Xs
 cloudbrain_client/ai_conversation_helper.py,sha256=FJ2DhVBoH7jsbJ0Cd2fAJBuOi2oxWdsdJzU4A0qScQA,22104
 cloudbrain_client/ai_websocket_client.py,sha256=a41Zwg6401qLFqQperkfc5TOPTaxO3C00OAGqsQbQw4,16200
 cloudbrain_client/cloudbrain_client.py,sha256=20AYB27rQI6G42jNiO7OpUTJu235wUA95DcWUcJfye8,25997
-cloudbrain_client/cloudbrain_collaboration_helper.py,sha256=703QqL1svx7evySO2aUVUBXXiCRrdvmHnQ9JNRjjT80,19439
+cloudbrain_client/cloudbrain_collaboration_helper.py,sha256=q17khympO1UF3pHAhgE8zGZ8seBWj7aKUV1ILuKG1m8,22297
 cloudbrain_client/cloudbrain_quick.py,sha256=sC7em8X61YiXXlMtvepUJPiwEz_2SaWBNGj_d-m2Ff4,4324
 cloudbrain_client/message_poller.py,sha256=flo3vfPQEGImLTlW7eYAlbOHmDUwdJ5LgMT4V8vPyTU,7055
-cloudbrain_client-1.1.2.dist-info/METADATA,sha256=-csFGzDYXKdKwSB6klebbJqvni1tGrOUSrIXYhiTjfA,7508
-cloudbrain_client-1.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-cloudbrain_client-1.1.2.dist-info/entry_points.txt,sha256=ES0E1Al-dyBoKksvgjg6jjgcU4eoIq_ZWvBBvcJp_kY,113
-cloudbrain_client-1.1.2.dist-info/top_level.txt,sha256=ksJ13MTscvck0-1Y6ADFYFzho5swJf-wY-n5r5IYZsU,18
-cloudbrain_client-1.1.2.dist-info/RECORD,,
+cloudbrain_client-1.1.4.dist-info/METADATA,sha256=9PdgLG-F6gc4fj564sEnJDtCZiybTEO2B_G5bPLd9C4,7533
+cloudbrain_client-1.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+cloudbrain_client-1.1.4.dist-info/entry_points.txt,sha256=ES0E1Al-dyBoKksvgjg6jjgcU4eoIq_ZWvBBvcJp_kY,113
+cloudbrain_client-1.1.4.dist-info/top_level.txt,sha256=ksJ13MTscvck0-1Y6ADFYFzho5swJf-wY-n5r5IYZsU,18
+cloudbrain_client-1.1.4.dist-info/RECORD,,