pycityagent 2.0.0a18__py3-none-any.whl → 2.0.0a20__py3-none-any.whl

pycityagent/simulation/simulation.py

@@ -3,23 +3,25 @@ import json
 import logging
 import os
 import random
+import time
 import uuid
-from collections.abc import Sequence
+from collections.abc import Callable, Sequence
 from concurrent.futures import ThreadPoolExecutor
 from datetime import datetime, timezone
 from pathlib import Path
-from typing import Any, Callable, Dict, List, Optional, Union
+from typing import Any, Optional, Union
 
 import pycityproto.city.economy.v2.economy_pb2 as economyv2
+import ray
 import yaml
 from mosstool.map._map_util.const import AOI_START_ID
 
-from pycityagent.environment.simulator import Simulator
-from pycityagent.memory.memory import Memory
-from pycityagent.message.messager import Messager
-from pycityagent.survey import Survey
-
 from ..agent import Agent, InstitutionAgent
+from ..environment.simulator import Simulator
+from ..memory.memory import Memory
+from ..message.messager import Messager
+from ..metrics import init_mlflow_connection
+from ..survey import Survey
 from .agentgroup import AgentGroup
 
 logger = logging.getLogger("pycityagent")
@@ -50,15 +52,16 @@ class AgentSimulation:
             self.agent_class = [agent_class]
         self.logging_level = logging_level
         self.config = config
+        self.exp_name = exp_name
         self._simulator = Simulator(config["simulator_request"])
         self.agent_prefix = agent_prefix
-        self._agents: Dict[uuid.UUID, Agent] = {}
-        self._groups: Dict[str, AgentGroup] = {}  # type:ignore
-        self._agent_uuid2group: Dict[uuid.UUID, AgentGroup] = {}  # type:ignore
-        self._agent_uuids: List[uuid.UUID] = []
-        self._user_chat_topics: Dict[uuid.UUID, str] = {}
-        self._user_survey_topics: Dict[uuid.UUID, str] = {}
-        self._user_interview_topics: Dict[uuid.UUID, str] = {}
+        self._agents: dict[uuid.UUID, Agent] = {}
+        self._groups: dict[str, AgentGroup] = {}  # type:ignore
+        self._agent_uuid2group: dict[uuid.UUID, AgentGroup] = {}  # type:ignore
+        self._agent_uuids: list[uuid.UUID] = []
+        self._user_chat_topics: dict[uuid.UUID, str] = {}
+        self._user_survey_topics: dict[uuid.UUID, str] = {}
+        self._user_interview_topics: dict[uuid.UUID, str] = {}
         self._loop = asyncio.get_event_loop()
 
         self._messager = Messager(
@@ -69,18 +72,37 @@ class AgentSimulation:
         )
         asyncio.create_task(self._messager.connect())
 
-        self._enable_avro = config["storage"]["avro"]["enabled"]
+        # storage
+        _storage_config: dict[str, Any] = config.get("storage", {})
+        # avro
+        _avro_config: dict[str, Any] = _storage_config.get("avro", {})
+        self._enable_avro = _avro_config.get("enabled", False)
         if not self._enable_avro:
+            self._avro_path = None
             logger.warning("AVRO is not enabled, NO AVRO LOCAL STORAGE")
-        self._avro_path = Path(config["storage"]["avro"]["path"]) / f"{self.exp_id}"
-        self._avro_path.mkdir(parents=True, exist_ok=True)
-
-        self._enable_pgsql = config["storage"]["pgsql"]["enabled"]
-        self._pgsql_host = config["storage"]["pgsql"]["host"]
-        self._pgsql_port = config["storage"]["pgsql"]["port"]
-        self._pgsql_database = config["storage"]["pgsql"]["database"]
-        self._pgsql_user = config["storage"]["pgsql"]["user"]
-        self._pgsql_password = config["storage"]["pgsql"]["password"]
+        else:
+            self._avro_path = Path(_avro_config["path"]) / f"{self.exp_id}"
+            self._avro_path.mkdir(parents=True, exist_ok=True)
+
+        # pg
+        _pgsql_config: dict[str, Any] = _storage_config.get("pgsql", {})
+        self._enable_pgsql = _pgsql_config.get("enabled", False)
+        if not self._enable_pgsql:
+            logger.warning("PostgreSQL is not enabled, NO POSTGRESQL DATABASE STORAGE")
+            self._pgsql_args = ("", "", "", "", "")
+        else:
+            self._pgsql_host = _pgsql_config["host"]
+            self._pgsql_port = _pgsql_config["port"]
+            self._pgsql_database = _pgsql_config["database"]
+            self._pgsql_user = _pgsql_config.get("user", None)
+            self._pgsql_password = _pgsql_config.get("password", None)
+            self._pgsql_args: tuple[str, str, str, str, str] = (
+                self._pgsql_host,
+                self._pgsql_port,
+                self._pgsql_database,
+                self._pgsql_user,
+                self._pgsql_password,
+            )
 
         # 添加实验信息相关的属性
         self._exp_info = {
@@ -96,14 +118,34 @@ class AgentSimulation:
         }
 
         # 创建异步任务保存实验信息
-        self._exp_info_file = self._avro_path / "experiment_info.yaml"
-        with open(self._exp_info_file, "w") as f:
-            yaml.dump(self._exp_info, f)
+        if self._enable_avro:
+            assert self._avro_path is not None
+            self._exp_info_file = self._avro_path / "experiment_info.yaml"
+            with open(self._exp_info_file, "w") as f:
+                yaml.dump(self._exp_info, f)
+
+    @property
+    def enable_avro(
+        self,
+    ) -> bool:
+        return self._enable_avro
+
+    @property
+    def enable_pgsql(
+        self,
+    ) -> bool:
+        return self._enable_pgsql
 
     @property
-    def agents(self):
+    def agents(self) -> dict[uuid.UUID, Agent]:
         return self._agents
 
+    @property
+    def avro_path(
+        self,
+    ) -> Path:
+        return self._avro_path  # type:ignore
+
     @property
     def groups(self):
         return self._groups
@@ -122,13 +164,26 @@ class AgentSimulation:
         agents: list[Agent],
         config: dict,
         exp_id: str,
+        exp_name: str,
         enable_avro: bool,
         avro_path: Path,
+        enable_pgsql: bool,
+        pgsql_copy_writer: ray.ObjectRef,
+        mlflow_run_id: str = None,  # type: ignore
         logging_level: int = logging.WARNING,
     ):
         """创建远程组"""
         group = AgentGroup.remote(
-            agents, config, exp_id, enable_avro, avro_path, logging_level
+            agents,
+            config,
+            exp_id,
+            exp_name,
+            enable_avro,
+            avro_path,
+            enable_pgsql,
+            pgsql_copy_writer,
+            mlflow_run_id,
+            logging_level,
         )
         return group_name, group, agents
 
@@ -174,7 +229,6 @@ class AgentSimulation:
                     memory_config_func.append(self.default_memory_config_institution)
                 else:
                     memory_config_func.append(self.default_memory_config_citizen)
-
         # 使用线程池并行创建 AgentGroup
         group_creation_params = []
         class_init_index = 0
@@ -218,6 +272,16 @@ class AgentSimulation:
 
             class_init_index += agent_count_i
 
+        # 初始化mlflow连接
+        _mlflow_config = self.config.get("metric_request", {}).get("mlflow")
+        if _mlflow_config:
+            mlflow_run_id, _ = init_mlflow_connection(
+                config=_mlflow_config,
+                mlflow_run_name=f"EXP_{self.exp_name}_{1000*int(time.time())}",
+                experiment_name=self.exp_name,
+            )
+        else:
+            mlflow_run_id = None
         # 收集所有创建组的参数
         creation_tasks = []
         for group_name, agents in group_creation_params:
@@ -226,8 +290,14 @@ class AgentSimulation:
                 agents,
                 self.config,
                 self.exp_id,
-                self._enable_avro,
-                self._avro_path,
+                self.exp_name,
+                self.enable_avro,
+                self.avro_path,
+                self.enable_pgsql,
+                # TODO:
+                # self._pgsql_copy_writer, # type:ignore
+                None,
+                mlflow_run_id,
                 self.logging_level,
             )
             creation_tasks.append((group_name, group, agents))
@@ -393,7 +463,7 @@ class AgentSimulation:
         return EXTRA_ATTRIBUTES, PROFILE, BASE
 
     async def send_survey(
-        self, survey: Survey, agent_uuids: Optional[List[uuid.UUID]] = None
+        self, survey: Survey, agent_uuids: Optional[list[uuid.UUID]] = None
     ):
         """发送问卷"""
         survey_dict = survey.to_dict()
@@ -410,7 +480,7 @@ class AgentSimulation:
             await self._messager.send_message(topic, payload)
 
     async def send_interview_message(
-        self, content: str, agent_uuids: Union[uuid.UUID, List[uuid.UUID]]
+        self, content: str, agent_uuids: Union[uuid.UUID, list[uuid.UUID]]
     ):
         """发送面试消息"""
         payload = {
@@ -438,10 +508,17 @@ class AgentSimulation:
     async def _save_exp_info(self) -> None:
         """异步保存实验信息到YAML文件"""
         try:
-            with open(self._exp_info_file, "w") as f:
-                yaml.dump(self._exp_info, f)
+            if self.enable_avro:
+                with open(self._exp_info_file, "w") as f:
+                    yaml.dump(self._exp_info, f)
         except Exception as e:
-            logger.error(f"保存实验信息失败: {str(e)}")
+            logger.error(f"Avro保存实验信息失败: {str(e)}")
+        try:
+            if self.enable_pgsql:
+                #   TODO
+                pass
+        except Exception as e:
+            logger.error(f"PostgreSQL保存实验信息失败: {str(e)}")
 
     async def _update_exp_status(self, status: int, error: str = "") -> None:
         """更新实验状态并保存"""
@@ -492,7 +569,6 @@ class AgentSimulation:
                     tasks = []
                     for group in self._groups.values():
                         tasks.append(group.run.remote())
-
                     # 等待所有group运行完成
                     await asyncio.gather(*tasks)
 

pycityagent/utils/parsers/parser_base.py

@@ -4,7 +4,7 @@ Base class of parser
 
 import re
 from abc import ABC, abstractmethod
-from typing import Any, Dict, List, Optional, Tuple, Union
+from typing import Any, Union
 
 
 class ParserBase(ABC):

pycityagent/workflow/__init__.py

@@ -4,14 +4,16 @@
 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 .block import (Block, log_and_check, log_and_check_with_memory,
+                    trigger_class)
 from .prompt import FormatPrompt
-from .tool import GetMap, SencePOI, Tool
-from .trigger import MemoryChangeTrigger, TimeTrigger, EventTrigger
+from .tool import ExportMlflowMetrics, GetMap, SencePOI, Tool
+from .trigger import EventTrigger, MemoryChangeTrigger, TimeTrigger
 
 __all__ = [
     "SencePOI",
     "Tool",
+    "ExportMlflowMetrics",
     "GetMap",
     "MemoryChangeTrigger",
     "TimeTrigger",

pycityagent/workflow/block.py

@@ -3,12 +3,11 @@ import functools
 import inspect
 from typing import Any, Callable, Coroutine, Optional, Union
 
-from pycityagent.environment.simulator import Simulator
-from pycityagent.workflow.trigger import EventTrigger
-
+from ..environment.simulator import Simulator
 from ..llm import LLM
 from ..memory import Memory
 from ..utils.decorators import record_call_aio
+from ..workflow.trigger import EventTrigger
 
 TRIGGER_INTERVAL = 1
 

pycityagent/workflow/prompt.py

@@ -1,4 +1,4 @@
-from typing import Any, Callable, Dict, List, Optional, Union
+from typing import Optional, Union
 import re
 
 
@@ -10,7 +10,7 @@ class FormatPrompt:
     Attributes:
         template (str): The template string containing placeholders.
         system_prompt (Optional[str]): An optional system prompt to add to the dialog.
-        variables (List[str]): A list of variable names extracted from the template.
+        variables (list[str]): A list of variable names extracted from the template.
         formatted_string (str): The formatted string derived from the template and provided variables.
     """
 
@@ -27,12 +27,12 @@ class FormatPrompt:
         self.variables = self._extract_variables()
         self.formatted_string = ""  # To store the formatted string
 
-    def _extract_variables(self) -> List[str]:
+    def _extract_variables(self) -> list[str]:
         """
         Extracts variable names from the template string.
 
         Returns:
-            List[str]: A list of variable names found within the template.
+            list[str]: A list of variable names found within the template.
         """
         return re.findall(r"\{(\w+)\}", self.template)
 
@@ -51,12 +51,12 @@ class FormatPrompt:
         )  # Store the formatted string
         return self.formatted_string
 
-    def to_dialog(self) -> List[Dict[str, str]]:
+    def to_dialog(self) -> list[dict[str, str]]:
         """
         Converts the formatted prompt and optional system prompt into a dialog format.
 
         Returns:
-            List[Dict[str, str]]: A list representing the dialog with roles and content.
+            list[dict[str, str]]: A list representing the dialog with roles and content.
         """
         dialog = []
         if self.system_prompt:

pycityagent/workflow/tool.py

@@ -1,7 +1,11 @@
-from typing import Any, Callable, Dict, List, Optional, Union
+import time
+from typing import Any, Optional, Union
+from collections.abc import Callable
+from mlflow.entities import Metric
 
 from ..agent import Agent
-from ..environment import LEVEL_ONE_PRE, POI_TYPE_DICT, AoiService, PersonService
+from ..environment import (LEVEL_ONE_PRE, POI_TYPE_DICT, AoiService,
+                           PersonService)
 from ..workflow import Block
 
 
@@ -72,7 +76,7 @@ class SencePOI(Tool):
     Attributes:
         radius (int): The radius within which to search for POIs.
         category_prefix (str): The prefix for the categories of POIs to consider.
-        variables (List[str]): A list of variables relevant to the tool's operation.
+        variables (list[str]): A list of variables relevant to the tool's operation.
 
     Args:
         radius (int, optional): The circular search radius. Defaults to 100.
@@ -139,7 +143,7 @@ class UpdateWithSimulator(Tool):
         if agent._simulator is None:
             return
         if not agent._has_bound_to_simulator:
-            await agent._bind_to_simulator() # type: ignore
+            await agent._bind_to_simulator()  # type: ignore
         simulator = agent.simulator
         memory = agent.memory
         person_id = await memory.get("id")
@@ -181,3 +185,48 @@ class ResetAgentPosition(Tool):
             lane_id=lane_id,
             s=s,
         )
+
+
+class ExportMlflowMetrics(Tool):
+    def __init__(self, log_batch_size: int = 100) -> None:
+        self._log_batch_size = log_batch_size
+        # TODO:support other log types
+        self.metric_log_cache: list[Metric] = []
+
+    async def __call__(
+        self,
+        metric: Union[Metric, dict],
+        clear_cache: bool = False,
+    ):
+        agent = self.agent
+        batch_size = self._log_batch_size
+        if len(self.metric_log_cache) > batch_size:
+            client = agent.mlflow_client
+            await client.log_batch(
+                metrics=self.metric_log_cache[:batch_size],
+            )
+            self.metric_log_cache = self.metric_log_cache[batch_size:]
+        else:
+            if isinstance(metric, Metric):
+                self.metric_log_cache.append(metric)
+            else:
+                _metric = Metric(
+                    key=metric["key"],
+                    value=metric["value"],
+                    timestamp=metric.get("timestamp", int(1000 * time.time())),
+                    step=metric["step"],
+                )
+                self.metric_log_cache.append(_metric)
+        if clear_cache:
+            await self._clear_cache()
+
+    async def _clear_cache(
+        self,
+    ):
+        agent = self.agent
+        client = agent.mlflow_client
+        if len(self.metric_log_cache) > 0:
+            await client.log_batch(
+                metrics=self.metric_log_cache,
+            )
+            self.metric_log_cache = []

pycityagent/workflow/trigger.py

@@ -1,5 +1,5 @@
 import asyncio
-from typing import Any, Callable, Dict, List, Optional, Union, Type
+from typing import Optional
 import socket
 from ..memory import Memory
 from ..environment import Simulator
@@ -11,7 +11,7 @@ class EventTrigger:
     """Base class for event triggers that wait for specific conditions to be met."""
 
     # 定义该trigger需要的组件类型
-    required_components: List[Type] = []
+    required_components: list[type] = []
 
     def __init__(self, block=None):
         self.block = block

{pycityagent-2.0.0a18.dist-info → pycityagent-2.0.0a20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycityagent
-Version: 2.0.0a18
+Version: 2.0.0a20
 Summary: LLM-based城市环境agent构建库
 License: MIT
 Author: Yuwei Yan
@@ -26,6 +26,7 @@ Requires-Dist: gradio (>=5.7.1,<6.0.0)
 Requires-Dist: grpcio (==1.67.1)
 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)
 Requires-Dist: mosstool (==1.0.24)
 Requires-Dist: networkx (==3.2.1)
 Requires-Dist: numpy (>=1.20.0,<2.0.0)
@@ -33,8 +34,9 @@ Requires-Dist: openai (>=1.58.1,<2.0.0)
 Requires-Dist: pandavro (>=1.8.0,<2.0.0)
 Requires-Dist: poetry (>=1.2.2)
 Requires-Dist: protobuf (<=4.24.0)
+Requires-Dist: psycopg[binary] (>=3.2.3,<4.0.0)
 Requires-Dist: pycitydata (==1.0.0)
-Requires-Dist: pycityproto (>=2.1.4,<3.0.0)
+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)

{pycityagent-2.0.0a18.dist-info → pycityagent-2.0.0a20.dist-info}/RECORD

@@ -1,7 +1,7 @@
 pycityagent/__init__.py,sha256=EDxt3Su3lH1IMh9suNw7GeGL7UrXeWiZTw5KWNznDzc,637
-pycityagent/agent.py,sha256=t9W9sKxtQ0EkMxL78kAjAu-rXigEK6gyLY0IEA4DbnQ,23143
+pycityagent/agent.py,sha256=HRFAG_iM1q3nvXtV0T-Dz01foOtty4IWka7h4WD97CU,24268
 pycityagent/economy/__init__.py,sha256=aonY4WHnx-6EGJ4WKrx4S-2jAkYNLtqUA04jp6q8B7w,75
-pycityagent/economy/econ_client.py,sha256=DE11Ng_NO_foW65A-LxFW0VED-HLrnn4GwUf_Xn-Tlg,11189
+pycityagent/economy/econ_client.py,sha256=GuHK9ZBnhqW3Z7F8ViDJn_iN73yOBbbwFyJv1wLEBDk,12211
 pycityagent/environment/__init__.py,sha256=awHxlOud-btWbk0FCS4RmGJ13W84oVCkbGfcrhKqihA,240
 pycityagent/environment/interact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycityagent/environment/interact/interact.py,sha256=ifxPPzuHeqLHIZ_6zvfXMoBOnBsXNIP4bYp7OJ7pnEQ,6588
@@ -21,23 +21,23 @@ pycityagent/environment/sim/person_service.py,sha256=nIvOsoBoqOTDYtsiThg07-4ZBgk
 pycityagent/environment/sim/road_service.py,sha256=phKTwTyhc_6Ht2mddEXpdENfl-lRXIVY0CHAlw1yHjI,1264
 pycityagent/environment/sim/sim_env.py,sha256=HI1LcS_FotDKQ6vBnx0e49prXSABOfA20aU9KM-ZkCY,4625
 pycityagent/environment/sim/social_service.py,sha256=6Iqvq6dz8H2jhLLdtaITc6Js9QnQw-Ylsd5AZgUj3-E,1993
-pycityagent/environment/simulator.py,sha256=K7IyhiGC9BxanW28bpML4M0YREdMp1h7yMoWBlbf3RY,12504
+pycityagent/environment/simulator.py,sha256=XjcxbyBIbB3Ht9z087z_oWIPAN6pP5Eq1lyf4W5atb8,12502
 pycityagent/environment/utils/__init__.py,sha256=1m4Q1EfGvNpUsa1bgQzzCyWhfkpElnskNImjjFD3Znc,237
 pycityagent/environment/utils/base64.py,sha256=hoREzQo3FXMN79pqQLO2jgsDEvudciomyKii7MWljAM,374
 pycityagent/environment/utils/const.py,sha256=3RMNy7_bE7-23K90j9DFW_tWEzu8s7hSTgKbV-3BFl4,5327
-pycityagent/environment/utils/geojson.py,sha256=Ieg8Bzw63kKhJlhDIOVDoh-wQO4Sbtoe47FtIOy5wWg,686
+pycityagent/environment/utils/geojson.py,sha256=LVHAdEhnZM8d0BoUnuPiIL_gaeXBIIglrLrfje5M0b4,661
 pycityagent/environment/utils/grpc.py,sha256=6EJwKXXktIWb1NcUiJzIRmfrY0S03QAXXGcCDHqAT00,1998
-pycityagent/environment/utils/map_utils.py,sha256=oqrRgQICC3SYw6gwjjPe_MAif7_t6dlrQpY8E32Fexs,5777
+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=Y0xhm_Ny6cawqzlendXb-mAS2QAuuEez1UtTR5-Kb2Q,4293
-pycityagent/llm/llm.py,sha256=BtxBvPK4tb8QlZIfxO5XJ73lKXwF8L31LqVbejWB8eo,15121
+pycityagent/llm/embedding.py,sha256=2psX_EK67oPlYe77g43EYUYams4M9AiJqxpHTFHG0n8,4253
+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/const.py,sha256=6zpJPJXWoH9-yf4RARYYff586agCoud9BRn7sPERB1g,932
-pycityagent/memory/memory.py,sha256=FjKVL_MgNBnSc0sox2tuxLqXg9_MQQr9vYdRDHMdDL4,18183
+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
@@ -45,9 +45,12 @@ pycityagent/memory/state.py,sha256=5W0c1yJ-aaPpE74B2LEcw3Ygpm77tyooHv8NylyrozE,5
 pycityagent/memory/utils.py,sha256=wLNlNlZ-AY9VB8kbUIy0UQSYh26FOQABbhmKQkit5o8,850
 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=jYaqaNpzM5M_e_ykISS_M-mIyYdzJXJWhgpfBpA6l5k,111
-pycityagent/simulation/agentgroup.py,sha256=M19XWJRWyjMAYS0_RIOBQ2C7I1MuVYIaX3DgehGZL2Y,12541
-pycityagent/simulation/simulation.py,sha256=9VH-VaRufUcQcVhqXcQJTMtrquns8pbDpRTfK6736Io,19318
+pycityagent/simulation/agentgroup.py,sha256=6p8-OP2x_syaQ1pWLgll0LHs823OHnappuY-8XzL_LU,13383
+pycityagent/simulation/simulation.py,sha256=pxFdaNMLdJMcpW8gnOcxjfbz4gkpcvWsTR0rNP8rE9k,21675
 pycityagent/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
 pycityagent/survey/manager.py,sha256=S5IkwTdelsdtZETChRcfCEczzwSrry_Fly9MY4s3rbk,1681
 pycityagent/survey/models.py,sha256=YE50UUt5qJ0O_lIUsSY6XFCGUTkJVNu_L1gAhaCJ2fs,3546
@@ -57,13 +60,13 @@ pycityagent/utils/decorators.py,sha256=Gk3r41hfk6awui40tbwpq3C7wC7jHaRmLRlcJFlLQ
 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/parser_base.py,sha256=k6DVqwAMK3jJdOP4IeLE-aFPm3V2F-St5qRBuRdx4aU,1742
+pycityagent/utils/parsers/parser_base.py,sha256=KBKO4zLZPNdGjPAGqIus8LseZ8W3Tlt2y0QxqeCd25Q,1713
 pycityagent/utils/survey_util.py,sha256=Be9nptmu2JtesFNemPgORh_2GsN7rcDYGQS9Zfvc5OI,2169
-pycityagent/workflow/__init__.py,sha256=EyCcjB6LyBim-5iAOPe4m2qfvghEPqu1ZdGfy4KPeZ8,551
-pycityagent/workflow/block.py,sha256=6EmiRMLdOZC1wMlmLMIjfrp9TuiI7Gw4s3nnXVMbrnw,6031
-pycityagent/workflow/prompt.py,sha256=tY69nDO8fgYfF_dOA-iceR8pAhkYmCqoox8uRPqEuGY,2956
-pycityagent/workflow/tool.py,sha256=_bCluIX8HTC8ZW6a-wrMB3Uhx2yzD8sM8XFDI3vd0MM,6642
-pycityagent/workflow/trigger.py,sha256=t5X_i0WtL32bipZSsq_E3UUyYYudYLxQUpvxbgClp2s,5683
-pycityagent-2.0.0a18.dist-info/METADATA,sha256=iajlG1hMfjQA2qBuTYITWemGBg4TEPU8hHpSU1l0MDs,7760
-pycityagent-2.0.0a18.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-pycityagent-2.0.0a18.dist-info/RECORD,,
+pycityagent/workflow/__init__.py,sha256=QNkUV-9mACMrR8c0cSKna2gC1mMZdxXbxWzjE-Uods0,621
+pycityagent/workflow/block.py,sha256=WkE2On97DCZS_9n8aIgT8wxv9Oaff4Fdf2tLqbKfMtE,6010
+pycityagent/workflow/prompt.py,sha256=6jI0Rq54JLv3-IXqZLYug62vse10wTI83xvf4ZX42nk,2929
+pycityagent/workflow/tool.py,sha256=SGY18lT71hBLKagopirFbxRjPY_387Dobo9SUwjHIn0,8215
+pycityagent/workflow/trigger.py,sha256=Df-MOBEDWBbM-v0dFLQLXteLsipymT4n8vqexmK2GiQ,5643
+pycityagent-2.0.0a20.dist-info/METADATA,sha256=aygfImVFM4jG0nXVuunu5LZDNi4JyikBULFce0WGWeM,7848
+pycityagent-2.0.0a20.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pycityagent-2.0.0a20.dist-info/RECORD,,