pycityagent 2.0.0a38__py3-none-any.whl → 2.0.0a40__py3-none-any.whl

pycityagent/simulation/agentgroup.py

@@ -199,11 +199,40 @@ class AgentGroup:
                     profile = profile[0]
                     profile["id"] = agent._uuid
                     profiles.append(
-                        (agent._uuid, profile.get("name", ""), json.dumps(profile))
+                        (
+                            agent._uuid,
+                            profile.get("name", ""),
+                            json.dumps(
+                                {
+                                    k: v
+                                    for k, v in profile.items()
+                                    if k not in {"id", "name"}
+                                }
+                            ),
+                        )
                     )
-                await self._pgsql_writer.async_write_profile.remote(  # type:ignore
-                    profiles
-                )
+            else:
+                profiles: list[Any] = []
+                for agent in self.agents:
+                    profile = await agent.memory._profile.export()
+                    profile = profile[0]
+                    profile["id"] = agent._uuid
+                    profiles.append(
+                        (
+                            agent._uuid,
+                            profile.get("name", ""),
+                            json.dumps(
+                                {
+                                    k: v
+                                    for k, v in profile.items()
+                                    if k not in {"id", "name"}
+                                }
+                            ),
+                        )
+                    )
+            await self._pgsql_writer.async_write_profile.remote(  # type:ignore
+                profiles
+            )
         self.initialized = True
         logger.debug(f"-----AgentGroup {self._uuid} initialized")
 

pycityagent/simulation/storage/pg.py

@@ -139,7 +139,12 @@ class PgWriter:
                 exec_str = "SELECT * FROM {table_name} WHERE id=%s".format(
                     table_name=table_name
                 ), (self.exp_id,)
-                await cur.execute(exec_str)  # type:ignore
+                await cur.execute(
+                    "SELECT * FROM {table_name} WHERE id=%s".format(
+                        table_name=table_name
+                    ),
+                    (self.exp_id,),
+                )  # type:ignore
                 logger.debug(f"table:{table_name} sql: {exec_str}")
                 record_exists = await cur.fetchall()
                 if record_exists:

{pycityagent-2.0.0a38.dist-info → pycityagent-2.0.0a40.dist-info}/METADATA

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

{pycityagent-2.0.0a38.dist-info → pycityagent-2.0.0a40.dist-info}/RECORD

@@ -50,9 +50,9 @@ 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=e6K388TzfwBDyZmfPAxLl_Z1eKcyEn3qgvSlbyw-ggI,24158
+pycityagent/simulation/agentgroup.py,sha256=ZPzgHgin-XOz3owxMbyAxcz8AiU4I6XM-ppOUsFsAIE,25284
 pycityagent/simulation/simulation.py,sha256=aE-U9wEay9ksheSOotxyLfZOGxlKxP0OFOuRt3mDmSY,23246
-pycityagent/simulation/storage/pg.py,sha256=UuBO0J7e1Oi2LsWygIiSHQ9W-RQO4X7Vpy9Ci0aMP8s,8272
+pycityagent/simulation/storage/pg.py,sha256=jjdYvqKmDms3weqALqOO__gxWF-Z0YcqeD85XbP4Qks,8455
 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
@@ -70,6 +70,6 @@ pycityagent/workflow/block.py,sha256=C2aWdVRffb3LknP955GvPcBMsm3VPXN9ZuAtCgITFTo
 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.0a38.dist-info/METADATA,sha256=HUMRqZM9uvEGwXX9uD0ELpw5HkB6MwgZaKHmJmlxSos,8046
-pycityagent-2.0.0a38.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-pycityagent-2.0.0a38.dist-info/RECORD,,
+pycityagent-2.0.0a40.dist-info/METADATA,sha256=ZgQbu-AdFrvCiiQJ0Cgs9dILTtHOIqs5VEMfbQ1SFn0,8046
+pycityagent-2.0.0a40.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+pycityagent-2.0.0a40.dist-info/RECORD,,