chore: Update default model and message formatting in demo

Farmerobot · Nov 7, 2024 · 2d9d338 · 2d9d338
1 parent 51d0209
commit 2d9d338
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/llm_postor/demo_gui.py b/src/llm_postor/demo_gui.py
@@ -17,8 +17,8 @@ def load_game_engine():
     game_engine = GameEngine()
 
     # Only for new game. Saved games have llm_model selected in the game state
-    model_name = "meta-llama/llama-3.1-8b-instruct"
-    player_model_name = "meta-llama/llama-3.1-8b-instruct"
+    model_name = "meta-llama/llama-3.2-3b-instruct"
+    player_model_name = "meta-llama/llama-3.2-3b-instruct"
     # model_name = "google/gemini-flash-1.5-exp"
 
     game_engine.init_game()

diff --git a/src/llm_postor/game/agents/discussion_agent.py b/src/llm_postor/game/agents/discussion_agent.py
@@ -73,7 +73,7 @@ def respond_to_statements(self, points: str) -> str:
             player_name=self.player_name,
             player_role=self.role,
             points=points,
-            messages="\n".join(self.messages)
+            messages=self.messages
         ))
         response = self.llm.invoke([system_message, user_message])
         self.add_token_usage(response.usage_metadata)