pycityagent 2.0.0a12__py3-none-any.whl → 2.0.0a13__py3-none-any.whl

pycityagent/agent.py

@@ -427,7 +427,9 @@ class InstitutionAgent(Agent):
             simulator,
             memory,
         )
-
+        # 添加响应收集器
+        self._gather_responses: Dict[str, asyncio.Future] = {}
+        
     async def bind_to_simulator(self):
         await self._bind_to_economy()
 
@@ -514,18 +516,47 @@ class InstitutionAgent(Agent):
 
     async def handle_gather_message(self, payload: dict):
         """处理收到的消息，识别发送者"""
-        # 从消息中解析发送者 ID 和消息内容
         content = payload["content"]
         sender_id = payload["from"]
-        print(
-            f"Agent {self._uuid} received gather message: '{content}' from Agent {sender_id}"
-        )
-
-    async def gather_messages(self, agent_ids: list[UUID], target: str):
-        """从多个智能体收集消息"""
+        
+        # 将响应存储到对应的Future中
+        response_key = str(sender_id)
+        if response_key in self._gather_responses:
+            self._gather_responses[response_key].set_result({
+                "from": sender_id,
+                "content": content,
+            })
+
+    async def gather_messages(self, agent_ids: list[UUID], target: str) -> List[dict]:
+        """从多个智能体收集消息
+        
+        Args:
+            agent_ids: 目标智能体ID列表
+            target: 要收集的信息类型
+            
+        Returns:
+            List[dict]: 收集到的所有响应
+        """
+        # 为每个agent创建Future
+        futures = {}
+        for agent_id in agent_ids:
+            response_key = str(agent_id)
+            futures[response_key] = asyncio.Future()
+            self._gather_responses[response_key] = futures[response_key]
+            
+        # 发送gather请求
         payload = {
             "from": self._uuid,
             "target": target,
         }
         for agent_id in agent_ids:
             await self._send_message(agent_id, payload, "gather")
+            
+        try:
+            # 等待所有响应
+            responses = await asyncio.gather(*futures.values())
+            return responses
+        finally:
+            # 清理Future
+            for key in futures:
+                self._gather_responses.pop(key, None)

pycityagent/simulation/agentgroup.py

@@ -71,7 +71,8 @@ class AgentGroup:
                 topic = f"exps/{self.exp_id}/agents/{agent._uuid}/gather"
                 await self.messager.subscribe(topic, agent)
         self.initialized = True
-
+        self.message_dispatch_task = asyncio.create_task(self.message_dispatch())
+        
     async def gather(self, content: str):
         results = {}
         for agent in self.agents:
@@ -82,49 +83,46 @@ class AgentGroup:
         agent = self.id2agent[target_agent_uuid]
         await agent.memory.update(target_key, content)
 
+    async def message_dispatch(self):
+        while True:
+            if not self.messager.is_connected():
+                logging.warning("Messager is not connected. Skipping message processing.")
+
+            # Step 1: 获取消息
+            messages = await self.messager.fetch_messages()
+            logging.info(f"Group {self._uuid} received {len(messages)} messages")
+
+            # Step 2: 分发消息到对应的 Agent
+            for message in messages:
+                topic = message.topic.value
+                payload = message.payload
+
+                # 添加解码步骤，将bytes转换为str
+                if isinstance(payload, bytes):
+                    payload = payload.decode("utf-8")
+                    payload = json.loads(payload)
+
+                # 提取 agent_id（主题格式为 "exps/{exp_id}/agents/{agent_uuid}/{topic_type}"）
+                _, _, _, agent_uuid, topic_type = topic.strip("/").split("/")
+                    
+                if uuid.UUID(agent_uuid) in self.id2agent:
+                    agent = self.id2agent[uuid.UUID(agent_uuid)]
+                    # topic_type: agent-chat, user-chat, user-survey, gather
+                    if topic_type == "agent-chat":
+                        await agent.handle_agent_chat_message(payload)
+                    elif topic_type == "user-chat":
+                        await agent.handle_user_chat_message(payload)
+                    elif topic_type == "user-survey":
+                        await agent.handle_user_survey_message(payload)
+                    elif topic_type == "gather":
+                        await agent.handle_gather_message(payload)
+
+            await asyncio.sleep(1)
+
     async def step(self):
         if not self.initialized:
             await self.init_agents()
 
-        # Step 1: 如果 Messager 无法连接，则跳过消息接收
-        if not self.messager.is_connected():
-            logging.warning("Messager is not connected. Skipping message processing.")
-            # 跳过接收和分发消息
-            tasks = [agent.run() for agent in self.agents]
-            await asyncio.gather(*tasks)
-            return
-
-        # Step 2: 从 Messager 获取消息
-        messages = await self.messager.fetch_messages()
-
-        logging.info(f"Group {self._uuid} received {len(messages)} messages")
-
-        # Step 3: 分发消息到对应的 Agent
-        for message in messages:
-            topic = message.topic.value
-            payload = message.payload
-
-            # 添加解码步骤，将bytes转换为str
-            if isinstance(payload, bytes):
-                payload = payload.decode("utf-8")
-                payload = json.loads(payload)
-
-            # 提取 agent_id（主题格式为 "exps/{exp_id}/agents/{agent_uuid}/{topic_type}"）
-            _, _, _, agent_uuid, topic_type = topic.strip("/").split("/")
-                
-            if uuid.UUID(agent_uuid) in self.id2agent:
-                agent = self.id2agent[uuid.UUID(agent_uuid)]
-                # topic_type: agent-chat, user-chat, user-survey, gather
-                if topic_type == "agent-chat":
-                    await agent.handle_agent_chat_message(payload)
-                elif topic_type == "user-chat":
-                    await agent.handle_user_chat_message(payload)
-                elif topic_type == "user-survey":
-                    await agent.handle_user_survey_message(payload)
-                elif topic_type == "gather":
-                    await agent.handle_gather_message(payload)
-
-        # Step 4: 调用每个 Agent 的运行逻辑
         tasks = [agent.run() for agent in self.agents]
         await asyncio.gather(*tasks)
 

{pycityagent-2.0.0a12.dist-info → pycityagent-2.0.0a13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycityagent
-Version: 2.0.0a12
+Version: 2.0.0a13
 Summary: LLM-based城市环境agent构建库
 License: MIT
 Author: Yuwei Yan

{pycityagent-2.0.0a12.dist-info → pycityagent-2.0.0a13.dist-info}/RECORD

@@ -1,5 +1,5 @@
 pycityagent/__init__.py,sha256=n56bWkAUEcvjDsb7LcJpaGjlrriSKPnR0yBhwRfEYBA,212
-pycityagent/agent.py,sha256=h0p9G5bLlfiv4yB8bkc9l2Z4jpa7s4uYVwAT4IrxusY,17396
+pycityagent/agent.py,sha256=7BCHK5OH1TvW16JLiYR0iWAiWGBuP4eJsiGU1y7exbY,18417
 pycityagent/economy/__init__.py,sha256=aonY4WHnx-6EGJ4WKrx4S-2jAkYNLtqUA04jp6q8B7w,75
 pycityagent/economy/econ_client.py,sha256=qQb_kZneEXGBRaS_y5Jdoi95I8GyjKEsDSC4s6V6R7w,10829
 pycityagent/environment/__init__.py,sha256=awHxlOud-btWbk0FCS4RmGJ13W84oVCkbGfcrhKqihA,240
@@ -46,7 +46,7 @@ pycityagent/memory/utils.py,sha256=wLNlNlZ-AY9VB8kbUIy0UQSYh26FOQABbhmKQkit5o8,8
 pycityagent/message/__init__.py,sha256=TCjazxqb5DVwbTu1fF0sNvaH_EPXVuj2XQ0p6W-QCLU,55
 pycityagent/message/messager.py,sha256=Iv4pK83JvHAQSZyGNACryPBey2wRoiok3Hb1eIwHbww,2506
 pycityagent/simulation/__init__.py,sha256=jYaqaNpzM5M_e_ykISS_M-mIyYdzJXJWhgpfBpA6l5k,111
-pycityagent/simulation/agentgroup.py,sha256=DzZooUZstnhObO3X7NjB9-8zBofEu_1NevaIhspuGME,6113
+pycityagent/simulation/agentgroup.py,sha256=fM93ydsuZ_w8YVnHnX1tWskWVYmt6sztKjzgDXcg1Xw,6106
 pycityagent/simulation/interview.py,sha256=S2uv8MFCB4-u_4Q202VFoPJOIleqpKK9Piju0BDSb_0,1158
 pycityagent/simulation/simulation.py,sha256=4BHavfSHPM81Y5Su-MzzBYM0sR9cHRk4YOv-VkRkzrA,11595
 pycityagent/simulation/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
@@ -65,6 +65,6 @@ pycityagent/workflow/block.py,sha256=6EmiRMLdOZC1wMlmLMIjfrp9TuiI7Gw4s3nnXVMbrnw
 pycityagent/workflow/prompt.py,sha256=tY69nDO8fgYfF_dOA-iceR8pAhkYmCqoox8uRPqEuGY,2956
 pycityagent/workflow/tool.py,sha256=wD9WZ5rma6HCKugtHTwbShNE0f-Rjlwvn_1be3fCAsk,6682
 pycityagent/workflow/trigger.py,sha256=t5X_i0WtL32bipZSsq_E3UUyYYudYLxQUpvxbgClp2s,5683
-pycityagent-2.0.0a12.dist-info/METADATA,sha256=DNsYXJBF2nVXYdiNJrhDty3AwVMkqmPbnm_Ld-Vtt8k,7622
-pycityagent-2.0.0a12.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-pycityagent-2.0.0a12.dist-info/RECORD,,
+pycityagent-2.0.0a13.dist-info/METADATA,sha256=MdIVm_sOeGM99wtIKpcDEt6zt3Gp-loHI5MxGTz_6og,7622
+pycityagent-2.0.0a13.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pycityagent-2.0.0a13.dist-info/RECORD,,