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

pycityagent/agent/agent_base.py

@@ -245,12 +245,12 @@ class Agent(ABC):
             config = json.load(f)
             return cls.import_block_config(config)
 
-    def load_from_config(self, config: dict[str, list[dict]]) -> None:
+    def load_from_config(self, config: dict[str, Any]) -> None:
         """
         Update the current Agent instance's Block hierarchy using the provided configuration.
 
         - **Args**:
-            - `config` (`dict[str, list[dict]]`): A dictionary containing the configuration for updating the agent and its blocks.
+            - `config` (`dict[str, Any]`): A dictionary containing the configuration for updating the agent and its blocks.
 
         - **Returns**:
             - `None`

pycityagent/simulation/agentgroup.py

@@ -47,7 +47,7 @@ class AgentGroup:
         mlflow_run_id: str,
         embedding_model: Embeddings,
         logging_level: int,
-        agent_config_file: Optional[dict[type[Agent], str]] = None,
+        agent_config_file: Optional[dict[type[Agent], Any]] = None,
         llm_semaphore: int = 200,
         environment: Optional[dict] = None,
     ):
@@ -76,7 +76,7 @@ class AgentGroup:
             - `mlflow_run_id` (str): Run identifier for MLflow tracking.
             - `embedding_model` (Embeddings): Model used for generating embeddings.
             - `logging_level` (int): Logging level for the agent group.
-            - `agent_config_file` (Optional[Dict[Type[Agent], str]], optional): File paths for loading agent configurations. Defaults to None.
+            - `agent_config_file` (Optional[Dict[Type[Agent], Any]], optional): File paths for loading agent configurations. Defaults to None.
             - `environment` (Optional[Dict[str, str]], optional): Environment variables for the simulator. Defaults to None.
         """
         logger.setLevel(logging_level)
@@ -105,8 +105,6 @@ class AgentGroup:
                 "status": self.avro_path / f"status.avro",
                 "survey": self.avro_path / f"survey.avro",
             }
-        if self.enable_pgsql:
-            pass
         # Mlflow
         metric_config = config.prop_metric_request
         if metric_config is not None and metric_config.mlflow is not None:
@@ -147,7 +145,7 @@ class AgentGroup:
         # prepare Simulator
         logger.info(f"-----Initializing Simulator in AgentGroup {self._uuid} ...")
         self.simulator = Simulator(config)
-        self.simulator.set_environment(environment)
+        self.simulator.set_environment(environment if environment else {})
         self.simulator.set_map(map_ref)
         self.projector = pyproj.Proj(
             ray.get(self.simulator.map.get_projector.remote())  # type:ignore

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pycityagent
-Version: 2.0.0a97
+Version: 2.0.0a98
 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.0a97.dist-info → pycityagent-2.0.0a98.dist-info}/RECORD

@@ -1,9 +1,3 @@
-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
@@ -26,7 +20,7 @@ pycityagent/memory/faiss_query.py,sha256=KPeyzIjD0dzkxr-TlOeCiIlkdh1WAyyipCAXUEt
 pycityagent/memory/state.py,sha256=JFCBuK1AS-HqscvdGS9ATF9AUU8t29_2leDY_6iO2_4,5158
 pycityagent/simulation/simulation.py,sha256=YPRzB7wCL-K85W3hgvBWukefyEgZXZCi4egH6rZa6XA,47099
 pycityagent/simulation/__init__.py,sha256=u1WpgwVxPboXWMxrUQKMXJNmTKQYAeCaqZv9HmSxESY,118
-pycityagent/simulation/agentgroup.py,sha256=DYx68TckA5ox9x_GHID-3Gg19qABY-NR9QAO2BKKfMc,36814
+pycityagent/simulation/agentgroup.py,sha256=lCz1Ruuz1TtuwzCnRVSPROey7FaXObiQmvNpv3lnzZw,36790
 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
@@ -41,7 +35,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=Fmp9wYwiDskEQxslweO3Jc-1OwbB9vgjv1VfZZIHSGI,39252
+pycityagent/agent/agent_base.py,sha256=jYatA3gxdJTRRqvtcKBABD7NywwhlV18qUDnUiw8qRM,39238
 pycityagent/agent/__init__.py,sha256=U20yKu9QwSqAx_PHk5JwipfODkDfxONtumVfnsKjWFg,180
 pycityagent/agent/agent.py,sha256=zJs09Tzz53av59SzmrlJj6Yd-XId6voCYaxguLbbq3s,18991
 pycityagent/cli/wrapper.py,sha256=bAwmfWTXbGld-uiMnIMhKp22IVRQjhLUToknyUDk9rQ,1147
@@ -99,3 +93,9 @@ pycityagent/cityagent/blocks/mobility_block.py,sha256=bUoExMdxlMzbnWHe1hx6bkR2Jj
 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.0a98.dist-info/RECORD,,
+pycityagent-2.0.0a98.dist-info/LICENSE,sha256=n2HPXiupinpyHMnIkbCf3OTYd3KMqbmldu1e7av0CAU,1084
+pycityagent-2.0.0a98.dist-info/WHEEL,sha256=md3JO_ifs5j508p3TDNMgtQVtnQblpGEt_Wo4W56l8Y,107
+pycityagent-2.0.0a98.dist-info/entry_points.txt,sha256=BZcne49AAIFv-hawxGnPbblea7X3MtAtoPyDX8L4OC4,132
+pycityagent-2.0.0a98.dist-info/top_level.txt,sha256=yOmeu6cSXmiUtScu53a3s0p7BGtLMaV0aff83EHCTic,43
+pycityagent-2.0.0a98.dist-info/METADATA,sha256=n8hlDfjXb8yJTLVUOsIdUc2XpEYSc8yyO8lfJ3aL8XU,8657