Merge pull request #45 from TheDataGuild/feature/itune-ready-production

Feature/itune ready production
TheDataGuild · Oct 13, 2023 · 209a623 · 209a623
2 parents e5a2edc + 4a1d628
commit 209a623
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/itune_strategy.pkl b/itune_strategy.pkl
diff --git a/mind_palace/app.py b/mind_palace/app.py
@@ -1,3 +1,5 @@
+import logging
+
 import extract
 import index
 import openai
@@ -9,7 +11,19 @@
 openai.api_key = st.secrets.openai_key
 xml_dir = "./resources/xmls/dennis-oct-10/"
 gpt_model = "gpt-3.5-turbo"
-itune = Tune(strategy=MultiArmedBandit())
+
+
+def is_production_env():
+    return "app_env" in st.secrets.keys() and st.secrets.app_env == "production"
+
+
+if not is_production_env():
+    logging.basicConfig(level=logging.INFO)
+
+itune = Tune(
+    strategy=MultiArmedBandit(),
+    only_choose_winning_params=is_production_env(),
+)
 
 itune.load()
 

diff --git a/requirements/build.txt b/requirements/build.txt
@@ -1,4 +1,4 @@
-itune==0.1.1
+itune==0.2.0
 pypdf==3.16.0
 grobid-tei-xml==0.1.3
 nltk==3.8.1