pycityagent 2.0.0a94__cp311-cp311-macosx_11_0_arm64.whl → 2.0.0a96__cp311-cp311-macosx_11_0_arm64.whl

pycityagent/cityagent/blocks/mobility_block.py

@@ -1,17 +1,19 @@
+import json
+import logging
 import math
+import random
+from operator import itemgetter
 from typing import List
 
+import numpy as np
 import ray
 
 from pycityagent.environment.simulator import Simulator
 from pycityagent.llm import LLM
 from pycityagent.memory import Memory
 from pycityagent.workflow.block import Block
-import numpy as np
-from operator import itemgetter
-import random
-import logging
 from pycityagent.workflow.prompt import FormatPrompt
+
 from .dispatcher import BlockDispatcher
 
 logger = logging.getLogger("pycityagent")
@@ -22,6 +24,11 @@ As an intelligent decision system, please determine the type of place the user n
 User Plan: {plan}
 User requirement: {intention}
 Your output must be a single selection from {poi_category} without any additional text or explanation.
+
+Please response in json format (Do not return any other text), example:
+{{
+    "place_type": "shopping"
+}}
 """
 
 PLACE_SECOND_TYPE_SELECTION_PROMPT = """
@@ -29,6 +36,11 @@ As an intelligent decision system, please determine the type of place the user n
 User Plan: {plan}
 User requirement: {intention}
 Your output must be a single selection from {poi_category} without any additional text or explanation.
+
+Please response in json format (Do not return any other text), example:
+{{
+    "place_type": "shopping"
+}}
 """
 
 PLACE_ANALYSIS_PROMPT = """
@@ -37,6 +49,11 @@ User Plan: {plan}
 User requirement: {intention}
 
 Your output must be a single selection from ['home', 'workplace', 'other'] without any additional text or explanation.
+
+Please response in json format (Do not return any other text), example:
+{{
+    "place_type": "home"
+}}
 """
 
 RADIUS_PROMPT = """As an intelligent decision system, please determine the maximum travel radius (in meters) based on the current emotional state.
@@ -48,7 +65,11 @@ Your current thought: {thought}
 
 Please analyze how these emotions would affect travel willingness and return only a single integer number between 3000-200000 representing the maximum travel radius in meters. A more positive emotional state generally leads to greater willingness to travel further.
 
-Return only the integer number without any additional text or explanation."""
+Please response in json format (Do not return any other text), example:
+{{
+    "radius": 10000
+}}
+"""
 
 
 def gravity_model(pois):
@@ -121,7 +142,7 @@ class PlaceSelectionBlock(Block):
     """
 
     configurable_fields: List[str] = ["search_limit"]
-    default_values = {"search_limit": 1000}
+    default_values = {"search_limit": 50}
 
     def __init__(self, llm: LLM, memory: Memory, simulator: Simulator):
         super().__init__(
@@ -134,7 +155,7 @@ class PlaceSelectionBlock(Block):
         )
         self.radiusPrompt = FormatPrompt(RADIUS_PROMPT)
         # configurable fields
-        self.search_limit = 1000
+        self.search_limit = 50
 
     async def forward(self, step, context):
         poi_cate = self.simulator.get_poi_cate()
@@ -143,8 +164,9 @@ class PlaceSelectionBlock(Block):
             intention=step["intention"],
             poi_category=list(poi_cate.keys()),
         )
-        levelOneType = await self.llm.atext_request(self.typeSelectionPrompt.to_dialog())  # type: ignore
+        levelOneType = await self.llm.atext_request(self.typeSelectionPrompt.to_dialog(), response_format={"type": "json_object"})  # type: ignore
         try:
+            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}")
@@ -153,7 +175,12 @@ class PlaceSelectionBlock(Block):
         self.secondTypeSelectionPrompt.format(
             plan=context["plan"], intention=step["intention"], poi_category=sub_category
         )
-        levelTwoType = await self.llm.atext_request(self.secondTypeSelectionPrompt.to_dialog())  # type: ignore
+        levelTwoType = await self.llm.atext_request(self.secondTypeSelectionPrompt.to_dialog(), response_format={"type": "json_object"})  # type: ignore
+        try:
+            levelTwoType = json.loads(levelTwoType)["place_type"]
+        except Exception as e:
+            logger.warning(f"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"])
         self.radiusPrompt.format(
@@ -162,23 +189,28 @@ class PlaceSelectionBlock(Block):
             weather=self.simulator.sence("weather"),
             temperature=self.simulator.sence("temperature"),
         )
-        radius = int(await self.llm.atext_request(self.radiusPrompt.to_dialog()))  # type: ignore
+        radius = await self.llm.atext_request(self.radiusPrompt.to_dialog(), response_format={"type": "json_object"})  # type: ignore
         try:
-            pois = ray.get(self.simulator.map.query_pois.remote(
-                center=center,
-                category_prefix=levelTwoType,
-                radius=radius,
-                limit=self.search_limit,
-            ))
+            radius = int(json.loads(radius)["radius"])
+            pois = ray.get(
+                self.simulator.map.query_pois.remote(
+                    center=center,
+                    category_prefix=levelTwoType,
+                    radius=radius,
+                    limit=self.search_limit,
+                )
+            )
         except Exception as e:
             logger.warning(f"Error querying pois: {e}")
-            levelTwoType = random.choice(sub_category)
-            pois = ray.get(self.simulator.map.query_pois.remote(
-                center=center,
-                category_prefix=levelTwoType,
-                radius=radius,
-                limit=self.search_limit,
-            ))
+            radius = 10000
+            pois = ray.get(
+                self.simulator.map.query_pois.remote(
+                    center=center,
+                    category_prefix=levelTwoType,
+                    radius=radius,
+                    limit=self.search_limit,
+                )
+            )
         if len(pois) > 0:
             pois = gravity_model(pois)
             probabilities = [item[2] for item in pois]
@@ -233,10 +265,12 @@ class MoveBlock(Block):
         self.placeAnalysisPrompt.format(
             plan=context["plan"], intention=step["intention"]
         )
-        number_poi_visited = await self.memory.status.get("number_poi_visited")
-        number_poi_visited += 1
-        await self.memory.status.update("number_poi_visited", number_poi_visited)
-        response = await self.llm.atext_request(self.placeAnalysisPrompt.to_dialog())  # type: ignore
+        response = await self.llm.atext_request(self.placeAnalysisPrompt.to_dialog(), response_format={"type": "json_object"})  # type: ignore
+        try:
+            response = json.loads(response)["place_type"]
+        except Exception as e:
+            logger.warning(f"Wrong type of poi, raw response: {response}")
+            response = "home"
         if response == "home":
             # 返回到家
             home = await self.memory.status.get("home")
@@ -260,6 +294,9 @@ class MoveBlock(Block):
                 person_id=agent_id,
                 target_positions=home,
             )
+            number_poi_visited = await self.memory.status.get("number_poi_visited")
+            number_poi_visited += 1
+            await self.memory.status.update("number_poi_visited", number_poi_visited)
             return {
                 "success": True,
                 "evaluation": f"Successfully returned home",
@@ -290,6 +327,9 @@ class MoveBlock(Block):
                 person_id=agent_id,
                 target_positions=work,
             )
+            number_poi_visited = await self.memory.status.get("number_poi_visited")
+            number_poi_visited += 1
+            await self.memory.status.update("number_poi_visited", number_poi_visited)
             return {
                 "success": True,
                 "evaluation": f"Successfully reached the workplace",
@@ -322,6 +362,9 @@ class MoveBlock(Block):
                     person_id=agent_id,
                     target_positions=next_place[1],
                 )
+            number_poi_visited = await self.memory.status.get("number_poi_visited")
+            number_poi_visited += 1
+            await self.memory.status.update("number_poi_visited", number_poi_visited)
             return {
                 "success": True,
                 "evaluation": f"Successfully reached the destination: {next_place}",

pycityagent/cityagent/blocks/needs_block.py

@@ -31,7 +31,7 @@ Current satisfaction levels (0-1 float values, lower means less satisfied):
 - safety_satisfaction: Safety satisfaction level (Normally, the agent will be more satisfied with safety when they have high income and currency)
 - social_satisfaction: Social satisfaction level
 
-Example response format (Do not return any other text):
+Please response in json format (Do not return any other text), example:
 {{
     "current_satisfaction": {{
         "hunger_satisfaction": 0.8,
@@ -64,12 +64,12 @@ Notes:
 2. If the current need is not "whatever", only return the new value for the current need. Otherwise, return both safe and social need values.
 3. Ensure the return value is in valid JSON format, examples below:
 
-Example response format for specific need (hungry here) adjustment (Do not return any other text):
+Please response in json format for specific need (hungry here) adjustment (Do not return any other text), example:
 {{
     "hunger_satisfaction": new_need_value
 }}
 
-Example response format for whatever need adjustment (Do not return any other text):
+Please response in json format for whatever need adjustment (Do not return any other text), example:
 {{
     "safety_satisfaction": new_safe_value,
     "social_satisfaction": new_social_value
@@ -93,23 +93,22 @@ class NeedsBlock(Block):
         self.token_consumption = 0
         self.initialized = False
         self.alpha_H, self.alpha_D, self.alpha_P, self.alpha_C = (
-            0.2,
+            0.15,
             0.08,
             0.05,
             0.1,
         )  # Hunger decay rate, Energy decay rate, Safety decay rate, Social decay rate
         self.T_H, self.T_D, self.T_P, self.T_C = (
-            0.1,
             0.2,
             0.2,
-            0.4,
+            0.2,
+            0.3,
         )  # Hunger threshold, Energy threshold, Safety threshold, Social threshold
 
     async def initialize(self):
         day = await self.simulator.get_simulator_day()
         if day != self.now_day:
             self.now_day = day
-            self.initialized = False
             workday = self.simulator.sence("day")
             if workday == "Workday":
                 self.need_work = True
@@ -126,7 +125,7 @@ class NeedsBlock(Block):
                 income=await self.memory.status.get("income"),
                 now_time=await self.simulator.get_time(format_time=True),
             )
-            response = await self.llm.atext_request(self.initial_prompt.to_dialog())
+            response = await self.llm.atext_request(self.initial_prompt.to_dialog(), response_format={"type": "json_object"})
             response = self.clean_json_response(response)
             retry = 3
             while retry > 0:
@@ -229,13 +228,13 @@ class NeedsBlock(Block):
                 await self.memory.status.update("current_need", "hungry")
             elif energy_satisfaction <= self.T_D:
                 await self.memory.status.update("current_need", "tired")
+            elif self.need_work:
+                await self.memory.status.update("current_need", "safe")
+                self.need_work = False
             elif safety_satisfaction <= self.T_P:
                 await self.memory.status.update("current_need", "safe")
             elif social_satisfaction <= self.T_C:
                 await self.memory.status.update("current_need", "social")
-            elif self.need_work:
-                await self.memory.status.update("current_need", "safe")
-                self.need_work = False
             else:
                 await self.memory.status.update("current_need", "whatever")
         else:
@@ -308,7 +307,7 @@ class NeedsBlock(Block):
 
         retry = 3
         while retry > 0:
-            response = await self.llm.atext_request(self.evaluation_prompt.to_dialog())
+            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
                 # 更新所有需求的数值

pycityagent/cityagent/blocks/other_block.py

@@ -25,7 +25,7 @@ class SleepBlock(Block):
             intention=step["intention"],
             emotion_types=await self.memory.status.get("emotion_types"),
         )
-        result = await self.llm.atext_request(self.guidance_prompt.to_dialog())
+        result = await self.llm.atext_request(self.guidance_prompt.to_dialog(), response_format={"type": "json_object"})
         result = clean_json_response(result)
         node_id = await self.memory.stream.add_other(description=f"I slept")
         try:
@@ -63,7 +63,7 @@ class OtherNoneBlock(Block):
             intention=step["intention"],
             emotion_types=await self.memory.status.get("emotion_types"),
         )
-        result = await self.llm.atext_request(self.guidance_prompt.to_dialog())
+        result = await self.llm.atext_request(self.guidance_prompt.to_dialog(), response_format={"type": "json_object"})
         result = clean_json_response(result)
         node_id = await self.memory.stream.add_other(
             description=f"I {step['intention']}"

pycityagent/cityagent/blocks/plan_block.py

@@ -31,7 +31,7 @@ Please evaluate and select the most appropriate option based on these three dime
 2. Subjective Norm: Social environment and others' views on this behavior
 3. Perceived Control: Difficulty and controllability of executing this option
 
-Please return the evaluation results in JSON format (Do not return any other text):
+Please response in json format (Do not return any other text), example:
 {{
     "selected_option": "Select the most suitable option from available options",
     "evaluation": {{
@@ -54,19 +54,6 @@ Current time: {current_time}
 Your emotion: {emotion_types}
 Your thought: {thought}
 
-Please generate specific execution steps and return in JSON format:
-{{
-    "plan": {{
-        "target": "Specific goal",
-        "steps": [
-            {{
-                "intention": "Specific intention",
-                "type": "Step type"
-            }}
-        ]
-    }}
-}}
-
 Notes:
 1. type can only be one of these four: mobility, social, economy, other
     1.1 mobility: Decisions or behaviors related to large-scale spatial movement, such as location selection, going to a place, etc.
@@ -76,7 +63,7 @@ Notes:
 2. steps should only include steps necessary to fulfill the target (limited to {max_plan_steps} steps)
 3. intention in each step should be concise and clear
 
-Example outputs (Do not return any other text):
+Please response in json format (Do not return any other text), example:
 {{
     "plan": {{
         "target": "Eat at home",
@@ -172,10 +159,10 @@ class PlanBlock(Block):
         self.token_consumption = 0
         self.guidance_options = {
             "hungry": ["Eat at home", "Eat outside"],
-            "tired": ["Sleep", "Take a nap"],
+            "tired": ["Sleep"],
             "safe": ["Go to work"],
             "social": ["Contact with friends", "Shopping"],
-            "whatever": ["Contact with friends", "Hang out"],
+            "whatever": ["Contact with friends", "Hang out", "Entertainment"],
         }
 
         # configurable fields
@@ -211,7 +198,7 @@ class PlanBlock(Block):
         )
 
         response = await self.llm.atext_request(
-            self.guidance_prompt.to_dialog()
+            self.guidance_prompt.to_dialog(), response_format={"type": "json_object"}
         )  # type: ignore
         retry = 3
         while retry > 0:
@@ -219,8 +206,15 @@ class PlanBlock(Block):
                 result = json.loads(self.clean_json_response(response))  # type: ignore
                 if "selected_option" not in result or "evaluation" not in result:
                     raise ValueError("Invalid guidance selection format")
-                if "attitude" not in result["evaluation"] or "subjective_norm" not in result["evaluation"] or "perceived_control" not in result["evaluation"] or "reasoning" not in result["evaluation"]:
-                    raise ValueError("Evaluation must include attitude, subjective_norm, perceived_control, and reasoning")
+                if (
+                    "attitude" not in result["evaluation"]
+                    or "subjective_norm" not in result["evaluation"]
+                    or "perceived_control" not in result["evaluation"]
+                    or "reasoning" not in result["evaluation"]
+                ):
+                    raise ValueError(
+                        "Evaluation must include attitude, subjective_norm, perceived_control, and reasoning"
+                    )
                 return result
             except Exception as e:
                 logger.warning(f"Error parsing guidance selection response: {str(e)}")
@@ -228,7 +222,7 @@ class PlanBlock(Block):
         return None
 
     async def generate_detailed_plan(
-        self, current_need: str, selected_option: str
+        self, selected_option: str
     ) -> Dict:
         """Generate detailed execution plan"""
         position_now = await self.memory.status.get("position")
@@ -262,7 +256,11 @@ class PlanBlock(Block):
         while retry > 0:
             try:
                 result = json.loads(self.clean_json_response(response))  # type: ignore
-                if "plan" not in result or "target" not in result["plan"] or "steps" not in result["plan"]:
+                if (
+                    "plan" not in result
+                    or "target" not in result["plan"]
+                    or "steps" not in result["plan"]
+                ):
                     raise ValueError("Invalid plan format")
                 for step in result["plan"]["steps"]:
                     if "intention" not in step or "type" not in step:
@@ -282,10 +280,10 @@ class PlanBlock(Block):
 
         # Step 2: Generate detailed plan
         detailed_plan = await self.generate_detailed_plan(
-            current_need, guidance_result["selected_option"]
+            guidance_result["selected_option"]
         )
 
-        if not detailed_plan or "plan" not in detailed_plan:
+        if not detailed_plan:
             await self.memory.status.update("current_plan", None)
             await self.memory.status.update(
                 "current_step", {"intention": "", "type": ""}

pycityagent/cityagent/blocks/social_block.py

@@ -18,12 +18,10 @@ logger = logging.getLogger("pycityagent")
 
 
 class MessagePromptManager:
-    def __init__(self, template: str, to_discuss: List[str]):
-        self.template = template
-        self.format_prompt = FormatPrompt(self.template)
-        self.to_discuss = to_discuss
+    def __init__(self):
+        pass
 
-    async def get_prompt(self, memory, step: Dict[str, Any], target: str) -> str:
+    async def get_prompt(self, memory, step: Dict[str, Any], target: str, template: str) -> str:
         """在这里改给模板输入的数据"""
         # 获取数据
         relationships = await memory.status.get("relationships") or {}
@@ -31,14 +29,17 @@ class MessagePromptManager:
 
         # 构建讨论话题约束
         discussion_constraint = ""
-        if self.to_discuss:
-            topics = ", ".join(f'"{topic}"' for topic in self.to_discuss)
+        topics = await memory.status.get("attitude")
+        topics = topics.keys()
+        if topics:
+            topics = ", ".join(f'"{topic}"' for topic in topics)
             discussion_constraint = (
                 f"Limit your discussion to the following topics: {topics}."
             )
 
         # 格式化提示
-        self.format_prompt.format(
+        format_prompt = FormatPrompt(template)
+        format_prompt.format(
             gender=await memory.status.get("gender") or "",
             education=await memory.status.get("education") or "",
             personality=await memory.status.get("personality") or "",
@@ -55,7 +56,7 @@ class MessagePromptManager:
             discussion_constraint=discussion_constraint,
         )
 
-        return self.format_prompt.to_dialog()
+        return format_prompt.to_dialog()
 
 
 class SocialNoneBlock(Block):
@@ -75,7 +76,7 @@ class SocialNoneBlock(Block):
             intention=step["intention"],
             emotion_types=await self.memory.status.get("emotion_types"),
         )
-        result = await self.llm.atext_request(self.guidance_prompt.to_dialog())
+        result = await self.llm.atext_request(self.guidance_prompt.to_dialog(), response_format={"type": "json_object"})
         result = clean_json_response(result)
         try:
             result = json.loads(result)
@@ -270,11 +271,8 @@ class MessageBlock(Block):
         The message should reflect my personality and background.
         {discussion_constraint}
         """
-        self.to_discuss = []
 
-        self.prompt_manager = MessagePromptManager(
-            self.default_message_template, self.to_discuss
-        )
+        self.prompt_manager = MessagePromptManager()
 
     def _serialize_message(self, message: str, propagation_count: int) -> str:
         try:
@@ -305,7 +303,7 @@ class MessageBlock(Block):
 
             # Get formatted prompt using prompt manager
             formatted_prompt = await self.prompt_manager.get_prompt(
-                self.memory, step, target
+                self.memory, step, target, self.default_message_template
             )
 
             # Generate message
@@ -319,8 +317,8 @@ class MessageBlock(Block):
                 chat_histories = {}
             if target not in chat_histories:
                 chat_histories[target] = ""
-            if chat_histories[target]:
-                chat_histories[target] += "，"
+            elif len(chat_histories[target]) > 0:
+                chat_histories[target] += ", "
             chat_histories[target] += f"me: {message}"
 
             await self.memory.status.update("chat_histories", chat_histories)

pycityagent/cityagent/blocks/utils.py

@@ -17,12 +17,13 @@ Examples:
 - "Read a book": {{"time": 90}}
 - "Exercise": {{"time": 45}}
 
-Please return the result in JSON format (Do not return any other text):
+Please return the result in JSON format (Do not return any other text), example:
 {{
-    "time": estimated completion time (integer, in minutes)
+    "time": 10
 }}
 """
 
+
 def prettify_document(document: str) -> str:
     # Remove sequences of whitespace characters (including newlines)
     cleaned = re.sub(r"\s+", " ", document).strip()

pycityagent/cityagent/firmagent.py

@@ -3,7 +3,7 @@ from typing import Optional
 import numpy as np
 from pycityagent import Simulator, InstitutionAgent
 from pycityagent.llm import LLM
-from pycityagent.economy import EconomyClient
+from pycityagent.environment import EconomyClient
 from pycityagent.message import Messager
 from pycityagent.memory import Memory
 import logging

pycityagent/cityagent/governmentagent.py

@@ -4,7 +4,7 @@ from typing import Optional
 import numpy as np
 from pycityagent import Simulator, InstitutionAgent
 from pycityagent.llm.llm import LLM
-from pycityagent.economy import EconomyClient
+from pycityagent.environment import EconomyClient
 from pycityagent.message import Messager
 from pycityagent.memory import Memory
 import logging

pycityagent/cityagent/initial.py

@@ -63,7 +63,7 @@ async def initialize_social_network(simulation):
 
             # Initialize empty chat histories and interaction records
             await simulation.update(
-                agent_id, "chat_histories", {friend_id: [] for friend_id in friends}
+                agent_id, "chat_histories", {friend_id: "" for friend_id in friends}
             )
             await simulation.update(
                 agent_id, "interactions", {friend_id: [] for friend_id in friends}

pycityagent/cityagent/memory_config.py

@@ -88,7 +88,6 @@ def memory_config_societyagent():
         "relation_types": (dict, {}, False),
         "chat_histories": (dict, {}, False),  # all chat histories
         "interactions": (dict, {}, False),  # all interaction records
-        "to_discuss": (dict, {}, False),
         # mobility
         "number_poi_visited": (int, 1, False),
     }

pycityagent/cityagent/message_intercept.py

@@ -1,5 +1,4 @@
 import asyncio
-import json
 
 from pycityagent.llm import LLM
 from pycityagent.message import MessageBlockBase, MessageBlockListenerBase
@@ -8,15 +7,15 @@ from pycityagent.message import MessageBlockBase, MessageBlockListenerBase
 async def check_message(
     from_uuid: str, to_uuid: str, llm_client: LLM, content: str
 ) -> bool:
-    print(f"\n检查消息: {from_uuid} -> {to_uuid}: {content}")
+    print(f"\n Checking Message: {from_uuid} -> {to_uuid}: {content}")
     is_valid = True
     prompt = f"""
-        请判断以下消息是否具有情绪煽动性：
-        发送者ID: {from_uuid}
-        接收者ID: {to_uuid}
-        消息内容: {content}
+        Please determine if the following message is emotionally provocative:
+        Sender ID: {from_uuid}
+        Receiver ID: {to_uuid}
+        Message content: {content}
         
-        如果消息具有情绪煽动性，请返回 False；如果消息正常，请返回 True。
+        If the message is emotionally provocative, please return False; if the message is normal, please return True.
         """
     for _ in range(10):
         try:
@@ -33,7 +32,7 @@ async def check_message(
             pass
     else:
         raise RuntimeError(f"Request for message interception prompt=`{prompt}` failed")
-    print(f"消息检查结果: {'合规' if is_valid else '不合规'}")
+    print(f"Message Check: {'Valid' if is_valid else 'Invalid'}")
     return is_valid
 
 

pycityagent/cityagent/nbsagent.py

@@ -5,7 +5,7 @@ from typing import Optional
 import numpy as np
 
 from pycityagent import InstitutionAgent, Simulator
-from pycityagent.economy import EconomyClient
+from pycityagent.environment import EconomyClient
 from pycityagent.llm.llm import LLM
 from pycityagent.memory import Memory
 from pycityagent.message import Messager

pycityagent/cityagent/societyagent.py

@@ -6,7 +6,7 @@ from typing import Optional
 
 from pycityagent import CitizenAgent, Simulator
 from pycityagent.agent import Agent
-from pycityagent.economy import EconomyClient
+from pycityagent.environment import EconomyClient
 from pycityagent.llm.llm import LLM
 from pycityagent.memory import Memory
 from pycityagent.tools import UpdateWithSimulator
@@ -427,7 +427,6 @@ class SocietyAgent(CitizenAgent):
                 )
 
                 if should_respond.strip().upper() != "YES":  # type:ignore
-                    await self.memory.status.update("chat_histories", chat_histories)
                     return ""
 
                 response_prompt = f"""Based on:

pycityagent/configs/__init__.py

@@ -0,0 +1,18 @@
+from typing import TYPE_CHECKING
+
+from .exp_config import AgentConfig, ExpConfig, WorkflowStep
+from .sim_config import (LLMRequestConfig, MapRequestConfig, MlflowConfig,
+                         SimConfig, SimulatorRequestConfig)
+from .utils import load_config_from_file
+
+__all__ = [
+    "SimConfig",
+    "SimulatorRequestConfig",
+    "MapRequestConfig",
+    "MlflowConfig",
+    "LLMRequestConfig",
+    "ExpConfig",
+    "load_config_from_file",
+    "WorkflowStep",
+    "AgentConfig",
+]