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

cloudbrain_client/ai_websocket_client.py

@@ -17,13 +17,14 @@ 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.message_handlers = {}  # For request-response pattern
+        self.registered_handlers = []  # For general message handlers
+        self.ai_name = ai_name
         self.ai_expertise = None
         self.ai_version = None
         
@@ -33,10 +34,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
@@ -120,7 +123,7 @@ class AIWebSocketClient:
             print(f"⚠️  Unknown message type: {message_type}")
         
         # Call registered handlers
-        for handler in self.message_handlers:
+        for handler in self.registered_handlers:
             try:
                 await handler(data)
             except Exception as e:

cloudbrain_client/cloudbrain_collaboration_helper.py

@@ -281,7 +281,7 @@ class CloudBrainCollaborationHelper:
     async def connect(self):
         """Connect to CloudBrain server"""
         try:
-            self.client = AIWebSocketClient(self.ai_id, self.server_url)
+            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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cloudbrain-client
-Version: 1.1.4
+Version: 1.1.6
 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.4.dist-info → cloudbrain_client-1.1.6.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=AjNR1tgt8766n5iXgghMyIgplsgxifA2V1c1lq2gYyg,16418
 cloudbrain_client/cloudbrain_client.py,sha256=20AYB27rQI6G42jNiO7OpUTJu235wUA95DcWUcJfye8,25997
-cloudbrain_client/cloudbrain_collaboration_helper.py,sha256=q17khympO1UF3pHAhgE8zGZ8seBWj7aKUV1ILuKG1m8,22297
+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.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,,
+cloudbrain_client-1.1.6.dist-info/METADATA,sha256=hKnuFtxgKWmoj-U1vE3iHAj2PcKAJOvtY8k4uk20u_A,7533
+cloudbrain_client-1.1.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+cloudbrain_client-1.1.6.dist-info/entry_points.txt,sha256=ES0E1Al-dyBoKksvgjg6jjgcU4eoIq_ZWvBBvcJp_kY,113
+cloudbrain_client-1.1.6.dist-info/top_level.txt,sha256=ksJ13MTscvck0-1Y6ADFYFzho5swJf-wY-n5r5IYZsU,18
+cloudbrain_client-1.1.6.dist-info/RECORD,,