pycityagent 2.0.0a95__cp39-cp39-macosx_11_0_arm64.whl → 2.0.0a97__cp39-cp39-macosx_11_0_arm64.whl

pycityagent/agent/agent_base.py

@@ -260,20 +260,16 @@ class Agent(ABC):
             - Recursively updates or creates top-level Blocks as specified in the configuration.
             - Raises a `KeyError` if a required Block is not found in the agent.
         """
-        # 使用配置更新当前Agent实例的Block层次结构。
-        # 更新当前Agent的基础参数
         for field in self.configurable_fields:
             if field in config["config"]:
                 if config["config"][field] != "default_value":
                     setattr(self, field, config["config"][field])
 
-        # 递归更新或创建顶层Block
         for block_data in config.get("blocks", []):
             block_name = block_data["name"]
             existing_block = getattr(self, block_name, None)  # type:ignore
 
             if existing_block:
-                # 如果Block已经存在，则递归更新
                 existing_block.load_from_config(block_data)
             else:
                 raise KeyError(

pycityagent/cityagent/blocks/mobility_block.py

@@ -13,9 +13,10 @@ from pycityagent.llm import LLM
 from pycityagent.memory import Memory
 from pycityagent.workflow.block import Block
 from pycityagent.workflow.prompt import FormatPrompt
-
+from .utils import clean_json_response
 from .dispatcher import BlockDispatcher
 
+
 logger = logging.getLogger("pycityagent")
 
 
@@ -166,10 +167,11 @@ class PlaceSelectionBlock(Block):
         )
         levelOneType = await self.llm.atext_request(self.typeSelectionPrompt.to_dialog(), response_format={"type": "json_object"})  # type: ignore
         try:
+            levelOneType = clean_json_response(levelOneType)
             levelOneType = json.loads(levelOneType)["place_type"]
             sub_category = poi_cate[levelOneType]
         except Exception as e:
-            logger.warning(f"Wrong type of poi, raw response: {levelOneType}")
+            logger.warning(f"Level One Type Selection: wrong type of poi, raw response: {levelOneType}")
             levelOneType = random.choice(list(poi_cate.keys()))
             sub_category = poi_cate[levelOneType]
         self.secondTypeSelectionPrompt.format(
@@ -177,9 +179,10 @@ class PlaceSelectionBlock(Block):
         )
         levelTwoType = await self.llm.atext_request(self.secondTypeSelectionPrompt.to_dialog(), response_format={"type": "json_object"})  # type: ignore
         try:
+            levelTwoType = clean_json_response(levelTwoType)
             levelTwoType = json.loads(levelTwoType)["place_type"]
         except Exception as e:
-            logger.warning(f"Wrong type of poi, raw response: {levelTwoType}")
+            logger.warning(f"Level Two Type Selection: wrong type of poi, raw response: {levelTwoType}")
             levelTwoType = random.choice(sub_category)
         center = await self.memory.status.get("position")
         center = (center["xy_position"]["x"], center["xy_position"]["y"])
@@ -260,19 +263,18 @@ class MoveBlock(Block):
         self.placeAnalysisPrompt = FormatPrompt(PLACE_ANALYSIS_PROMPT)
 
     async def forward(self, step, context):
-        # 这里应该添加移动的具体逻辑
         agent_id = await self.memory.status.get("id")
         self.placeAnalysisPrompt.format(
             plan=context["plan"], intention=step["intention"]
         )
         response = await self.llm.atext_request(self.placeAnalysisPrompt.to_dialog(), response_format={"type": "json_object"})  # type: ignore
         try:
+            response = clean_json_response(response)
             response = json.loads(response)["place_type"]
         except Exception as e:
-            logger.warning(f"Wrong type of poi, raw response: {response}")
+            logger.warning(f"Place Analysis: wrong type of place, raw response: {response}")
             response = "home"
         if response == "home":
-            # 返回到家
             home = await self.memory.status.get("home")
             home = home["aoi_position"]["aoi_id"]
             nowPlace = await self.memory.status.get("position")

pycityagent/cityagent/blocks/needs_block.py

@@ -8,6 +8,7 @@ from pycityagent.llm import LLM
 from pycityagent.memory import Memory
 from pycityagent.workflow.block import Block
 from pycityagent.workflow.prompt import FormatPrompt
+from .utils import clean_json_response
 
 logger = logging.getLogger("pycityagent")
 
@@ -126,7 +127,7 @@ class NeedsBlock(Block):
                 now_time=await self.simulator.get_time(format_time=True),
             )
             response = await self.llm.atext_request(self.initial_prompt.to_dialog(), response_format={"type": "json_object"})
-            response = self.clean_json_response(response)
+            response = clean_json_response(response)
             retry = 3
             while retry > 0:
                 try:
@@ -309,7 +310,7 @@ class NeedsBlock(Block):
         while retry > 0:
             response = await self.llm.atext_request(self.evaluation_prompt.to_dialog(), response_format={"type": "json_object"})
             try:
-                new_satisfaction = json.loads(self.clean_json_response(response))  # type: ignore
+                new_satisfaction = json.loads(clean_json_response(response))  # type: ignore
                 # 更新所有需求的数值
                 for need_type, new_value in new_satisfaction.items():
                     if need_type in [
@@ -330,11 +331,6 @@ class NeedsBlock(Block):
                 logger.warning(f"Original response: {response}")
                 retry -= 1
 
-    def clean_json_response(self, response: str) -> str:
-        """清理LLM响应中的特殊字符"""
-        response = response.replace("```json", "").replace("```", "")
-        return response.strip()
-
     async def forward(self):
         await self.initialize()
 

pycityagent/cityagent/blocks/plan_block.py

@@ -10,7 +10,7 @@ from pycityagent.llm import LLM
 from pycityagent.memory import Memory
 from pycityagent.workflow import Block
 from pycityagent.workflow.prompt import FormatPrompt
-
+from .utils import clean_json_response
 logger = logging.getLogger("pycityagent")
 
 GUIDANCE_SELECTION_PROMPT = """As an intelligent agent's decision system, please select the most suitable option from the following choices to satisfy the current need.
@@ -203,7 +203,7 @@ class PlanBlock(Block):
         retry = 3
         while retry > 0:
             try:
-                result = json.loads(self.clean_json_response(response))  # type: ignore
+                result = json.loads(clean_json_response(response))  # type: ignore
                 if "selected_option" not in result or "evaluation" not in result:
                     raise ValueError("Invalid guidance selection format")
                 if (
@@ -255,7 +255,7 @@ class PlanBlock(Block):
         retry = 3
         while retry > 0:
             try:
-                result = json.loads(self.clean_json_response(response))  # type: ignore
+                result = json.loads(clean_json_response(response))  # type: ignore
                 if (
                     "plan" not in result
                     or "target" not in result["plan"]
@@ -316,8 +316,3 @@ Execution Steps: \n{formated_steps}
             "current_step", steps[0] if steps else {"intention": "", "type": ""}
         )
         await self.memory.status.update("execution_context", {"plan": formated_plan})
-
-    def clean_json_response(self, response: str) -> str:
-        """Clean special characters in LLM response"""
-        response = response.replace("```json", "").replace("```", "")
-        return response.strip()

pycityagent/cityagent/blocks/utils.py

@@ -57,6 +57,6 @@ def extract_dict_from_string(input_string):
 
 
 def clean_json_response(response: str) -> str:
-    """清理LLM响应中的特殊字符"""
+    """remove the special characters in the response"""
     response = response.replace("```json", "").replace("```", "")
-    return response.strip()
+    return response.strip()

pycityagent/configs/exp_config.py

@@ -29,7 +29,7 @@ class AgentConfig(BaseModel):
     number_of_nbs: int = Field(1, description="Number of neighborhood-based services")
     group_size: int = Field(100, description="Size of agent groups")
     embedding_model: Any = Field(None, description="Embedding model")
-    agent_class_configs: Optional[dict[Any, dict[str, list[dict]]]] = None
+    agent_class_configs: Optional[dict[Any, dict[str, Any]]] = None
     memory_config_func: Optional[dict[type["Any"], Callable]] = None
     memory_config_init_func: Optional[Callable] = Field(None)
     init_func: Optional[list[Callable[["AgentSimulation"], None]]] = None
@@ -47,7 +47,7 @@ class AgentConfig(BaseModel):
         number_of_nbs: int = 1,
         group_size: int = 100,
         embedding_model: Any = None,
-        agent_class_configs: Optional[dict[Any, dict[str, list[dict]]]] = None,
+        agent_class_configs: Optional[dict[Any, dict[str, Any]]] = None,
         enable_institution: bool = True,
         memory_config_func: Optional[dict[type["Any"], Callable]] = None,
         memory_config_init_func: Optional[Callable] = None,

pycityagent/simulation/agentgroup.py

@@ -192,7 +192,7 @@ class AgentGroup:
                     self.agent_config_file is not None
                     and self.agent_config_file[agent_class_i]
                 ):
-                    agent.load_from_file(self.agent_config_file[agent_class_i])
+                    agent.load_from_config(self.agent_config_file[agent_class_i])
                 if self._message_interceptor is not None:
                     agent.set_message_interceptor(self._message_interceptor)
                 self.agents.append(agent)

pycityagent/simulation/simulation.py

@@ -374,7 +374,7 @@ class AgentSimulation:
                 await simulation.pause_simulator()
             elif step.type == WorkflowType.RESUME:
                 await simulation.resume_simulator()
-            elif step.type == WorkflowType.FUNCTION:
+            else:
                 _func = cast(Callable, step.func)
                 await _func(simulation)
         logger.info("Simulation finished")

{pycityagent-2.0.0a95.dist-info → pycityagent-2.0.0a97.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pycityagent
-Version: 2.0.0a95
+Version: 2.0.0a97
 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

{pycityagent-2.0.0a95.dist-info → pycityagent-2.0.0a97.dist-info}/RECORD

@@ -1,3 +1,9 @@
+pycityagent-2.0.0a97.dist-info/RECORD,,
+pycityagent-2.0.0a97.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
+pycityagent-2.0.0a97.dist-info/WHEEL,sha256=md3JO_ifs5j508p3TDNMgtQVtnQblpGEt_Wo4W56l8Y,107
+pycityagent-2.0.0a97.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
+pycityagent-2.0.0a97.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
+pycityagent-2.0.0a97.dist-info/METADATA,sha256=M8Yf50T1rDcPjB8_SoCuFSONew2E-mMI-D-Y2aomCbo,8657
 pycityagent/pycityagent-sim,sha256=_xsIrY1raLYagebaymahAZc2YSustRmQM2jm1B4p27U,36972738
 pycityagent/__init__.py,sha256=PUKWTXc-xdMG7px8oTNclodsILUgypANj2Z647sY63k,808
 pycityagent/pycityagent-ui,sha256=Ur95yZygIaZ5l_CDqP9394M5GQ66iV5PkcNPYFWqzvk,41225346
@@ -18,9 +24,9 @@ pycityagent/memory/utils.py,sha256=oJWLdPeJy_jcdKcDTo9JAH9kDZhqjoQhhv_zT9qWC0w,8
 pycityagent/memory/const.py,sha256=Wgsx0oztDrJDEpI9Dr6XeI6GXrABcP0UTOuEvMp1AD8,934
 pycityagent/memory/faiss_query.py,sha256=KPeyzIjD0dzkxr-TlOeCiIlkdh1WAyyipCAXUEt97Lk,17350
 pycityagent/memory/state.py,sha256=JFCBuK1AS-HqscvdGS9ATF9AUU8t29_2leDY_6iO2_4,5158
-pycityagent/simulation/simulation.py,sha256=GrD5VWaEvrKrBFwhEKUUfQfk2rweTEZnROaBZJXSni0,47134
+pycityagent/simulation/simulation.py,sha256=YPRzB7wCL-K85W3hgvBWukefyEgZXZCi4egH6rZa6XA,47099
 pycityagent/simulation/__init__.py,sha256=u1WpgwVxPboXWMxrUQKMXJNmTKQYAeCaqZv9HmSxESY,118
-pycityagent/simulation/agentgroup.py,sha256=JFDEywmFziq-I8aUDZ3QhVD8cXliVsUSav482BmPDO4,36812
+pycityagent/simulation/agentgroup.py,sha256=DYx68TckA5ox9x_GHID-3Gg19qABY-NR9QAO2BKKfMc,36814
 pycityagent/simulation/storage/pg.py,sha256=xRshSOGttW-p0re0fNBOjOpb-nQ5msIE2LsdT79_E_Y,8425
 pycityagent/message/message_interceptor.py,sha256=Z7I1vqs4BrKadPAtQi-nfbkgboBAflXDSIcbIpDAxB8,17777
 pycityagent/message/__init__.py,sha256=f5QH7DKPqEAMyfSlBMnl3uouOKlsoel909STlIe7nUk,276
@@ -35,7 +41,7 @@ pycityagent/utils/parsers/__init__.py,sha256=AN2xgiPxszWK4rpX7zrqRsqNwfGF3WnCA5-
 pycityagent/utils/parsers/code_block_parser.py,sha256=jy6wHx9zypvJTWneCKuUaLQL19d4yLK6C_awc7y8A-k,1179
 pycityagent/utils/parsers/parser_base.py,sha256=ZbqRcqOBL9hfHVZSltQSQmJnWrtuJS9aYeOJU-FlJ0E,1725
 pycityagent/utils/parsers/json_parser.py,sha256=sakKohUuMcUH2toR85CAG_37D02ZoAn6LlzLgftUjmo,2923
-pycityagent/agent/agent_base.py,sha256=MM_kBWheTk9Y3pKoDo6Yl1Bvn2usT6v2icloYx_ykCM,39467
+pycityagent/agent/agent_base.py,sha256=Fmp9wYwiDskEQxslweO3Jc-1OwbB9vgjv1VfZZIHSGI,39252
 pycityagent/agent/__init__.py,sha256=U20yKu9QwSqAx_PHk5JwipfODkDfxONtumVfnsKjWFg,180
 pycityagent/agent/agent.py,sha256=zJs09Tzz53av59SzmrlJj6Yd-XId6voCYaxguLbbq3s,18991
 pycityagent/cli/wrapper.py,sha256=bAwmfWTXbGld-uiMnIMhKp22IVRQjhLUToknyUDk9rQ,1147
@@ -43,7 +49,7 @@ pycityagent/workflow/__init__.py,sha256=WVBMF-UT3iOBoyFihRrkeqAyLvD_Iibn2axAPOIM
 pycityagent/workflow/prompt.py,sha256=rzenP4EFGxbWE1aq-x2036b6umKvi5cQx2xtWULwgIk,3392
 pycityagent/workflow/block.py,sha256=phr5julVru-Yv-oWyUNMNmm4okhPMNZIF7Up71aoUY0,12249
 pycityagent/workflow/trigger.py,sha256=4nzAwywGQsFabgo6gzp-vD2EV4wII7Z0LHGEAsflSEY,7608
-pycityagent/configs/exp_config.py,sha256=D_OyUBxBx-I5Eyaot52Kl6hb5wEvF5CjwcBUDM6agG8,7319
+pycityagent/configs/exp_config.py,sha256=5cVEfCgBQIXNeidXttvRw31wDIlksWGiwcB0JbyHMb8,7305
 pycityagent/configs/__init__.py,sha256=JAvFkZ6yKTtWvkgO2FaeQfpR-NsStDZPadLH9FOIROU,486
 pycityagent/configs/utils.py,sha256=ILVL5wjLJqxyBVGkV1QOkqp4b9EulSmNrNbD0JQIVqc,429
 pycityagent/configs/sim_config.py,sha256=cCdVu0IypmyOEeWiOLsocL7y7d8uBnLjXsH4_t-pXDk,8209
@@ -81,21 +87,15 @@ pycityagent/cityagent/societyagent.py,sha256=z1i7yPRz7Uupw5oGpMlFwyDi-8R4wHSCjgh
 pycityagent/cityagent/message_intercept.py,sha256=Am0u-hGFOBcvcNXE_WJos5iwh2QoJLihdSIz3sr_U9Y,4439
 pycityagent/cityagent/governmentagent.py,sha256=P3ufIfM5ldh2KC9y_y1ONowsV3FsFF5LWe0iQ3MVmgA,3067
 pycityagent/cityagent/blocks/dispatcher.py,sha256=gd3lyggvtkCq9Ou_icgua60H0UjHTzwHoynLl_iZoes,2939
-pycityagent/cityagent/blocks/needs_block.py,sha256=PDReR1GjF3qafR06Wfw_mizzBZ6xRQRq6KqgA9r8WOA,15321
+pycityagent/cityagent/blocks/needs_block.py,sha256=1Mnr2vStBxSbmh2WByJdefOvTToKOENYDhyZFuL0KiM,15142
 pycityagent/cityagent/blocks/cognition_block.py,sha256=vy7wEvK72-sOieSSWidIXd5nKPLR-4f-ZYL7gyvje-0,15552
 pycityagent/cityagent/blocks/social_block.py,sha256=C1zMcN3RZBYewqw0MEYgtTf5Jzsol1Up3Ew9ovTSm1Y,15169
 pycityagent/cityagent/blocks/__init__.py,sha256=Tmy1CyQ8s3Iltp4xYWNg73Y28UWQv_LNWgMzgzquQT0,420
 pycityagent/cityagent/blocks/economy_block.py,sha256=w1rbTb4CqqXD9BDQuOAtCmrJ4KJjJYiQ5hJHEmCs6CA,20924
-pycityagent/cityagent/blocks/utils.py,sha256=YQkthaMkSCnT-1yfOkioGDvi2c-NLc5B-08BBlW4z0k,1781
+pycityagent/cityagent/blocks/utils.py,sha256=dNIQGv7Br3hWOFznW1JRrFVeGw2vn3QRiPq732we5kQ,1792
 pycityagent/cityagent/blocks/other_block.py,sha256=-H2umOQ-l7_9SjP-SHu7J4oHDh4V-r649_4kGS9Qrng,4447
-pycityagent/cityagent/blocks/plan_block.py,sha256=-3ww7WPzEoDHXJp_WRudllL9GphAQw_-nR426OUEiGY,11897
-pycityagent/cityagent/blocks/mobility_block.py,sha256=rrWzJO6mCMxUFfl1txOqq9BY7FmQ3LE3-Kr0d-4jnCc,16669
+pycityagent/cityagent/blocks/plan_block.py,sha256=3TfXg6Iqa9Z_fh22KV2ZLNffNFglIV22vgCfGmorNe4,11710
+pycityagent/cityagent/blocks/mobility_block.py,sha256=bUoExMdxlMzbnWHe1hx6bkR2Jjc1CZglux8RAXrWaRE,16876
 pycityagent/survey/models.py,sha256=g3xni4GcA1Py3vlGt6z4ltutjgQ4G0uINYAM8vKRJAw,5225
 pycityagent/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
 pycityagent/survey/manager.py,sha256=QZEApC6tKe-g1waDa_O4Eg4CFkyL8KWa7tUGZnT_7WI,3397
-pycityagent-2.0.0a95.dist-info/RECORD,,
-pycityagent-2.0.0a95.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
-pycityagent-2.0.0a95.dist-info/WHEEL,sha256=md3JO_ifs5j508p3TDNMgtQVtnQblpGEt_Wo4W56l8Y,107
-pycityagent-2.0.0a95.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
-pycityagent-2.0.0a95.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
-pycityagent-2.0.0a95.dist-info/METADATA,sha256=WUcWf5gbldc2LyaR71a_QS8cZuGgUs9q0hrbXOdZ8_o,8657