pycityagent 2.0.0a21__py3-none-any.whl → 2.0.0a24__py3-none-any.whl

pycityagent/memory/memory_base.py

@@ -6,7 +6,8 @@ import asyncio
 import logging
 import time
 from abc import ABC, abstractmethod
-from typing import Any, Callable, Dict, Optional, Sequence, Union
+from collections.abc import Callable, Sequence
+from typing import Any, Optional, Union
 
 from .const import *
 
@@ -16,8 +17,8 @@ logger = logging.getLogger("pycityagent")
 class MemoryUnit:
     def __init__(
         self,
-        content: Optional[Dict] = None,
-        required_attributes: Optional[Dict] = None,
+        content: Optional[dict] = None,
+        required_attributes: Optional[dict] = None,
         activate_timestamp: bool = False,
     ) -> None:
         self._content = {}
@@ -52,7 +53,7 @@ class MemoryUnit:
         else:
             setattr(self, f"{SELF_DEFINE_PREFIX}{property_name}", property_value)
 
-    async def update(self, content: Dict) -> None:
+    async def update(self, content: dict) -> None:
         await self._lock.acquire()
         for k, v in content.items():
             if k in self._content:
@@ -111,14 +112,14 @@ class MemoryUnit:
 
     async def dict_values(
         self,
-    ) -> Dict[Any, Any]:
+    ) -> dict[Any, Any]:
         return self._content
 
 
 class MemoryBase(ABC):
 
     def __init__(self) -> None:
-        self._memories: Dict[Any, Dict] = {}
+        self._memories: dict[Any, dict] = {}
         self._lock = asyncio.Lock()
 
     @abstractmethod

pycityagent/memory/profile.py

@@ -2,8 +2,9 @@
 Agent Profile
 """
 
+from collections.abc import Callable, Sequence
 from copy import deepcopy
-from typing import Any, Callable, Dict, Optional, Sequence, Union, cast
+from typing import Any, Optional, Union, cast
 
 from ..utils.decorators import lock_decorator
 from .const import *
@@ -14,7 +15,7 @@ from .utils import convert_msg_to_sequence
 class ProfileMemoryUnit(MemoryUnit):
     def __init__(
         self,
-        content: Optional[Dict] = None,
+        content: Optional[dict] = None,
         activate_timestamp: bool = False,
     ) -> None:
         super().__init__(
@@ -28,7 +29,7 @@ class ProfileMemory(MemoryBase):
     def __init__(
         self,
         msg: Optional[
-            Union[ProfileMemoryUnit, Sequence[ProfileMemoryUnit], Dict, Sequence[Dict]]
+            Union[ProfileMemoryUnit, Sequence[ProfileMemoryUnit], dict, Sequence[dict]]
         ] = None,
         activate_timestamp: bool = False,
     ) -> None:
@@ -74,7 +75,7 @@ class ProfileMemory(MemoryBase):
     @lock_decorator
     async def load(
         self,
-        snapshots: Union[Dict, Sequence[Dict]],
+        snapshots: Union[dict, Sequence[dict]],
         reset_memory: bool = False,
     ) -> None:
         if reset_memory:
@@ -91,7 +92,7 @@ class ProfileMemory(MemoryBase):
     @lock_decorator
     async def export(
         self,
-    ) -> Sequence[Dict]:
+    ) -> Sequence[dict]:
         _res = []
         for m in self._memories.keys():
             m = cast(ProfileMemoryUnit, m)
@@ -145,7 +146,7 @@ class ProfileMemory(MemoryBase):
                     self._memories[unit] = {}
 
     @lock_decorator
-    async def update_dict(self, to_update_dict: Dict, store_snapshot: bool = False):
+    async def update_dict(self, to_update_dict: dict, store_snapshot: bool = False):
         _latest_memories = self._fetch_recent_memory()
         _latest_memory: ProfileMemoryUnit = _latest_memories[-1]
         if not store_snapshot:

pycityagent/memory/self_define.py

@@ -2,8 +2,9 @@
 Self Define Data
 """
 
+from collections.abc import Callable, Sequence
 from copy import deepcopy
-from typing import Any, Callable, Dict, Optional, Sequence, Union, cast
+from typing import Any, Optional, Union, cast
 
 from ..utils.decorators import lock_decorator
 from .const import *
@@ -14,8 +15,8 @@ from .utils import convert_msg_to_sequence
 class DynamicMemoryUnit(MemoryUnit):
     def __init__(
         self,
-        content: Optional[Dict] = None,
-        required_attributes: Optional[Dict] = None,
+        content: Optional[dict] = None,
+        required_attributes: Optional[dict] = None,
         activate_timestamp: bool = False,
     ) -> None:
         super().__init__(
@@ -29,7 +30,7 @@ class DynamicMemory(MemoryBase):
 
     def __init__(
         self,
-        required_attributes: Dict[Any, Any],
+        required_attributes: dict[Any, Any],
         activate_timestamp: bool = False,
     ) -> None:
         super().__init__()
@@ -69,7 +70,7 @@ class DynamicMemory(MemoryBase):
     @lock_decorator
     async def load(
         self,
-        snapshots: Union[Dict, Sequence[Dict]],
+        snapshots: Union[dict, Sequence[dict]],
         reset_memory: bool = False,
     ) -> None:
         if reset_memory:
@@ -86,7 +87,7 @@ class DynamicMemory(MemoryBase):
     @lock_decorator
     async def export(
         self,
-    ) -> Sequence[Dict]:
+    ) -> Sequence[dict]:
         _res = []
         for m in self._memories.keys():
             m = cast(DynamicMemoryUnit, m)
@@ -143,7 +144,7 @@ class DynamicMemory(MemoryBase):
                     self._memories[unit] = {}
 
     @lock_decorator
-    async def update_dict(self, to_update_dict: Dict, store_snapshot: bool = False):
+    async def update_dict(self, to_update_dict: dict, store_snapshot: bool = False):
         _latest_memories = self._fetch_recent_memory()
         _latest_memory: DynamicMemoryUnit = _latest_memories[-1]
         if not store_snapshot:

pycityagent/memory/state.py

@@ -2,8 +2,9 @@
 Agent State
 """
 
+from collections.abc import Callable, Sequence
 from copy import deepcopy
-from typing import Any, Callable, Dict, Optional, Sequence, Union, cast
+from typing import Any, Optional, Union, cast
 
 from ..utils.decorators import lock_decorator
 from .const import *
@@ -14,7 +15,7 @@ from .utils import convert_msg_to_sequence
 class StateMemoryUnit(MemoryUnit):
     def __init__(
         self,
-        content: Optional[Dict] = None,
+        content: Optional[dict] = None,
         activate_timestamp: bool = False,
     ) -> None:
         super().__init__(
@@ -28,7 +29,7 @@ class StateMemory(MemoryBase):
     def __init__(
         self,
         msg: Optional[
-            Union[MemoryUnit, Sequence[MemoryUnit], Dict, Sequence[Dict]]
+            Union[MemoryUnit, Sequence[MemoryUnit], dict, Sequence[dict]]
         ] = None,
         activate_timestamp: bool = False,
     ) -> None:
@@ -73,7 +74,7 @@ class StateMemory(MemoryBase):
     @lock_decorator
     async def load(
         self,
-        snapshots: Union[Dict, Sequence[Dict]],
+        snapshots: Union[dict, Sequence[dict]],
         reset_memory: bool = False,
     ) -> None:
 
@@ -91,7 +92,7 @@ class StateMemory(MemoryBase):
     @lock_decorator
     async def export(
         self,
-    ) -> Sequence[Dict]:
+    ) -> Sequence[dict]:
 
         _res = []
         for m in self._memories.keys():
@@ -151,7 +152,7 @@ class StateMemory(MemoryBase):
                     self._memories[unit] = {}
 
     @lock_decorator
-    async def update_dict(self, to_update_dict: Dict, store_snapshot: bool = False):
+    async def update_dict(self, to_update_dict: dict, store_snapshot: bool = False):
 
         _latest_memories = self._fetch_recent_memory()
         _latest_memory: StateMemoryUnit = _latest_memories[-1]

pycityagent/memory/utils.py

@@ -1,4 +1,5 @@
-from typing import Any, Sequence, Union
+from collections.abc import Sequence
+from typing import Any, Union
 
 from .memory_base import MemoryUnit
 

pycityagent/simulation/agentgroup.py

@@ -10,12 +10,14 @@ from uuid import UUID
 
 import fastavro
 import ray
+from langchain_core.embeddings import Embeddings
 
 from ..agent import Agent, CitizenAgent, InstitutionAgent
 from ..economy.econ_client import EconomyClient
 from ..environment.simulator import Simulator
 from ..llm.llm import LLM
 from ..llm.llmconfig import LLMConfig
+from ..memory import FaissQuery
 from ..message import Messager
 from ..metrics import MlflowClient
 from ..utils import (DIALOG_SCHEMA, INSTITUTION_STATUS_SCHEMA, PROFILE_SCHEMA,
@@ -37,6 +39,7 @@ class AgentGroup:
         enable_pgsql: bool,
         pgsql_writer: ray.ObjectRef,
         mlflow_run_id: str,
+        embedding_model: Embeddings,
         logging_level: int,
     ):
         logger.setLevel(logging_level)
@@ -46,6 +49,7 @@ class AgentGroup:
         self.exp_id = exp_id
         self.enable_avro = enable_avro
         self.enable_pgsql = enable_pgsql
+        self.embedding_model = embedding_model
         if enable_avro:
             self.avro_path = avro_path / f"{self._uuid}"
             self.avro_path.mkdir(parents=True, exist_ok=True)
@@ -99,6 +103,13 @@ class AgentGroup:
         else:
             self.mlflow_client = None
 
+        # set FaissQuery
+        if self.embedding_model is not None:
+            self.faiss_query = FaissQuery(
+                embeddings=self.embedding_model,
+            )
+        else:
+            self.faiss_query = None
         for agent in self.agents:
             agent.set_exp_id(self.exp_id)  # type: ignore
             agent.set_llm_client(self.llm)
@@ -112,6 +123,12 @@ class AgentGroup:
                 agent.set_avro_file(self.avro_file)  # type: ignore
             if self.enable_pgsql:
                 agent.set_pgsql_writer(self._pgsql_writer)
+            # set memory.faiss_query
+            if self.faiss_query is not None:
+                agent.memory.set_faiss_query(self.faiss_query)
+            # set memory.embedding model
+            if self.embedding_model is not None:
+                agent.memory.set_embedding_model(self.embedding_model)
 
     async def init_agents(self):
         logger.debug(f"-----Initializing Agents in AgentGroup {self._uuid} ...")
@@ -376,32 +393,32 @@ class AgentGroup:
                             "created_at": _date_time,
                         }
                         _statuses_time_list.append((_status_dict, _date_time))
-        to_update_statues: list[tuple] = []
-        for _status_dict, _ in _statuses_time_list:
-            BASIC_KEYS = [
-                "id",
-                "day",
-                "t",
-                "lng",
-                "lat",
-                "parent_id",
-                "action",
-                "created_at",
-            ]
-            _data = [_status_dict[k] for k in BASIC_KEYS if k != "created_at"]
-            _other_dict = json.dumps(
-                {k: v for k, v in _status_dict.items() if k not in BASIC_KEYS}
-            )
-            _data.append(_other_dict)
-            _data.append(_status_dict["created_at"])
-            to_update_statues.append(tuple(_data))
-        if self._last_asyncio_pg_task is not None:
-            await self._last_asyncio_pg_task
-        self._last_asyncio_pg_task = (
-            self._pgsql_writer.async_write_status.remote(  # type:ignore
-                to_update_statues
+            to_update_statues: list[tuple] = []
+            for _status_dict, _ in _statuses_time_list:
+                BASIC_KEYS = [
+                    "id",
+                    "day",
+                    "t",
+                    "lng",
+                    "lat",
+                    "parent_id",
+                    "action",
+                    "created_at",
+                ]
+                _data = [_status_dict[k] for k in BASIC_KEYS if k != "created_at"]
+                _other_dict = json.dumps(
+                    {k: v for k, v in _status_dict.items() if k not in BASIC_KEYS}
+                )
+                _data.append(_other_dict)
+                _data.append(_status_dict["created_at"])
+                to_update_statues.append(tuple(_data))
+            if self._last_asyncio_pg_task is not None:
+                await self._last_asyncio_pg_task
+            self._last_asyncio_pg_task = (
+                self._pgsql_writer.async_write_status.remote(  # type:ignore
+                    to_update_statues
+                )
             )
-        )
 
     async def step(self):
         if not self.initialized:

pycityagent/simulation/simulation.py

@@ -14,11 +14,13 @@ from typing import Any, Optional, Union
 import pycityproto.city.economy.v2.economy_pb2 as economyv2
 import ray
 import yaml
+from langchain_core.embeddings import Embeddings
 from mosstool.map._map_util.const import AOI_START_ID
 
 from ..agent import Agent, InstitutionAgent
 from ..environment.simulator import Simulator
-from ..memory.memory import Memory
+from ..llm import SimpleEmbedding
+from ..memory import FaissQuery, Memory
 from ..message.messager import Messager
 from ..metrics import init_mlflow_connection
 from ..survey import Survey
@@ -76,6 +78,8 @@ class AgentSimulation:
 
         # storage
         _storage_config: dict[str, Any] = config.get("storage", {})
+        if _storage_config is None:
+            _storage_config = {}
         # avro
         _avro_config: dict[str, Any] = _storage_config.get("avro", {})
         self._enable_avro = _avro_config.get("enabled", False)
@@ -164,6 +168,7 @@ class AgentSimulation:
         enable_pgsql: bool,
         pgsql_writer: ray.ObjectRef,
         mlflow_run_id: str = None,  # type: ignore
+        embedding_model: Embeddings = None,  # type: ignore
         logging_level: int = logging.WARNING,
     ):
         """创建远程组"""
@@ -177,6 +182,7 @@ class AgentSimulation:
             enable_pgsql,
             pgsql_writer,
             mlflow_run_id,
+            embedding_model,
             logging_level,
         )
         return group_name, group, agents
@@ -186,6 +192,7 @@ class AgentSimulation:
         agent_count: Union[int, list[int]],
         group_size: int = 1000,
         pg_sql_writers: int = 32,
+        embedding_model: Embeddings = SimpleEmbedding(),
         memory_config_func: Optional[Union[Callable, list[Callable]]] = None,
     ) -> None:
         """初始化智能体
@@ -305,6 +312,7 @@ class AgentSimulation:
                 self.enable_pgsql,
                 _workers[i % _num_workers],  # type:ignore
                 mlflow_run_id,  # type:ignore
+                embedding_model,
                 self.logging_level,
             )
             creation_tasks.append((group_name, group, agents))

pycityagent/utils/parsers/json_parser.py

@@ -1,6 +1,6 @@
 import json
 import logging
-from typing import Any, Dict
+from typing import Any
 
 from .parser_base import ParserBase
 
@@ -67,14 +67,14 @@ class JsonDictParser(JsonObjectParser):
         """Initialize the JsonDictParser with default tags."""
         super().__init__()
 
-    def parse(self, response: str) -> Dict:
+    def parse(self, response: str) -> dict:
         """Parse the response string to extract and return a JSON object as a dictionary.
 
         Parameters:
             response (str): The response string containing the JSON object.
 
         Returns:
-            Dict: The parsed JSON object as a dictionary.
+            dict: The parsed JSON object as a dictionary.
         """
         parsed_json = super().parse(response)
         if not isinstance(parsed_json, dict):

pycityagent/workflow/block.py

@@ -1,7 +1,8 @@
 import asyncio
 import functools
 import inspect
-from typing import Any, Callable, Coroutine, Optional, Union
+from collections.abc import Awaitable, Callable, Coroutine
+from typing import Any, Optional, Union
 
 from ..environment.simulator import Simulator
 from ..llm import LLM

{pycityagent-2.0.0a21.dist-info → pycityagent-2.0.0a24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycityagent
-Version: 2.0.0a21
+Version: 2.0.0a24
 Summary: LLM-based城市环境agent构建库
 License: MIT
 Author: Yuwei Yan
@@ -20,10 +20,12 @@ Requires-Dist: aiohttp (==3.10.10)
 Requires-Dist: aiomqtt (>=2.3.0,<3.0.0)
 Requires-Dist: citystreetview (==1.2.4)
 Requires-Dist: dashscope (==1.14.1)
+Requires-Dist: faiss-cpu (>=1.9.0.post1,<2.0.0)
 Requires-Dist: fastavro (>=1.10.0,<2.0.0)
 Requires-Dist: geojson (==3.1.0)
 Requires-Dist: gradio (>=5.7.1,<6.0.0)
 Requires-Dist: grpcio (==1.67.1)
+Requires-Dist: langchain-community (>=0.3.13,<0.4.0)
 Requires-Dist: langchain-core (>=0.3.28,<0.4.0)
 Requires-Dist: matplotlib (==3.8.3)
 Requires-Dist: mlflow (>=2.19.0,<3.0.0)
@@ -40,6 +42,8 @@ Requires-Dist: pycityproto (>=2.1.5,<3.0.0)
 Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
 Requires-Dist: ray (>=2.40.0,<3.0.0)
 Requires-Dist: sidecar (==0.7.0)
+Requires-Dist: torch (>=2.5.1,<3.0.0)
+Requires-Dist: transformers (>=4.47.1,<5.0.0)
 Requires-Dist: zhipuai (>=2.1.5.20230904,<3.0.0.0)
 Description-Content-Type: text/markdown
 

{pycityagent-2.0.0a21.dist-info → pycityagent-2.0.0a24.dist-info}/RECORD

@@ -1,5 +1,5 @@
-pycityagent/__init__.py,sha256=EDxt3Su3lH1IMh9suNw7GeGL7UrXeWiZTw5KWNznDzc,637
-pycityagent/agent.py,sha256=FMplKAgcz2Exkl8EiE2RwQ0Hd5U08krRZ3CFFLoF_4g,28450
+pycityagent/__init__.py,sha256=fv0mzNGbHBF6m550yYqnuUpB8iQPWS-7EatYRK7DO4s,693
+pycityagent/agent.py,sha256=l8Oa95_K5JBWKzvZmbQe_QM_E_vaG-YstuuR55kgC6Y,29005
 pycityagent/economy/__init__.py,sha256=aonY4WHnx-6EGJ4WKrx4S-2jAkYNLtqUA04jp6q8B7w,75
 pycityagent/economy/econ_client.py,sha256=GuHK9ZBnhqW3Z7F8ViDJn_iN73yOBbbwFyJv1wLEBDk,12211
 pycityagent/environment/__init__.py,sha256=awHxlOud-btWbk0FCS4RmGJ13W84oVCkbGfcrhKqihA,240
@@ -11,16 +11,16 @@ pycityagent/environment/sence/static.py,sha256=9s7jz8HitstTrk-GqpnVB26oPrjuTyNeL
 pycityagent/environment/sidecar/__init__.py,sha256=RFbOf40aYBP4WwRpFkua5tlRE_OtMcMNyp1Lew_aaAU,235
 pycityagent/environment/sidecar/sidecarv2.py,sha256=beKlYZgt38EQbV1x6NWQo7xVXyB-5QHfbwJexyXu7Tg,3252
 pycityagent/environment/sim/__init__.py,sha256=JVG6sSD2Hbohl1TtKjuQi7_M7tKMrFh9vl3QV3VA5O0,724
-pycityagent/environment/sim/aoi_service.py,sha256=yM50xhYPLLExwjl5MDlnAuohqJpK0KbIvr_cWLZJEAk,1203
+pycityagent/environment/sim/aoi_service.py,sha256=2UjvUTF4CW4E_L30IRcdwv6t_q1ZdXN3TTLOKSOaaXE,1230
 pycityagent/environment/sim/client.py,sha256=MHR7Hhu-Cr7B61d3K_QDpSJh4HrUU9JQf6Cu7_8pdsE,3493
-pycityagent/environment/sim/clock_service.py,sha256=YG_A_IA7iJp9rOQE3bCPgppzuzzUwWMCzlxSUdOy088,1326
-pycityagent/environment/sim/economy_services.py,sha256=ZggCyAK4Tv5SS881lUBe0WyymKq-0PtdWpe8p_pTDiI,7085
-pycityagent/environment/sim/lane_service.py,sha256=6ljWiX_GbZd-a-fEcktDZncxBkraLA2LDE_pVV99uLs,3896
-pycityagent/environment/sim/light_service.py,sha256=ezYGcV_PkZ6I-yv48LNEjsytwwwQx-1Qp2QCWXWIhdQ,4215
-pycityagent/environment/sim/person_service.py,sha256=nIvOsoBoqOTDYtsiThg07-4ZBgkTUDEbb3dHyOjzyb8,10652
-pycityagent/environment/sim/road_service.py,sha256=phKTwTyhc_6Ht2mddEXpdENfl-lRXIVY0CHAlw1yHjI,1264
+pycityagent/environment/sim/clock_service.py,sha256=fgYXacf_-ixhVAn5uKUvqvemBS6A0oQK8JOZukkhXyY,1353
+pycityagent/environment/sim/economy_services.py,sha256=xoc-1_H8JmQwJ24oWRS1fD-hGYtz2I-x6BOkQ4yENzU,7106
+pycityagent/environment/sim/lane_service.py,sha256=N2dUe-3XuqqKLsNXt1k4NN8uV-J_ruo08yhaUd_hwOI,3916
+pycityagent/environment/sim/light_service.py,sha256=KVwt7ii_iLGA7gANZe3n6-4RiiPQt1w9H6ZOoizMI04,4242
+pycityagent/environment/sim/person_service.py,sha256=5r1F2Itn7dKJ2U4hSLovrk5p4qy-2n77MTAv_OlTIwA,10673
+pycityagent/environment/sim/road_service.py,sha256=bKyn3_me0sGmaJVyF6eNeFbdU-9C1yWsa9L7pieDJzg,1285
 pycityagent/environment/sim/sim_env.py,sha256=HI1LcS_FotDKQ6vBnx0e49prXSABOfA20aU9KM-ZkCY,4625
-pycityagent/environment/sim/social_service.py,sha256=6Iqvq6dz8H2jhLLdtaITc6Js9QnQw-Ylsd5AZgUj3-E,1993
+pycityagent/environment/sim/social_service.py,sha256=9EFJAwVdUuUQkNkFRn9qZRDfD1brh2fqkvasnXUEBhQ,2014
 pycityagent/environment/simulator.py,sha256=XjcxbyBIbB3Ht9z087z_oWIPAN6pP5Eq1lyf4W5atb8,12502
 pycityagent/environment/utils/__init__.py,sha256=1m4Q1EfGvNpUsa1bgQzzCyWhfkpElnskNImjjFD3Znc,237
 pycityagent/environment/utils/base64.py,sha256=hoREzQo3FXMN79pqQLO2jgsDEvudciomyKii7MWljAM,374
@@ -29,28 +29,29 @@ pycityagent/environment/utils/geojson.py,sha256=LVHAdEhnZM8d0BoUnuPiIL_gaeXBIIgl
 pycityagent/environment/utils/grpc.py,sha256=6EJwKXXktIWb1NcUiJzIRmfrY0S03QAXXGcCDHqAT00,1998
 pycityagent/environment/utils/map_utils.py,sha256=lYOEoCFFK6-e9N5txLMMq4HUlxMqc8Uw1YrGW5oJmgg,5749
 pycityagent/environment/utils/port.py,sha256=3OM6kSUt3PxvDUOlgyiendBtETaWU8Mzk_8H0TzTmYg,295
-pycityagent/environment/utils/protobuf.py,sha256=0jBvK_s96y_n7tuMbG22TOtQmg71SGV4ONDy2IGsU9o,1148
-pycityagent/llm/__init__.py,sha256=7klKEmCcDWJIu-F4DoAukSuKfDbLhdczrSIhpwow-sY,145
-pycityagent/llm/embedding.py,sha256=2psX_EK67oPlYe77g43EYUYams4M9AiJqxpHTFHG0n8,4253
+pycityagent/environment/utils/protobuf.py,sha256=0BsM_G7x2B_6DMIBHe9bjVuQDOXUytNRQ03g9e05F3c,1170
+pycityagent/llm/__init__.py,sha256=iWs6FLgrbRVIiqOf4ILS89gkVCTvS7HFC3vG-MWuyko,205
+pycityagent/llm/embeddings.py,sha256=Nhf_tUIlaYJAZ93wW2QTCtS1wq7e8fUgdn2JketEAuQ,7600
 pycityagent/llm/llm.py,sha256=vJaaGqVuyV-GlBxrnvGKZnMDlxeTT_sGUTdxz5tYwEE,15141
 pycityagent/llm/llmconfig.py,sha256=4Ylf4OFSBEFy8jrOneeX0HvPhWEaF5jGvy1HkXK08Ro,436
 pycityagent/llm/utils.py,sha256=hoNPhvomb1u6lhFX0GctFipw74hVKb7bvUBDqwBzBYw,160
-pycityagent/memory/__init__.py,sha256=Hs2NhYpIG-lvpwPWwj4DydB1sxtjz7cuA4iDAzCXnjI,243
+pycityagent/memory/__init__.py,sha256=_Vfdo1HcLWsuuz34_i8e91nnLVYADpMlHHSVaB3xgIk,297
 pycityagent/memory/const.py,sha256=6zpJPJXWoH9-yf4RARYYff586agCoud9BRn7sPERB1g,932
-pycityagent/memory/memory.py,sha256=vJxHOI74aJDGZPFu2LbBr02ASfOYpig66fto6Gjr-6Q,18191
-pycityagent/memory/memory_base.py,sha256=euKZRCs4dbcKxjlZzpLCTnH066DAtRjj5g1JFKD40qQ,5633
-pycityagent/memory/profile.py,sha256=s4LnxSPGSjIGZXHXkkd8mMa6uYYZrytgyQdWjcaqGf4,5182
-pycityagent/memory/self_define.py,sha256=poPiexNhOLq_iTgK8s4mK_xoL_DAAcB8kMvInj7iE5E,5179
-pycityagent/memory/state.py,sha256=5W0c1yJ-aaPpE74B2LEcw3Ygpm77tyooHv8NylyrozE,5113
-pycityagent/memory/utils.py,sha256=wLNlNlZ-AY9VB8kbUIy0UQSYh26FOQABbhmKQkit5o8,850
+pycityagent/memory/faiss_query.py,sha256=Z0JS4udyPYCIzHMq464QtHscnswu35gh9fQptikAwkQ,12976
+pycityagent/memory/memory.py,sha256=UBh4yANNHDzYZwrsvyX4ZMSHXINbu1U6g0HLNCOOCk8,17883
+pycityagent/memory/memory_base.py,sha256=QG_j3BxZvkadFEeE3uBR_kjl_xcXD1aHUVs8GEF3d6w,5654
+pycityagent/memory/profile.py,sha256=q8ZS9IBmHCg_X1GONUvXK85P6tCepTKQgXKuvuXYNXw,5203
+pycityagent/memory/self_define.py,sha256=vpZ6CIxR2grNXEIOScdpsSc59FBg0mOKelwQuTElbtQ,5200
+pycityagent/memory/state.py,sha256=TYItiyDtehMEQaSBN7PpNrnNxdDM5jGppr9R9Ufv3kA,5134
+pycityagent/memory/utils.py,sha256=oJWLdPeJy_jcdKcDTo9JAH9kDZhqjoQhhv_zT9qWC0w,877
 pycityagent/message/__init__.py,sha256=TCjazxqb5DVwbTu1fF0sNvaH_EPXVuj2XQ0p6W-QCLU,55
 pycityagent/message/messager.py,sha256=W_OVlNGcreHSBf6v-DrEnfNCXExB78ySr0w26MSncfU,2541
 pycityagent/metrics/__init__.py,sha256=X08PaBbGVAd7_PRGLREXWxaqm7nS82WBQpD1zvQzcqc,128
 pycityagent/metrics/mlflow_client.py,sha256=g_tHxWkWTDijtbGL74-HmiYzWVKb1y8-w12QrY9jL30,4449
 pycityagent/metrics/utils/const.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycityagent/simulation/__init__.py,sha256=P5czbcg2d8S0nbbnsQXFIhwzO4CennAhZM8OmKvAeYw,194
-pycityagent/simulation/agentgroup.py,sha256=5p68wNoEaog4nDym3xsCTporBWmxNiQ1crN3mbOHFsE,19788
-pycityagent/simulation/simulation.py,sha256=7Go_RkpkC_DuBWW21JPqlV2yXY754RqSkqzM0vTdteU,23008
+pycityagent/simulation/agentgroup.py,sha256=r8arCAQkKMhv3yr35XsYJL-MfG6o6rWwHItBmxfDtA4,20589
+pycityagent/simulation/simulation.py,sha256=9kkdgXSEOAN8wiewVFyORksti4IdVNU0opObV6ZYa9k,23344
 pycityagent/simulation/storage/pg.py,sha256=Ws04mUgRcbbvWi_eQm3PXYa6w7AQUbDPWhSU7HFtsD8,6026
 pycityagent/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
 pycityagent/survey/manager.py,sha256=S5IkwTdelsdtZETChRcfCEczzwSrry_Fly9MY4s3rbk,1681
@@ -60,15 +61,15 @@ pycityagent/utils/avro_schema.py,sha256=DHM3bOo8m0dJf8oSwyOWzVeXrH6OERmzA_a5vS4S
 pycityagent/utils/decorators.py,sha256=Gk3r41hfk6awui40tbwpq3C7wC7jHaRmLRlcJFlLQCE,3160
 pycityagent/utils/parsers/__init__.py,sha256=AN2xgiPxszWK4rpX7zrqRsqNwfGF3WnCA5-PFTvbaKk,281
 pycityagent/utils/parsers/code_block_parser.py,sha256=Cs2Z_hm9VfNCpPPll1TwteaJF-HAQPs-3RApsOekFm4,1173
-pycityagent/utils/parsers/json_parser.py,sha256=FZ3XN1g8z4Dr2TFraUOoah1oQcze4fPd2m01hHoX0Mo,2917
+pycityagent/utils/parsers/json_parser.py,sha256=tjwyPluYfkWgsvLi0hzfJwFhO3L6yQfZMKza20HaGrY,2911
 pycityagent/utils/parsers/parser_base.py,sha256=KBKO4zLZPNdGjPAGqIus8LseZ8W3Tlt2y0QxqeCd25Q,1713
 pycityagent/utils/pg_query.py,sha256=h5158xcrxjUTR0nKwAaG1neFfTHPbN5guLmaXpC8yvs,1918
 pycityagent/utils/survey_util.py,sha256=Be9nptmu2JtesFNemPgORh_2GsN7rcDYGQS9Zfvc5OI,2169
 pycityagent/workflow/__init__.py,sha256=QNkUV-9mACMrR8c0cSKna2gC1mMZdxXbxWzjE-Uods0,621
-pycityagent/workflow/block.py,sha256=WkE2On97DCZS_9n8aIgT8wxv9Oaff4Fdf2tLqbKfMtE,6010
+pycityagent/workflow/block.py,sha256=l-z9iJo9_USZQRyj4TLMfihK0-tnNDG0a6jVk9WhG0o,6048
 pycityagent/workflow/prompt.py,sha256=6jI0Rq54JLv3-IXqZLYug62vse10wTI83xvf4ZX42nk,2929
 pycityagent/workflow/tool.py,sha256=xADxhNgVsjNiMxlhdwn3xGUstFOkLEG8P67ez8VmwSI,8555
 pycityagent/workflow/trigger.py,sha256=Df-MOBEDWBbM-v0dFLQLXteLsipymT4n8vqexmK2GiQ,5643
-pycityagent-2.0.0a21.dist-info/METADATA,sha256=sowWsIPV6PFjNPeQI30Pn0J1Fqz5KfZ7sMydvfaOAX0,7848
-pycityagent-2.0.0a21.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-pycityagent-2.0.0a21.dist-info/RECORD,,
+pycityagent-2.0.0a24.dist-info/METADATA,sha256=cHowSJH9VJmum92fAEfRvQYtWmbCJRnVgOmI2JZDlqw,8033
+pycityagent-2.0.0a24.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pycityagent-2.0.0a24.dist-info/RECORD,,