pycityagent 2.0.0a31__py3-none-any.whl → 2.0.0a32__py3-none-any.whl

pycityagent/simulation/agentgroup.py

@@ -9,6 +9,7 @@ from typing import Any
 from uuid import UUID
 
 import fastavro
+import pyproj
 import ray
 from langchain_core.embeddings import Embeddings
 
@@ -80,6 +81,7 @@ class AgentGroup:
         # Step:2 prepare Simulator
         logger.info(f"-----Creating Simulator in AgentGroup {self._uuid} ...")
         self.simulator = Simulator(config["simulator_request"])
+        self.projector = pyproj.Proj(self.simulator.map.header["projection"])
 
         # Step:3 prepare Economy client
         if "economy" in config["simulator_request"]:
@@ -264,8 +266,9 @@ class AgentGroup:
                 for agent in self.agents:
                     _date_time = datetime.now(timezone.utc)
                     position = await agent.memory.get("position")
-                    lng = position["longlat_position"]["longitude"]
-                    lat = position["longlat_position"]["latitude"]
+                    x = position["xy_position"]["x"]
+                    y = position["xy_position"]["y"]
+                    lng, lat = self.projector(x, y, inverse=True)
                     if "aoi_position" in position:
                         parent_id = position["aoi_position"]["aoi_id"]
                     elif "lane_position" in position:
@@ -379,8 +382,9 @@ class AgentGroup:
                     for agent in self.agents:
                         _date_time = datetime.now(timezone.utc)
                         position = await agent.memory.get("position")
-                        lng = position["longlat_position"]["longitude"]
-                        lat = position["longlat_position"]["latitude"]
+                        x = position["xy_position"]["x"]
+                        y = position["xy_position"]["y"]
+                        lng, lat = self.projector(x, y, inverse=True)
                         if "aoi_position" in position:
                             parent_id = position["aoi_position"]["aoi_id"]
                         elif "lane_position" in position:
@@ -409,6 +413,17 @@ class AgentGroup:
                 else:
                     for agent in self.agents:
                         _date_time = datetime.now(timezone.utc)
+                        position = await agent.memory.get("position")
+                        x = position["xy_position"]["x"]
+                        y = position["xy_position"]["y"]
+                        lng, lat = self.projector(x, y, inverse=True)
+                        if "aoi_position" in position:
+                            parent_id = position["aoi_position"]["aoi_id"]
+                        elif "lane_position" in position:
+                            parent_id = position["lane_position"]["lane_id"]
+                        else:
+                            # BUG: 需要处理
+                            parent_id = -1
                         try:
                             nominal_gdp = await agent.memory.get("nominal_gdp")
                         except:
@@ -457,9 +472,9 @@ class AgentGroup:
                             "id": agent._uuid,
                             "day": await self.simulator.get_simulator_day(),
                             "t": await self.simulator.get_simulator_second_from_start_of_day(),
-                            "lng": -1,
-                            "lat": -1,
-                            "parent_id": -1,
+                            "lng": lng,
+                            "lat": lat,
+                            "parent_id": parent_id,
                             "action": "",
                             "type": await agent.memory.get("type"),
                             "nominal_gdp": nominal_gdp,

{pycityagent-2.0.0a31.dist-info → pycityagent-2.0.0a32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycityagent
-Version: 2.0.0a31
+Version: 2.0.0a32
 Summary: LLM-based城市环境agent构建库
 License: MIT
 Author: Yuwei Yan

{pycityagent-2.0.0a31.dist-info → pycityagent-2.0.0a32.dist-info}/RECORD

@@ -50,7 +50,7 @@ pycityagent/metrics/__init__.py,sha256=X08PaBbGVAd7_PRGLREXWxaqm7nS82WBQpD1zvQzc
 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=dyUnl76ZlO9ZDwCVtg6hB98dkXSurUsD0ADJyGIlP_I,23587
+pycityagent/simulation/agentgroup.py,sha256=ARqycTmVR8kLmLojhTqP6GU1JCjjnwQ_T5SOb5-Spew,24406
 pycityagent/simulation/simulation.py,sha256=9kkdgXSEOAN8wiewVFyORksti4IdVNU0opObV6ZYa9k,23344
 pycityagent/simulation/storage/pg.py,sha256=qGrYzJIAzjv8-d3-cle0rY0AN6XB6MgnHkFLBoLmKWU,7251
 pycityagent/survey/__init__.py,sha256=rxwou8U9KeFSP7rMzXtmtp2fVFZxK4Trzi-psx9LPIs,153
@@ -70,6 +70,6 @@ pycityagent/workflow/block.py,sha256=l-z9iJo9_USZQRyj4TLMfihK0-tnNDG0a6jVk9WhG0o
 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.0a31.dist-info/METADATA,sha256=T-AOPv84-eFtT35uA3WvqIpCkQGoCuxR-Tl_zfSqefI,8033
-pycityagent-2.0.0a31.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-pycityagent-2.0.0a31.dist-info/RECORD,,
+pycityagent-2.0.0a32.dist-info/METADATA,sha256=V94OUuB_US8D7UljUrG8MHFqx27PUDQ2qMIX_XYNJSg,8033
+pycityagent-2.0.0a32.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pycityagent-2.0.0a32.dist-info/RECORD,,