From 9b6b7bb0802168cdbaaec2369aaac807da511fe0 Mon Sep 17 00:00:00 2001
From: Sverre Nystad <sverrenystad@gmail.com>
Date: Thu, 21 Sep 2023 00:32:54 +0200
Subject: [PATCH] Refactor: Make referee a part of outcome engine folder

---
 src/agents/dungeon_master.py        | 4 ++--
 src/{ => outcome_engine}/referee.py | 0
 tests/test_referee.py               | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)
 rename src/{ => outcome_engine}/referee.py (100%)

diff --git a/src/agents/dungeon_master.py b/src/agents/dungeon_master.py
index 261b9fd..6a87c03 100644
--- 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 b/src/outcome_engine/referee.py
similarity index 100%
rename from src/referee.py
rename to src/outcome_engine/referee.py
diff --git a/tests/test_referee.py b/tests/test_referee.py
index cb49347..c0a149b 100644
--- 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():