pycityagent 2.0.0a49__cp311-cp311-macosx_11_0_arm64.whl → 2.0.0a51__cp311-cp311-macosx_11_0_arm64.whl

pycityagent/simulation/simulation.py

@@ -13,6 +13,11 @@ import yaml
 from langchain_core.embeddings import Embeddings
 
 from ..agent import Agent, InstitutionAgent
+from ..cityagent import (BankAgent, FirmAgent, GovernmentAgent, NBSAgent,
+                         SocietyAgent, memory_config_bank, memory_config_firm,
+                         memory_config_government, memory_config_nbs,
+                         memory_config_societyagent)
+from ..cityagent.initial import bind_agent_info, initialize_social_network
 from ..environment.simulator import Simulator
 from ..llm import SimpleEmbedding
 from ..memory import Memory
@@ -22,11 +27,10 @@ from ..survey import Survey
 from ..utils import TO_UPDATE_EXP_INFO_KEYS_AND_TYPES
 from .agentgroup import AgentGroup
 from .storage.pg import PgWriter, create_pg_tables
-from ..cityagent import SocietyAgent, FirmAgent, BankAgent, NBSAgent, GovernmentAgent, memory_config_societyagent, memory_config_government, memory_config_firm, memory_config_bank, memory_config_nbs
-from ..cityagent.initial import bind_agent_info, initialize_social_network
 
 logger = logging.getLogger("pycityagent")
 
+
 class AgentSimulation:
     """城市智能体模拟器"""
 
@@ -52,7 +56,13 @@ class AgentSimulation:
             self.agent_class = agent_class
         elif agent_class is None:
             if enable_economy:
-                self.agent_class = [SocietyAgent, FirmAgent, BankAgent, NBSAgent, GovernmentAgent]
+                self.agent_class = [
+                    SocietyAgent,
+                    FirmAgent,
+                    BankAgent,
+                    NBSAgent,
+                    GovernmentAgent,
+                ]
                 self.default_memory_config_func = [
                     memory_config_societyagent,
                     memory_config_firm,
@@ -82,7 +92,7 @@ class AgentSimulation:
         # self._last_asyncio_pg_task = None  # 将SQL写入的IO隐藏到计算任务后
 
         self._messager = Messager.remote(
-            hostname=config["simulator_request"]["mqtt"]["server"], # type:ignore
+            hostname=config["simulator_request"]["mqtt"]["server"],  # type:ignore
             port=config["simulator_request"]["mqtt"]["port"],
             username=config["simulator_request"]["mqtt"].get("username", None),
             password=config["simulator_request"]["mqtt"].get("password", None),
@@ -143,7 +153,7 @@ class AgentSimulation:
         """Directly run from config file
         Basic config file should contain:
         - simulation_config: file_path
-        - agent_config: 
+        - agent_config:
             - agent_config_file: Optional[dict]
             - memory_config_func: Optional[Union[Callable, list[Callable]]]
             - init_func: Optional[list[Callable[AgentSimulation, None]]]
@@ -173,13 +183,14 @@ class AgentSimulation:
         if "workflow" not in config:
             raise ValueError("workflow is required")
         import yaml
+
         logger.info("Loading config file...")
         with open(config["simulation_config"], "r") as f:
             simulation_config = yaml.safe_load(f)
         logger.info("Creating AgentSimulation Task...")
         simulation = cls(
-            config=simulation_config, 
-            agent_config_file=config["agent_config"].get("agent_config_file", None), 
+            config=simulation_config,
+            agent_config_file=config["agent_config"].get("agent_config_file", None),
             exp_name=config.get("exp_name", "default_experiment"),
             logging_level=config.get("logging_level", logging.WARNING),
         )
@@ -193,21 +204,28 @@ class AgentSimulation:
         await simulation.init_agents(
             agent_count=agent_count,
             group_size=config["agent_config"].get("group_size", 10000),
-            embedding_model=config["agent_config"].get("embedding_model", SimpleEmbedding()),
+            embedding_model=config["agent_config"].get(
+                "embedding_model", SimpleEmbedding()
+            ),
             memory_config_func=config["agent_config"].get("memory_config_func", None),
         )
         logger.info("Running Init Functions...")
-        for init_func in config["agent_config"].get("init_func", [bind_agent_info, initialize_social_network]):
+        for init_func in config["agent_config"].get(
+            "init_func", [bind_agent_info, initialize_social_network]
+        ):
             await init_func(simulation)
         logger.info("Starting Simulation...")
         for step in config["workflow"]:
-            logger.info(f"Running step: type: {step['type']} - description: {step.get('description', 'no description')}")
+            logger.info(
+                f"Running step: type: {step['type']} - description: {step.get('description', 'no description')}"
+            )
             if step["type"] not in ["run", "step", "interview", "survey", "intervene"]:
                 raise ValueError(f"Invalid step type: {step['type']}")
             if step["type"] == "run":
                 await simulation.run(step.get("day", 1))
             elif step["type"] == "step":
-                await simulation.step(step.get("time", 1))
+                # await simulation.step(step.get("time", 1))
+                await simulation.step()
             else:
                 await step["step_func"](simulation)
         logger.info("Simulation finished")
@@ -241,11 +259,11 @@ class AgentSimulation:
     @property
     def agent_uuid2group(self):
         return self._agent_uuid2group
-    
+
     @property
     def messager(self):
         return self._messager
-    
+
     async def _save_exp_info(self) -> None:
         """异步保存实验信息到YAML文件"""
         try:
@@ -354,38 +372,44 @@ class AgentSimulation:
         # 分别处理机构智能体和普通智能体
         institution_params = []
         citizen_params = []
-        
+
         # 收集所有参数
         for i in range(len(self.agent_class)):
             agent_class = self.agent_class[i]
             agent_count_i = agent_count[i]
             memory_config_func_i = memory_config_func[i]
-            
+
             if self.agent_config_file is not None:
-                config_file = self.agent_config_file.get(agent_class, None) 
+                config_file = self.agent_config_file.get(agent_class, None)
             else:
                 config_file = None
-                
+
             if issubclass(agent_class, InstitutionAgent):
-                institution_params.append((agent_class, agent_count_i, memory_config_func_i, config_file))
+                institution_params.append(
+                    (agent_class, agent_count_i, memory_config_func_i, config_file)
+                )
             else:
-                citizen_params.append((agent_class, agent_count_i, memory_config_func_i, config_file))
+                citizen_params.append(
+                    (agent_class, agent_count_i, memory_config_func_i, config_file)
+                )
 
         # 处理机构智能体组
         if institution_params:
             total_institution_count = sum(p[1] for p in institution_params)
-            num_institution_groups = (total_institution_count + group_size - 1) // group_size
-            
+            num_institution_groups = (
+                total_institution_count + group_size - 1
+            ) // group_size
+
             for k in range(num_institution_groups):
                 start_idx = k * group_size
                 remaining = total_institution_count - start_idx
                 number_of_agents = min(remaining, group_size)
-                
+
                 agent_classes = []
                 agent_counts = []
                 memory_config_funcs = []
                 config_files = []
-                
+
                 # 分配每种类型的机构智能体到当前组
                 curr_start = start_idx
                 for agent_class, count, mem_func, conf_file in institution_params:
@@ -395,30 +419,32 @@ class AgentSimulation:
                         memory_config_funcs.append(mem_func)
                         config_files.append(conf_file)
                     curr_start = max(0, curr_start - count)
-                
-                group_creation_params.append((
-                    agent_classes,
-                    agent_counts, 
-                    memory_config_funcs,
-                    f"InstitutionGroup_{k}",
-                    config_files
-                ))
+
+                group_creation_params.append(
+                    (
+                        agent_classes,
+                        agent_counts,
+                        memory_config_funcs,
+                        f"InstitutionGroup_{k}",
+                        config_files,
+                    )
+                )
 
         # 处理普通智能体组
         if citizen_params:
             total_citizen_count = sum(p[1] for p in citizen_params)
             num_citizen_groups = (total_citizen_count + group_size - 1) // group_size
-            
+
             for k in range(num_citizen_groups):
                 start_idx = k * group_size
                 remaining = total_citizen_count - start_idx
                 number_of_agents = min(remaining, group_size)
-                
+
                 agent_classes = []
                 agent_counts = []
                 memory_config_funcs = []
                 config_files = []
-                
+
                 # 分配每种类型的普通智能体到当前组
                 curr_start = start_idx
                 for agent_class, count, mem_func, conf_file in citizen_params:
@@ -428,14 +454,16 @@ class AgentSimulation:
                         memory_config_funcs.append(mem_func)
                         config_files.append(conf_file)
                     curr_start = max(0, curr_start - count)
-                
-                group_creation_params.append((
-                    agent_classes,
-                    agent_counts,
-                    memory_config_funcs, 
-                    f"CitizenGroup_{k}",
-                    config_files
-                ))
+
+                group_creation_params.append(
+                    (
+                        agent_classes,
+                        agent_counts,
+                        memory_config_funcs,
+                        f"CitizenGroup_{k}",
+                        config_files,
+                    )
+                )
 
         # 初始化mlflow连接
         _mlflow_config = self.config.get("metric_request", {}).get("mlflow")
@@ -463,7 +491,13 @@ class AgentSimulation:
             self._pgsql_writers = _workers = [None for _ in range(_num_workers)]
 
         creation_tasks = []
-        for i, (agent_class, number_of_agents, memory_config_function_group, group_name, config_file) in enumerate(group_creation_params):
+        for i, (
+            agent_class,
+            number_of_agents,
+            memory_config_function_group,
+            group_name,
+            config_file,
+        ) in enumerate(group_creation_params):
             # 直接创建异步任务
             group = AgentGroup.remote(
                 agent_class,
@@ -489,7 +523,9 @@ class AgentSimulation:
             group_agent_uuids = ray.get(group.get_agent_uuids.remote())
             for agent_uuid in group_agent_uuids:
                 self._agent_uuid2group[agent_uuid] = group
-                self._user_chat_topics[agent_uuid] = f"exps/{self.exp_id}/agents/{agent_uuid}/user-chat"
+                self._user_chat_topics[agent_uuid] = (
+                    f"exps/{self.exp_id}/agents/{agent_uuid}/user-chat"
+                )
                 self._user_survey_topics[agent_uuid] = (
                     f"exps/{self.exp_id}/agents/{agent_uuid}/user-survey"
                 )
@@ -511,23 +547,26 @@ class AgentSimulation:
         for group in self._groups.values():
             gather_tasks.append(group.gather.remote(content))
         return await asyncio.gather(*gather_tasks)
-    
-    async def filter(self, 
-                     types: Optional[list[Type[Agent]]] = None, 
-                     keys: Optional[list[str]] = None, 
-                     values: Optional[list[Any]] = None) -> list[str]:
+
+    async def filter(
+        self,
+        types: Optional[list[Type[Agent]]] = None,
+        keys: Optional[list[str]] = None,
+        values: Optional[list[Any]] = None,
+    ) -> list[str]:
         """过滤出指定类型的智能体"""
         if not types and not keys and not values:
             return self._agent_uuids
         group_to_filter = []
-        for t in types:
-            if t in self._type2group:
-                group_to_filter.extend(self._type2group[t])
-            else:
-                raise ValueError(f"type {t} not found in simulation")
+        if types is not None:
+            for t in types:
+                if t in self._type2group:
+                    group_to_filter.extend(self._type2group[t])
+                else:
+                    raise ValueError(f"type {t} not found in simulation")
         filtered_uuids = []
         if keys:
-            if len(keys) != len(values):
+            if values is None or len(keys) != len(values):
                 raise ValueError("the length of key and value does not match")
             for group in group_to_filter:
                 filtered_uuids.extend(await group.filter.remote(types, keys, values))

pycityagent/tools/__init__.py

@@ -0,0 +1,11 @@
+from .tool import (ExportMlflowMetrics, GetMap, ResetAgentPosition, SencePOI,
+                   Tool, UpdateWithSimulator)
+
+__all__ = [
+    "SencePOI",
+    "Tool",
+    "ExportMlflowMetrics",
+    "GetMap",
+    "UpdateWithSimulator",
+    "ResetAgentPosition",
+]

pycityagent/{workflow → tools}/tool.py

@@ -7,9 +7,9 @@ from typing import Any, Optional, Union
 from mlflow.entities import Metric
 
 from ..agent import Agent
-from ..utils.decorators import lock_decorator
 from ..environment import (LEVEL_ONE_PRE, POI_TYPE_DICT, AoiService,
                            PersonService)
+from ..utils.decorators import lock_decorator
 from ..workflow import Block
 
 
@@ -199,6 +199,7 @@ class ExportMlflowMetrics(Tool):
         # TODO: support other log types
         self.metric_log_cache: dict[str, list[Metric]] = defaultdict(list)
         self._lock = asyncio.Lock()
+
     @lock_decorator
     async def __call__(
         self,
@@ -231,6 +232,7 @@ class ExportMlflowMetrics(Tool):
                 _cache = _cache[batch_size:]
         if clear_cache:
             await self._clear_cache()
+
     @lock_decorator
     async def _clear_cache(
         self,

pycityagent/workflow/__init__.py

@@ -7,14 +7,9 @@ This module contains classes for creating blocks and running workflows.
 from .block import (Block, log_and_check, log_and_check_with_memory,
                     trigger_class)
 from .prompt import FormatPrompt
-from .tool import ExportMlflowMetrics, GetMap, SencePOI, Tool
 from .trigger import EventTrigger, MemoryChangeTrigger, TimeTrigger
 
 __all__ = [
-    "SencePOI",
-    "Tool",
-    "ExportMlflowMetrics",
-    "GetMap",
     "MemoryChangeTrigger",
     "TimeTrigger",
     "EventTrigger",

pycityagent/workflow/block.py

@@ -1,10 +1,11 @@
 from __future__ import annotations
+
 import asyncio
 import functools
 import inspect
-from collections.abc import Awaitable, Callable, Coroutine
 import json
-from typing import Any, List, Optional, Union
+from collections.abc import Awaitable, Callable, Coroutine
+from typing import Any, Optional, Union
 
 from pyparsing import Dict
 
@@ -143,7 +144,7 @@ def trigger_class():
 
 # Define a Block, similar to a layer in PyTorch
 class Block:
-    configurable_fields: List[str] = []
+    configurable_fields: list[str] = []
     default_values: dict[str, Any] = {}
 
     def __init__(
@@ -164,22 +165,23 @@ class Block:
             trigger.initialize()  # 立即初始化trigger
         self.trigger = trigger
 
-    def export_config(self) -> Dict[str, Optional[str]]:
+    def export_config(self) -> dict[str, Optional[str]]:
         return {
             field: self.default_values.get(field, "default_value")
             for field in self.configurable_fields
         }
 
     @classmethod
-    def export_class_config(cls) -> Dict[str, str]:
+    def export_class_config(cls) -> dict[str, str]:
         return {
             field: cls.default_values.get(field, "default_value")
             for field in cls.configurable_fields
         }
 
     @classmethod
-    def import_config(cls, config: Dict[str, str]) -> "Block":
+    def import_config(cls, config: dict[str, Union[str, dict]]) -> Block:
         instance = cls(name=config["name"])
+        assert isinstance(config["config"], dict)
         for field, value in config["config"].items():
             if field in cls.configurable_fields:
                 setattr(instance, field, value)
@@ -190,8 +192,8 @@ class Block:
             setattr(instance, child_block.name.lower(), child_block)
 
         return instance
-    
-    def load_from_config(self, config: Dict[str, List[Dict]]) -> None:
+
+    def load_from_config(self, config: dict[str, list[Dict]]) -> None:
         """
         使用配置更新当前Block实例的参数，并递归更新子Block。
         """
@@ -201,8 +203,8 @@ class Block:
                 if config["config"][field] != "default_value":
                     setattr(self, field, config["config"][field])
 
-        def build_or_update_block(block_data: Dict) -> Block:
-            block_name = block_data["name"].lower()
+        def build_or_update_block(block_data: dict) -> Block:
+            block_name = block_data["name"].lower()  # type:ignore
             existing_block = getattr(self, block_name, None)
 
             if existing_block:

{pycityagent-2.0.0a49.dist-info → pycityagent-2.0.0a51.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycityagent
-Version: 2.0.0a49
+Version: 2.0.0a51
 Summary: LLM-based city environment agent building library
 Author-email: Yuwei Yan <pinkgranite86@gmail.com>, Junbo Yan <yanjb20thu@gmali.com>, Jun Zhang <zhangjun990222@gmali.com>
 License: MIT License
@@ -50,7 +50,6 @@ Requires-Dist: requests>=2.32.3
 Requires-Dist: Shapely>=2.0.6
 Requires-Dist: PyYAML>=6.0.2
 Requires-Dist: zhipuai>=2.1.5.20230904
-Requires-Dist: gradio>=5.7.1
 Requires-Dist: mosstool>=1.3.0
 Requires-Dist: ray>=2.40.0
 Requires-Dist: aiomqtt>=2.3.0

{pycityagent-2.0.0a49.dist-info → pycityagent-2.0.0a51.dist-info}/RECORD

@@ -1,12 +1,13 @@
 pycityagent/pycityagent-sim,sha256=n96jlVZRINlBec5SPOGAdUmeLWMoEKGgoH29iOVJ0wE,34081890
-pycityagent/__init__.py,sha256=9sHyzo8monlBPDe7LTlxCIHeNPYj8HJJDCeuJrYJLsQ,750
+pycityagent/__init__.py,sha256=PUKWTXc-xdMG7px8oTNclodsILUgypANj2Z647sY63k,808
 pycityagent/pycityagent-ui,sha256=cHZjqtrQ4Fh4qtRahFNCNbT2DNHLmUexiDAa-72Z3RQ,40333378
-pycityagent/agent.py,sha256=iaF83y344V29-jghx5bJD3yY8kMuSzInBPNi7cVxYiA,34026
 pycityagent/metrics/mlflow_client.py,sha256=g_tHxWkWTDijtbGL74-HmiYzWVKb1y8-w12QrY9jL30,4449
 pycityagent/metrics/__init__.py,sha256=X08PaBbGVAd7_PRGLREXWxaqm7nS82WBQpD1zvQzcqc,128
 pycityagent/metrics/utils/const.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycityagent/economy/__init__.py,sha256=aonY4WHnx-6EGJ4WKrx4S-2jAkYNLtqUA04jp6q8B7w,75
 pycityagent/economy/econ_client.py,sha256=GuHK9ZBnhqW3Z7F8ViDJn_iN73yOBbbwFyJv1wLEBDk,12211
+pycityagent/tools/__init__.py,sha256=XtdtGyWeFyK1YOUvWkykBWxemtmwQjWUIuuyU1-gosQ,261
+pycityagent/tools/tool.py,sha256=JD5APkoWqnS46gyimI17no2fPknjwkk173AFt-oFv_c,8778
 pycityagent/llm/llmconfig.py,sha256=4Ylf4OFSBEFy8jrOneeX0HvPhWEaF5jGvy1HkXK08Ro,436
 pycityagent/llm/__init__.py,sha256=iWs6FLgrbRVIiqOf4ILS89gkVCTvS7HFC3vG-MWuyko,205
 pycityagent/llm/llm.py,sha256=owTYuXmnHZnvXaAvwiiyD511P3wpU3K04xZArhhiJF0,15700
@@ -21,9 +22,9 @@ pycityagent/memory/utils.py,sha256=oJWLdPeJy_jcdKcDTo9JAH9kDZhqjoQhhv_zT9qWC0w,8
 pycityagent/memory/const.py,sha256=6zpJPJXWoH9-yf4RARYYff586agCoud9BRn7sPERB1g,932
 pycityagent/memory/faiss_query.py,sha256=Z0JS4udyPYCIzHMq464QtHscnswu35gh9fQptikAwkQ,12976
 pycityagent/memory/state.py,sha256=TYItiyDtehMEQaSBN7PpNrnNxdDM5jGppr9R9Ufv3kA,5134
-pycityagent/simulation/simulation.py,sha256=7Y1r5DAFaB1vuDd3l_85NWpHQUy2GLkFEb800xsJr0E,25565
+pycityagent/simulation/simulation.py,sha256=DkTYkdNknz2d7hIdYhxxLewUkYwmS9uOxnmEGTwbL_Q,26098
 pycityagent/simulation/__init__.py,sha256=P5czbcg2d8S0nbbnsQXFIhwzO4CennAhZM8OmKvAeYw,194
-pycityagent/simulation/agentgroup.py,sha256=DGN6q9S65K9gSv_ihBeT6iJ__LJhl27yXBFyLEk55t0,29302
+pycityagent/simulation/agentgroup.py,sha256=fMTN9xDC-PdTGZ4KCMovZeGjzv8HIQJ1AGVG7iAJnD4,29351
 pycityagent/simulation/storage/pg.py,sha256=5itxKOkNPlOzN7z2_3oKU1ZK0uLTDugfld8ZkRbD69I,8407
 pycityagent/message/__init__.py,sha256=TCjazxqb5DVwbTu1fF0sNvaH_EPXVuj2XQ0p6W-QCLU,55
 pycityagent/message/messager.py,sha256=78K31EPKfC5IxbISc-Lc2babC7VOh9Vbe3c0sO-YDLA,3333
@@ -36,11 +37,13 @@ pycityagent/utils/parsers/__init__.py,sha256=AN2xgiPxszWK4rpX7zrqRsqNwfGF3WnCA5-
 pycityagent/utils/parsers/code_block_parser.py,sha256=Cs2Z_hm9VfNCpPPll1TwteaJF-HAQPs-3RApsOekFm4,1173
 pycityagent/utils/parsers/parser_base.py,sha256=KBKO4zLZPNdGjPAGqIus8LseZ8W3Tlt2y0QxqeCd25Q,1713
 pycityagent/utils/parsers/json_parser.py,sha256=tjwyPluYfkWgsvLi0hzfJwFhO3L6yQfZMKza20HaGrY,2911
+pycityagent/agent/agent_base.py,sha256=78J6np8XrFXTUXTC9x_O28EUmyWsdgtARd1KfNcC0cI,23073
+pycityagent/agent/__init__.py,sha256=U20yKu9QwSqAx_PHk5JwipfODkDfxONtumVfnsKjWFg,180
+pycityagent/agent/agent.py,sha256=8biuC44Yx4T8Fntlj_RGTlnKWwS0XDNkfYQrv5xXUwE,11548
 pycityagent/cli/wrapper.py,sha256=2Tb52gOlEVgn11Ekt6ZkRXr_BGzte-EPyBKnR6g6hQ4,1143
-pycityagent/workflow/__init__.py,sha256=QNkUV-9mACMrR8c0cSKna2gC1mMZdxXbxWzjE-Uods0,621
+pycityagent/workflow/__init__.py,sha256=H08Ko3eliZvuuCMajbEri-IP4-SeswYU6UjHBNA4Ze0,490
 pycityagent/workflow/prompt.py,sha256=6jI0Rq54JLv3-IXqZLYug62vse10wTI83xvf4ZX42nk,2929
-pycityagent/workflow/block.py,sha256=8pK17DGX8MnFJHq5lJ4susdZ-CDAEIjlTV_2mJxhbbw,9600
-pycityagent/workflow/tool.py,sha256=ChuN5Sm-yltesemewjnTOYaxN6NUjfi2MezVuclf6Tc,8776
+pycityagent/workflow/block.py,sha256=3X6qzpYmmMyZyXA6Dxl4lS-mNR0eLzLoiRhHY-H1rzE,9667
 pycityagent/workflow/trigger.py,sha256=Df-MOBEDWBbM-v0dFLQLXteLsipymT4n8vqexmK2GiQ,5643
 pycityagent/environment/__init__.py,sha256=awHxlOud-btWbk0FCS4RmGJ13W84oVCkbGfcrhKqihA,240
 pycityagent/environment/simulator.py,sha256=1OUfODDzM4EN6Lw_Wzq4KeQb-EpcUBioZYc9fxfSPn0,12070
@@ -70,14 +73,14 @@ pycityagent/environment/sim/clock_service.py,sha256=fgYXacf_-ixhVAn5uKUvqvemBS6A
 pycityagent/environment/sim/road_service.py,sha256=bKyn3_me0sGmaJVyF6eNeFbdU-9C1yWsa9L7pieDJzg,1285
 pycityagent/environment/interact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycityagent/environment/interact/interact.py,sha256=ifxPPzuHeqLHIZ_6zvfXMoBOnBsXNIP4bYp7OJ7pnEQ,6588
-pycityagent/cityagent/memory_config.py,sha256=tI4SbzDdjVqIt5cTqx-CwNHu0xnbVAhIenZgwYa9wf8,10429
-pycityagent/cityagent/bankagent.py,sha256=IC4Qf-zgHg9Q65ecb_UAKjw-HbizYEDuzfhlPIske9Y,2166
+pycityagent/cityagent/memory_config.py,sha256=6dkgqPczANUqZpqGXDOt7wrS6vHdFSMNkWvEVwo6VIQ,10672
+pycityagent/cityagent/bankagent.py,sha256=9J0c_r6m0OJH2KpXxeCWstLqZVpjQ67xuWmDTbVm7KM,2217
 pycityagent/cityagent/__init__.py,sha256=gcBQ-a50XegFtjigQ7xDXRBZrywBKqifiQFSRnEF8gM,572
-pycityagent/cityagent/firmagent.py,sha256=cV2X1wtj6vbXEGNcFlhb9Nsp544coe2lyEjLwOmLOPE,3635
-pycityagent/cityagent/nbsagent.py,sha256=VFxz0FsjPN9hAir-VxXYVE8WR4zsEXF7oRn0G8SYNfo,4647
-pycityagent/cityagent/initial.py,sha256=_sKZicZ0wfuBN_hLOW82_Yp3wOMMlygEvcGT0sYXE6c,4538
-pycityagent/cityagent/societyagent.py,sha256=-O_lk2xk9-ldE9MTX48tqbRyJFIqcpYnmfMsrGUvRvw,13958
-pycityagent/cityagent/governmentagent.py,sha256=WdM-80YkRohYuNj1hB07ZZ_dgRN_uiVU8YkoS7Cs6vg,2713
+pycityagent/cityagent/firmagent.py,sha256=Mtb58YfW3V696gXA6TRw1qUL1WIzthahDBv2X5x_Iyg,4176
+pycityagent/cityagent/nbsagent.py,sha256=SIOigR8eiSHC0Gi8kg32hLxsqPKDgCHWF0HA89EU7eA,5202
+pycityagent/cityagent/initial.py,sha256=7hgCt_tGdnVTXGfEQOn1GTW5dAs1b-ru_FwXxRLI6tM,4549
+pycityagent/cityagent/societyagent.py,sha256=oVDWt-DDGMDxh_LpqMs4_CX-5hVQORyPC1ZW2a8CrhI,14960
+pycityagent/cityagent/governmentagent.py,sha256=9fjGueVmYok7zca14E1VL4N3ENEhnq2nubI6ab7Hzp0,2825
 pycityagent/cityagent/blocks/dispatcher.py,sha256=mEa1r3tRS3KI1BMZR_w_sbUGzOj6aUJuiUrsHv1n2n0,2943
 pycityagent/cityagent/blocks/needs_block.py,sha256=pPPjdVFCqFvAV0RM1LVevQIH5d0GALnB6xLekvplutE,13113
 pycityagent/cityagent/blocks/cognition_block.py,sha256=CGIYwuF1twS5g3DNhCkaSFT2s1IQXF5NjJO38EX6FLg,15633
@@ -92,9 +95,9 @@ pycityagent/cityagent/blocks/mobility_block.py,sha256=ZebloDhvZJYbV0YSWhAbnOLToi
 pycityagent/survey/models.py,sha256=YE50UUt5qJ0O_lIUsSY6XFCGUTkJVNu_L1gAhaCJ2fs,3546
 pycityagent/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
 pycityagent/survey/manager.py,sha256=S5IkwTdelsdtZETChRcfCEczzwSrry_Fly9MY4s3rbk,1681
-pycityagent-2.0.0a49.dist-info/RECORD,,
-pycityagent-2.0.0a49.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
-pycityagent-2.0.0a49.dist-info/WHEEL,sha256=DQkJubmXvyOE1DlaPsfq9czNuGB2yvzYGbMABjt_JVM,109
-pycityagent-2.0.0a49.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
-pycityagent-2.0.0a49.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
-pycityagent-2.0.0a49.dist-info/METADATA,sha256=AArJCX11voHGW5AlMrin1qZoj8iVGdOD9XLODRcQwUY,9139
+pycityagent-2.0.0a51.dist-info/RECORD,,
+pycityagent-2.0.0a51.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
+pycityagent-2.0.0a51.dist-info/WHEEL,sha256=DQkJubmXvyOE1DlaPsfq9czNuGB2yvzYGbMABjt_JVM,109
+pycityagent-2.0.0a51.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
+pycityagent-2.0.0a51.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
+pycityagent-2.0.0a51.dist-info/METADATA,sha256=zmYO5Z37qS-ENVQomAcXnJA_8TD1RDSe6IGkmYHq3MM,9110