pycityagent 2.0.0a5__py3-none-any.whl → 2.0.0a7__py3-none-any.whl

pycityagent/memory/profile.py

@@ -3,8 +3,7 @@ Agent Profile
 """
 
 from copy import deepcopy
-from typing import (Any, Callable, Dict, List, Optional, Sequence, Tuple,
-                    Union, cast)
+from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union, cast
 
 from ..utils.decorators import lock_decorator
 from .const import *

pycityagent/memory/self_define.py

@@ -3,8 +3,7 @@ Self Define Data
 """
 
 from copy import deepcopy
-from typing import (Any, Callable, Dict, List, Optional, Sequence, Tuple,
-                    Union, cast)
+from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union, cast
 
 from ..utils.decorators import lock_decorator
 from .const import *

pycityagent/memory/state.py

@@ -3,8 +3,7 @@ Agent State
 """
 
 from copy import deepcopy
-from typing import (Any, Callable, Dict, List, Optional, Sequence, Tuple,
-                    Union, cast)
+from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union, cast
 
 from ..utils.decorators import lock_decorator
 from .const import *

pycityagent/message/__init__.py

@@ -1,3 +1,3 @@
 from .messager import Messager
 
-__all__ = ["Messager"]
+__all__ = ["Messager"]

pycityagent/message/messager.py

@@ -4,9 +4,14 @@ import logging
 import math
 from aiomqtt import Client
 
+
 class Messager:
-    def __init__(self, broker, port=1883, timeout=math.inf):
-        self.client = Client(broker, port=port, timeout=timeout)
+    def __init__(
+        self, hostname, port=1883, username=None, password=None, timeout=math.inf
+    ):
+        self.client = Client(
+            hostname, port=port, username=username, password=password, timeout=timeout
+        )
         self.connected = False  # 是否已连接标志
         self.message_queue = asyncio.Queue()  # 用于存储接收到的消息
         self.subscribers = {}  # 订阅者信息，topic -> Agent 映射
@@ -31,7 +36,9 @@ class Messager:
 
     async def subscribe(self, topic, agent):
         if not self.is_connected():
-            logging.error(f"Cannot subscribe to {topic} because not connected to the Broker.")
+            logging.error(
+                f"Cannot subscribe to {topic} because not connected to the Broker."
+            )
             return
         await self.client.subscribe(topic)
         self.subscribers[topic] = agent
@@ -48,7 +55,7 @@ class Messager:
         while not self.message_queue.empty():
             messages.append(await self.message_queue.get())
         return messages
-    
+
     async def send_message(self, topic: str, payload: str, sender_id: int):
         """通过 Messager 发送消息，包含发送者 ID"""
         # 构造消息，payload 中加入 sender_id 以便接收者识别

pycityagent/simulation/__init__.py

@@ -4,4 +4,4 @@
 
 from .simulation import AgentSimulation
 
-__all__ = ["AgentSimulation"] 
+__all__ = ["AgentSimulation"]

pycityagent/simulation/agentgroup.py

@@ -8,12 +8,19 @@ from pycityagent.llm.llm import LLM
 from pycityagent.llm.llmconfig import LLMConfig
 from pycityagent.message import Messager
 
+
 @ray.remote
 class AgentGroup:
-    def __init__(self, agents: list[Agent], config: dict):
+    def __init__(self, agents: list[Agent], config: dict, exp_id: str):
         self.agents = agents
         self.config = config
-        self.messager = Messager(config["simulator_request"]["mqtt"]["server"], config["simulator_request"]["mqtt"]["port"])
+        self.exp_id = exp_id
+        self.messager = Messager(
+            hostname=config["simulator_request"]["mqtt"]["server"],
+            port=config["simulator_request"]["mqtt"]["port"],
+            username=config["simulator_request"]["mqtt"].get("username", None),
+            password=config["simulator_request"]["mqtt"].get("password", None),
+        )
         self.initialized = False
 
         # Step:1 prepare LLM client
@@ -27,9 +34,12 @@ class AgentGroup:
 
         # Step:3 prepare Economy client
         logging.info("-----Creating Economy client in remote...")
-        self.economy_client = EconomyClient(config["simulator_request"]["economy"]['server'])
+        self.economy_client = EconomyClient(
+            config["simulator_request"]["economy"]["server"]
+        )
 
         for agent in self.agents:
+            agent.set_exp_id(self.exp_id)
             agent.set_llm_client(self.llm)
             agent.set_simulator(self.simulator)
             agent.set_economy_client(self.economy_client)
@@ -44,7 +54,7 @@ class AgentGroup:
             await self.messager.start_listening()
             for agent in self.agents:
                 agent.set_messager(self.messager)
-                topic = f"/agents/{agent._agent_id}/chat"
+                topic = f"/exps/{self.exp_id}/agents/{agent._agent_id}/chat"
                 await self.messager.subscribe(topic, agent)
         self.initialized = True
 
@@ -70,10 +80,9 @@ class AgentGroup:
 
             # 添加解码步骤，将bytes转换为str
             if isinstance(payload, bytes):
-                payload = payload.decode('utf-8')
-
-            # 提取 agent_id（主题格式为 "/agents/{agent_id}/chat"）
-            _, agent_id, _ = topic.strip("/").split("/")
+                payload = payload.decode("utf-8")
+            # 提取 agent_id（主题格式为 "/exps/{exp_id}/agents/{agent_id}/chat"）
+            _, _, _, agent_id, _ = topic.strip("/").split("/")
             agent_id = int(agent_id)
 
             if agent_id in self.id2agent:
@@ -95,15 +104,14 @@ class AgentGroup:
             start_time = await self.simulator.get_time()
             # 计算结束时间（秒）
             end_time = start_time + day * 24 * 3600  # 将天数转换为秒
-            
+
             while True:
                 current_time = await self.simulator.get_time()
                 if current_time >= end_time:
                     break
-                
+
                 await self.step()
 
         except Exception as e:
             logging.error(f"模拟器运行错误: {str(e)}")
             raise
-

pycityagent/simulation/interview.py

@@ -2,20 +2,24 @@ from dataclasses import dataclass
 from datetime import datetime
 from typing import List, Optional
 
+
 @dataclass
 class InterviewRecord:
     """采访记录"""
+
     timestamp: datetime
     agent_name: str
     question: str
     response: str
     blocking: bool
 
+
 class InterviewManager:
     """采访管理器"""
+
     def __init__(self):
         self._history: List[InterviewRecord] = []
-        
+
     def add_record(self, agent_name: str, question: str, response: str, blocking: bool):
         """添加采访记录"""
         record = InterviewRecord(
@@ -23,14 +27,14 @@ class InterviewManager:
             agent_name=agent_name,
             question=question,
             response=response,
-            blocking=blocking
+            blocking=blocking,
         )
         self._history.append(record)
-        
+
     def get_agent_history(self, agent_name: str) -> List[InterviewRecord]:
         """获取指定智能体的采访历史"""
         return [r for r in self._history if r.agent_name == agent_name]
-    
+
     def get_recent_history(self, limit: int = 10) -> List[InterviewRecord]:
         """获取最近的采访记录"""
-        return sorted(self._history, key=lambda x: x.timestamp, reverse=True)[:limit] 
+        return sorted(self._history, key=lambda x: x.timestamp, reverse=True)[:limit]

pycityagent/simulation/simulation.py

@@ -1,33 +1,36 @@
 import asyncio
 import json
 import logging
+import uuid
 from datetime import datetime
 import random
 from typing import Dict, List, Optional, Callable
 from mosstool.map._map_util.const import AOI_START_ID
 
-from pycityagent.economy.econ_client import EconomyClient
-from pycityagent.environment.simulator import Simulator
 from pycityagent.memory.memory import Memory
-from pycityagent.message.messager import Messager
 
 from ..agent import Agent
 from .interview import InterviewManager
 from .survey import QuestionType, SurveyManager
 from .ui import InterviewUI
 from .agentgroup import AgentGroup
+
 logger = logging.getLogger(__name__)
 
 
 class AgentSimulation:
     """城市智能体模拟器"""
-    def __init__(self, agent_class: type[Agent], config: dict, agent_prefix: str = "agent_"):
+
+    def __init__(
+        self, agent_class: type[Agent], config: dict, agent_prefix: str = "agent_"
+    ):
         """
         Args:
             agent_class: 智能体类
             config: 配置
             agent_prefix: 智能体名称前缀
         """
+        self.exp_id = uuid.uuid4()
         self.agent_class = agent_class
         self.config = config
         self.agent_prefix = agent_prefix
@@ -42,16 +45,21 @@ class AgentSimulation:
         self._blocked_agents: List[str] = []  # 新增：持续阻塞的智能体列表
         self._survey_manager = SurveyManager()
 
-    async def init_agents(self, agent_count: int, group_size: int = 1000, memory_config_func: Callable = None) -> None:
+    async def init_agents(
+        self,
+        agent_count: int,
+        group_size: int = 1000,
+        memory_config_func: Callable = None,
+    ) -> None:
         """初始化智能体
-        
+
         Args:
             agent_count: 要创建的总智能体数量
             group_size: 每个组的智能体数量，每一个组为一个独立的ray actor
             memory_config_func: 返回Memory配置的函数，需要返回(EXTRA_ATTRIBUTES, PROFILE, BASE)元组
         """
         if memory_config_func is None:
-            memory_config_func = self.default_memory_config_func        
+            memory_config_func = self.default_memory_config_func
 
         for i in range(agent_count):
             agent_name = f"{self.agent_prefix}{i}"
@@ -59,11 +67,9 @@ class AgentSimulation:
             # 获取Memory配置
             extra_attributes, profile, base = memory_config_func()
             memory = Memory(
-                config=extra_attributes,
-                profile=profile.copy(),
-                base=base.copy()
+                config=extra_attributes, profile=profile.copy(), base=base.copy()
             )
-            
+
             # 创建智能体时传入Memory配置
             agent = self.agent_class(
                 name=agent_name,
@@ -74,55 +80,103 @@ class AgentSimulation:
 
         # 计算需要的组数,向上取整以处理不足一组的情况
         num_group = (agent_count + group_size - 1) // group_size
-        
+
         for i in range(num_group):
             # 计算当前组的起始和结束索引
             start_idx = i * group_size
             end_idx = min((i + 1) * group_size, agent_count)
-            
+
             # 获取当前组的agents
             agents = list(self._agents.values())[start_idx:end_idx]
             group_name = f"{self.agent_prefix}_group_{i}"
-            group = AgentGroup.remote(agents, self.config)
+            group = AgentGroup.remote(agents, self.config, self.exp_id)
             self._groups[group_name] = group
 
     def default_memory_config_func(self):
         """默认的Memory配置函数"""
         EXTRA_ATTRIBUTES = {
             # 需求信息
-            "needs": (dict, {
-                'hungry': random.random(),  # 饥饿感
-                'tired': random.random(),   # 疲劳感
-                'safe': random.random(),    # 安全需
-                'social': random.random(),  # 社会需求
-            }, True),
+            "needs": (
+                dict,
+                {
+                    "hungry": random.random(),  # 饥饿感
+                    "tired": random.random(),  # 疲劳感
+                    "safe": random.random(),  # 安全需
+                    "social": random.random(),  # 社会需求
+                },
+                True,
+            ),
             "current_need": (str, "none", True),
             "current_plan": (list, [], True),
             "current_step": (dict, {"intention": "", "type": ""}, True),
-            "execution_context" : (dict, {}, True),
+            "execution_context": (dict, {}, True),
             "plan_history": (list, [], True),
         }
 
         PROFILE = {
             "gender": random.choice(["male", "female"]),
-            "education": random.choice(["Doctor", "Master", "Bachelor", "College", "High School"]),
+            "education": random.choice(
+                ["Doctor", "Master", "Bachelor", "College", "High School"]
+            ),
             "consumption": random.choice(["sightly low", "low", "medium", "high"]),
-            "occupation": random.choice(["Student", "Teacher", "Doctor", "Engineer", "Manager", "Businessman", "Artist", "Athlete", "Other"]),
+            "occupation": random.choice(
+                [
+                    "Student",
+                    "Teacher",
+                    "Doctor",
+                    "Engineer",
+                    "Manager",
+                    "Businessman",
+                    "Artist",
+                    "Athlete",
+                    "Other",
+                ]
+            ),
             "age": random.randint(18, 65),
-            "skill": random.choice(["Good at problem-solving", "Good at communication", "Good at creativity", "Good at teamwork", "Other"]),
+            "skill": random.choice(
+                [
+                    "Good at problem-solving",
+                    "Good at communication",
+                    "Good at creativity",
+                    "Good at teamwork",
+                    "Other",
+                ]
+            ),
             "family_consumption": random.choice(["low", "medium", "high"]),
-            "personality": random.choice(["outgoint", "introvert", "ambivert", "extrovert"]),
+            "personality": random.choice(
+                ["outgoint", "introvert", "ambivert", "extrovert"]
+            ),
             "income": random.randint(1000, 10000),
             "currency": random.randint(10000, 100000),
             "residence": random.choice(["city", "suburb", "rural"]),
-            "race": random.choice(["Chinese", "American", "British", "French", "German", "Japanese", "Korean", "Russian", "Other"]),
-            "religion": random.choice(["none", "Christian", "Muslim", "Buddhist", "Hindu", "Other"]),
-            "marital_status": random.choice(["not married", "married", "divorced", "widowed"]),
-            }
+            "race": random.choice(
+                [
+                    "Chinese",
+                    "American",
+                    "British",
+                    "French",
+                    "German",
+                    "Japanese",
+                    "Korean",
+                    "Russian",
+                    "Other",
+                ]
+            ),
+            "religion": random.choice(
+                ["none", "Christian", "Muslim", "Buddhist", "Hindu", "Other"]
+            ),
+            "marital_status": random.choice(
+                ["not married", "married", "divorced", "widowed"]
+            ),
+        }
 
         BASE = {
-            "home": {"aoi_position": {"aoi_id": AOI_START_ID + random.randint(1, 50000)}},
-            "work": {"aoi_position": {"aoi_id": AOI_START_ID + random.randint(1, 50000)}},
+            "home": {
+                "aoi_position": {"aoi_id": AOI_START_ID + random.randint(1, 50000)}
+            },
+            "work": {
+                "aoi_position": {"aoi_id": AOI_START_ID + random.randint(1, 50000)}
+            },
         }
 
         return EXTRA_ATTRIBUTES, PROFILE, BASE
@@ -219,7 +273,7 @@ class AgentSimulation:
         except Exception as e:
             logger.error(f"采访过程出错: {str(e)}")
             return f"采访过程出现错误: {str(e)}"
-        
+
     async def submit_survey(self, agent_name: str, survey_id: str) -> str:
         """向智能体提交问卷
 
@@ -320,9 +374,7 @@ class AgentSimulation:
             prevent_thread_lock=True,
             quiet=True,
         )
-        logger.info(
-            f"Gradio Frontend is running on http://{server_name}:{server_port}"
-        )
+        logger.info(f"Gradio Frontend is running on http://{server_name}:{server_port}")
 
     async def step(self):
         """运行一步, 即每个智能体执行一次forward"""
@@ -334,7 +386,7 @@ class AgentSimulation:
         except Exception as e:
             logger.error(f"运行错误: {str(e)}")
             raise
-        
+
     async def run(
         self,
         day: int = 1,
@@ -349,7 +401,7 @@ class AgentSimulation:
             tasks = []
             for group in self._groups.values():
                 tasks.append(group.run.remote(day))
-            
+
             await asyncio.gather(*tasks)
 
         except Exception as e:

pycityagent/simulation/survey/__init__.py

@@ -1,9 +1,4 @@
 from .models import QuestionType, Question, Survey
 from .manager import SurveyManager
 
-__all__ = [
-    'QuestionType',
-    'Question', 
-    'Survey',
-    'SurveyManager'
-]
+__all__ = ["QuestionType", "Question", "Survey", "SurveyManager"]

pycityagent/simulation/survey/manager.py

@@ -4,14 +4,17 @@ import uuid
 import json
 from .models import Survey, Question, QuestionType
 
+
 class SurveyManager:
     def __init__(self):
         self._surveys: Dict[str, Survey] = {}
-        
-    def create_survey(self, title: str, description: str, questions: List[dict]) -> Survey:
+
+    def create_survey(
+        self, title: str, description: str, questions: List[dict]
+    ) -> Survey:
         """创建新问卷"""
         survey_id = str(uuid.uuid4())
-        
+
         # 转换问题数据
         survey_questions = []
         for q in questions:
@@ -21,47 +24,45 @@ class SurveyManager:
                 required=q.get("required", True),
                 options=q.get("options", []),
                 min_rating=q.get("min_rating", 1),
-                max_rating=q.get("max_rating", 5)
+                max_rating=q.get("max_rating", 5),
             )
             survey_questions.append(question)
-            
+
         survey = Survey(
             id=survey_id,
             title=title,
             description=description,
-            questions=survey_questions
+            questions=survey_questions,
         )
-        
+
         self._surveys[survey_id] = survey
         return survey
-        
+
     def get_survey(self, survey_id: str) -> Optional[Survey]:
         """获取指定问卷"""
         return self._surveys.get(survey_id)
-        
+
     def get_all_surveys(self) -> List[Survey]:
         """获取所有问卷"""
         return list(self._surveys.values())
-        
+
     def add_response(self, survey_id: str, agent_name: str, response: dict) -> bool:
         """添加问卷回答"""
         survey = self.get_survey(survey_id)
         if not survey:
             return False
-            
-        survey.responses[agent_name] = {
-            "timestamp": datetime.now(),
-            **response
-        }
+
+        survey.responses[agent_name] = {"timestamp": datetime.now(), **response}
         return True
-        
+
     def export_results(self, survey_id: str) -> str:
         """导出问卷结果"""
         survey = self.get_survey(survey_id)
         if not survey:
             return json.dumps({"error": "问卷不存在"})
-            
-        return json.dumps({
-            "survey": survey.to_dict(),
-            "responses": survey.responses
-        }, ensure_ascii=False, indent=2) 
+
+        return json.dumps(
+            {"survey": survey.to_dict(), "responses": survey.responses},
+            ensure_ascii=False,
+            indent=2,
+        )

pycityagent/simulation/survey/models.py

@@ -4,6 +4,7 @@ from datetime import datetime
 from enum import Enum
 import uuid
 
+
 class QuestionType(Enum):
     TEXT = "文本"
     SINGLE_CHOICE = "单选"
@@ -11,6 +12,7 @@ class QuestionType(Enum):
     RATING = "评分"
     LIKERT = "李克特量表"
 
+
 @dataclass
 class Question:
     content: str
@@ -19,7 +21,7 @@ class Question:
     options: List[str] = field(default_factory=list)
     min_rating: int = 1
     max_rating: int = 5
-    
+
     def to_dict(self) -> dict:
         return {
             "content": self.content,
@@ -27,9 +29,10 @@ class Question:
             "required": self.required,
             "options": self.options,
             "min_rating": self.min_rating,
-            "max_rating": self.max_rating
+            "max_rating": self.max_rating,
         }
 
+
 @dataclass
 class Survey:
     id: str
@@ -38,12 +41,12 @@ class Survey:
     questions: List[Question]
     responses: Dict[str, dict] = field(default_factory=dict)
     created_at: datetime = field(default_factory=datetime.now)
-    
+
     def to_dict(self) -> dict:
         return {
             "id": self.id,
             "title": self.title,
             "description": self.description,
             "questions": [q.to_dict() for q in self.questions],
-            "response_count": len(self.responses)
-        } 
+            "response_count": len(self.responses),
+        }

pycityagent/utils/decorators.py

@@ -2,18 +2,21 @@ import time
 import functools
 import inspect
 
-CALLING_STRING = "function: `{func_name}` in \"{file_path}\", line {line_number}, arguments: `{arguments}` start time: `{start_time}` end time: `{end_time}` output: `{output}`"
+CALLING_STRING = 'function: `{func_name}` in "{file_path}", line {line_number}, arguments: `{arguments}` start time: `{start_time}` end time: `{end_time}` output: `{output}`'
 
-__all__ =[
+__all__ = [
     "record_call_aio",
     "record_call",
     "lock_decorator",
 ]
+
+
 def record_call_aio(record_function_calling: bool = True):
     """
     Decorator to log the async function call details if `record_function_calling` is True.
     """
-    def decorator(func): 
+
+    def decorator(func):
         async def wrapper(*args, **kwargs):
             cur_frame = inspect.currentframe()
             assert cur_frame is not None
@@ -40,14 +43,18 @@ def record_call_aio(record_function_calling: bool = True):
                     )
                 )
             return result
+
         return wrapper
+
     return decorator
 
+
 def record_call(record_function_calling: bool = True):
     """
     Decorator to log the function call details if `record_function_calling` is True.
     """
-    def decorator(func):        
+
+    def decorator(func):
         def wrapper(*args, **kwargs):
             cur_frame = inspect.currentframe()
             assert cur_frame is not None
@@ -74,9 +81,12 @@ def record_call(record_function_calling: bool = True):
                     )
                 )
             return result
+
         return wrapper
+
     return decorator
 
+
 def lock_decorator(func):
     async def wrapper(self, *args, **kwargs):
         lock = self._lock

pycityagent/utils/parsers/__init__.py

@@ -1,6 +1,7 @@
 """Model response parser module."""
+
 from .parser_base import ParserBase
-from .json_parser import JsonDictParser,JsonObjectParser
+from .json_parser import JsonDictParser, JsonObjectParser
 from .code_block_parser import CodeBlockParser