pycityagent 2.0.0a38__tar.gz → 2.0.0a39__tar.gz

{pycityagent-2.0.0a38 → pycityagent-2.0.0a39}/PKG-INFO

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

{pycityagent-2.0.0a38 → pycityagent-2.0.0a39}/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-2.0.0a38 → pycityagent-2.0.0a39}/pyproject.toml

@@ -3,7 +3,7 @@ in-project = true
 
 [tool.poetry]
 name = "pycityagent"
-version = "2.0.0a38"
+version = "2.0.0a39"
 description = "LLM-based城市环境agent构建库"
 authors = ["Yuwei Yan <pinkgranite86@gmail.com>","Junbo Yan <yanjb20thu@gmali.com>"]
 license = "MIT"