cloudbrain-client 1.1.3__py3-none-any.whl → 1.1.5__py3-none-any.whl

cloudbrain_client/ai_websocket_client.py

@@ -17,13 +17,13 @@ from typing import Optional, Callable
 class AIWebSocketClient:
     """Generic WebSocket client for AI communication"""
     
-    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', ai_name: str = None):
         self.ai_id = ai_id
         self.server_url = server_url
         self.ws = None
         self.connected = False
         self.message_handlers = []
-        self.ai_name = None
+        self.ai_name = ai_name
         self.ai_expertise = None
         self.ai_version = None
         
@@ -33,10 +33,12 @@ class AIWebSocketClient:
             print(f"🔗 Connecting to {self.server_url}...")
             self.ws = await websockets.connect(self.server_url)
             
-            # Authenticate - libsql simulator expects just ai_id
+            # Authenticate - send ai_id and ai_name (for auto-assignment)
             auth_msg = {
                 'ai_id': self.ai_id
             }
+            if self.ai_name:
+                auth_msg['ai_name'] = self.ai_name
             await self.ws.send(json.dumps(auth_msg))
             
             # Wait for welcome message

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, self.ai_name)
+            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
     

{cloudbrain_client-1.1.3.dist-info → cloudbrain_client-1.1.5.dist-info}/METADATA

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

{cloudbrain_client-1.1.3.dist-info → cloudbrain_client-1.1.5.dist-info}/RECORD

@@ -1,12 +1,12 @@
 cloudbrain_client/__init__.py,sha256=-0LOV9olyyutc_zO-7YH4vSmq2fe1YiGS0HUT69i0Xs,7366
 cloudbrain_client/ai_conversation_helper.py,sha256=FJ2DhVBoH7jsbJ0Cd2fAJBuOi2oxWdsdJzU4A0qScQA,22104
-cloudbrain_client/ai_websocket_client.py,sha256=a41Zwg6401qLFqQperkfc5TOPTaxO3C00OAGqsQbQw4,16200
+cloudbrain_client/ai_websocket_client.py,sha256=XfbNVXfZYpvCg3E-FTvvtyTAc7lAzaVc80IoxXyfB7U,16313
 cloudbrain_client/cloudbrain_client.py,sha256=20AYB27rQI6G42jNiO7OpUTJu235wUA95DcWUcJfye8,25997
-cloudbrain_client/cloudbrain_collaboration_helper.py,sha256=L9LTr64wjfEi5esMYv6LTV5rTN3w5Sr_mK9AOVd7CdI,21607
+cloudbrain_client/cloudbrain_collaboration_helper.py,sha256=x13bRdF3QCv6LGWDsYybSuRDzadljbnF_YmL2ixI-gw,22311
 cloudbrain_client/cloudbrain_quick.py,sha256=sC7em8X61YiXXlMtvepUJPiwEz_2SaWBNGj_d-m2Ff4,4324
 cloudbrain_client/message_poller.py,sha256=flo3vfPQEGImLTlW7eYAlbOHmDUwdJ5LgMT4V8vPyTU,7055
-cloudbrain_client-1.1.3.dist-info/METADATA,sha256=cxDtX4ZROYBSui5IeyFdvhbzmNAb6Rh6AbcQMM6YQB0,7533
-cloudbrain_client-1.1.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-cloudbrain_client-1.1.3.dist-info/entry_points.txt,sha256=ES0E1Al-dyBoKksvgjg6jjgcU4eoIq_ZWvBBvcJp_kY,113
-cloudbrain_client-1.1.3.dist-info/top_level.txt,sha256=ksJ13MTscvck0-1Y6ADFYFzho5swJf-wY-n5r5IYZsU,18
-cloudbrain_client-1.1.3.dist-info/RECORD,,
+cloudbrain_client-1.1.5.dist-info/METADATA,sha256=OKFmTfj2HYvuSQfqZnwolqCeFvF8BpT0lF37P5F4QvE,7533
+cloudbrain_client-1.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+cloudbrain_client-1.1.5.dist-info/entry_points.txt,sha256=ES0E1Al-dyBoKksvgjg6jjgcU4eoIq_ZWvBBvcJp_kY,113
+cloudbrain_client-1.1.5.dist-info/top_level.txt,sha256=ksJ13MTscvck0-1Y6ADFYFzho5swJf-wY-n5r5IYZsU,18
+cloudbrain_client-1.1.5.dist-info/RECORD,,