pycityagent 2.0.0a65__cp310-cp310-macosx_11_0_arm64.whl → 2.0.0a67__cp310-cp310-macosx_11_0_arm64.whl

pycityagent/cityagent/societyagent.py

@@ -62,6 +62,7 @@ class PlanAndActionBlock(Block):
             llm=llm, memory=memory, simulator=simulator, economy_client=economy_client
         )
         self.otherBlock = OtherBlock(llm=llm, memory=memory)
+
     async def plan_generation(self):
         """Generate plan"""
         current_plan = await self.memory.status.get("current_plan")
@@ -151,8 +152,10 @@ class PlanAndActionBlock(Block):
         # step execution
         await self.step_execution()
 
+
 class MindBlock(Block):
     """Cognition workflow"""
+
     cognitionBlock: CognitionBlock
 
     def __init__(self, llm: LLM, memory: Memory, simulator: Simulator):
@@ -164,11 +167,12 @@ class MindBlock(Block):
     async def forward(self):
         await self.cognitionBlock.forward()
 
+
 class SocietyAgent(CitizenAgent):
     update_with_sim = UpdateWithSimulator()
     mindBlock: MindBlock
     planAndActionBlock: PlanAndActionBlock
-    update_with_sim = UpdateWithSimulator()
+    
     configurable_fields = [
         "enable_cognition",
         "enable_mobility",
@@ -234,9 +238,10 @@ class SocietyAgent(CitizenAgent):
         await self.update_with_sim()
 
         # check last step
-        if not await self.check_and_update_step():
+        ifpass = await self.check_and_update_step()
+        if not ifpass:
             return
-        
+
         await self.planAndActionBlock.forward()
 
         if self.enable_cognition:
@@ -361,7 +366,7 @@ class SocietyAgent(CitizenAgent):
 
                 if not content:
                     return ""
-                
+
                 # 添加记忆
                 description = f"You received a social message: {content}"
                 await self.memory.stream.add_social(description=description)

pycityagent/cli/wrapper.py

@@ -1,7 +1,7 @@
 import os
-import sys
-import subprocess
 import signal
+import subprocess
+import sys
 
 _script_dir = os.path.dirname(os.path.abspath(__file__))
 _parent_dir = os.path.dirname(_script_dir)