pycityagent 2.0.0a16__tar.gz → 2.0.0a18__tar.gz

{pycityagent-2.0.0a16 → pycityagent-2.0.0a18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pycityagent
-Version: 2.0.0a16
+Version: 2.0.0a18
 Summary: LLM-based城市环境agent构建库
 License: MIT
 Author: Yuwei Yan
@@ -34,7 +34,7 @@ Requires-Dist: pandavro (>=1.8.0,<2.0.0)
 Requires-Dist: poetry (>=1.2.2)
 Requires-Dist: protobuf (<=4.24.0)
 Requires-Dist: pycitydata (==1.0.0)
-Requires-Dist: pycityproto (>=2.1.3,<3.0.0)
+Requires-Dist: pycityproto (>=2.1.4,<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.0a16 → pycityagent-2.0.0a18}/pycityagent/economy/econ_client.py

@@ -307,3 +307,12 @@ class EconomyClient:
         )
         # current agent ids and org ids
         return (list(response.agent_ids), list(response.org_ids))
+
+    async def get_org_entity_ids(self, org_type: economyv2.OrgType)->list[int]:
+        request = org_service.GetOrgEntityIdsRequest(
+            type=org_type,
+        )
+        response: org_service.GetOrgEntityIdsResponse = (
+            await self._aio_stub.GetOrgEntityIds(request)
+        )
+        return list(response.org_ids)

{pycityagent-2.0.0a16 → pycityagent-2.0.0a18}/pycityagent/simulation/simulation.py

@@ -488,12 +488,13 @@ class AgentSimulation:
             monitor_task = asyncio.create_task(self._monitor_exp_status(stop_event))
 
             try:
-                tasks = []
-                for group in self._groups.values():
-                    tasks.append(group.run.remote())
+                for _ in range(day):
+                    tasks = []
+                    for group in self._groups.values():
+                        tasks.append(group.run.remote())
 
-                # 等待所有group运行完成
-                await asyncio.gather(*tasks)
+                    # 等待所有group运行完成
+                    await asyncio.gather(*tasks)
 
             finally:
                 # 设置停止事件

{pycityagent-2.0.0a16 → pycityagent-2.0.0a18}/pyproject.toml

@@ -3,7 +3,7 @@ in-project = true
 
 [tool.poetry]
 name = "pycityagent"
-version = "2.0.0a16"
+version = "2.0.0a18"
 description = "LLM-based城市环境agent构建库"
 authors = ["Yuwei Yan <pinkgranite86@gmail.com>","Junbo Yan <yanjb20thu@gmali.com>"]
 license = "MIT"
@@ -32,7 +32,7 @@ openai = "^1.58.1"
 Pillow = "11.0.0"
 protobuf = "<=4.24.0,<5.0.0"
 pycitydata = "1.0.0"
-pycityproto = "^2.1.3"
+pycityproto = "^2.1.4"
 Requests = "2.32.3"
 Shapely = "2.0.6"
 sidecar = "0.7.0"