Fix agent relationship bug. (#21)

simulatrex · Nov 29, 2023 · c356029 · c356029
1 parent e125fe5
commit c356029
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name="simulatrex",
-    version="0.0.4.0",
+    version="0.0.4.1",
     author="Dominik Scherm",
     author_email="me@dominikscherm.de",
     description="LLM-based simulation framework",

diff --git a/src/simulatrex/agent.py b/src/simulatrex/agent.py
@@ -191,19 +191,22 @@ async def _decide_on_converse(
         for thought in last_thoughts:
             agent_thoughts.append(thought.content)
 
-        agent_relationships: List[AgentRelationship] = []
+        agent_relationships: List[str] = []
         for relationship in self.relationships:
             agent_relationships.append(relationship.summary())
 
         if agent_relationships:
+            _agent_relationships_summary = "; ".join(
+                [str(r) for r in agent_relationships]
+            )
             _logger.info(
-                f"Relationships found for the agent: {[r.summary() for r in agent_relationships]}"
+                f"Relationships found for the agent: {_agent_relationships_summary}"
             )
             prompt = PromptManager().get_filled_template(
                 TemplateType.AGENT_DECIDE_ON_CONVERSATION,
                 agent_name=self.identity.name,
                 agent_thoughts=agent_thoughts,
-                agent_relationships=agent_relationships,
+                agent_relationships=_agent_relationships_summary,
                 environment=environment,
             )