Refactor: Make referee a part of outcome engine folder

SverreNystad · Sep 20, 2023 · 9b6b7bb · 9b6b7bb
1 parent f1f4975
commit 9b6b7bb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/agents/dungeon_master.py b/src/agents/dungeon_master.py
@@ -5,7 +5,7 @@
 from langchain.agents import initialize_agent
 import logging
 from src.npc_generation import generate_npc
-from src.referee import decide_difficulty
+from src.outcome_engine.referee import decide_difficulty
 from src.text_generation.text_generator import get_default_text_generator
 
 # Set up logging
@@ -67,7 +67,7 @@ def narrate(prompt: str) -> str:
     agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION, 
     verbose=False, 
     memory=memory,
-    max_iterations=1,
+    max_iterations=2,
     )
 
 def run_dungeon_master(prompt) -> str:

diff --git a/src/referee.py → src/outcome_engine/referee.py b/src/referee.py → src/outcome_engine/referee.py
diff --git a/tests/test_referee.py b/tests/test_referee.py
@@ -1,5 +1,5 @@
 import pytest
-from src.referee import Difficulty, decide_difficulty
+from src.outcome_engine.referee import Difficulty, decide_difficulty
 
 @pytest.fixture
 def TRIVIAL_CONTEXT():