Merge pull request #11 from NREL/debug_cox

debug_cox
NREL · Dec 14, 2023 · 323066e · 323066e
2 parents f05e0fc + a6d7386
commit 323066e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/elm/wizard.py b/elm/wizard.py
@@ -38,6 +38,7 @@ def __init__(self, corpus, model=None, token_budget=3500, ref_col=None):
         """
 
         super().__init__(model)
+
         self.corpus = self.preflight_corpus(corpus)
         self.token_budget = token_budget
         self.embedding_arr = np.vstack(self.corpus['embedding'].values)

diff --git a/examples/energy_wizard/retrieve_docs.py b/examples/energy_wizard/retrieve_docs.py
@@ -20,7 +20,7 @@
 # NREL-Azure endpoint. You can also use just the openai endpoint.
 # NOTE: embedding values are different between OpenAI and Azure models!
 openai.api_base = os.getenv("AZURE_OPENAI_ENDPOINT")
-openai.api_key = os.getenv("AZURE_OPENAI_API_KEY")
+openai.api_key = os.getenv("AZURE_OPENAI_KEY")
 openai.api_type = 'azure'
 openai.api_version = '2023-03-15-preview'
 

diff --git a/examples/energy_wizard/run_app.py b/examples/energy_wizard/run_app.py
@@ -4,6 +4,7 @@
 import openai
 from glob import glob
 import pandas as pd
+import sys
 
 from elm import EnergyWizard
 
@@ -61,7 +62,16 @@ def get_corpus():
 @st.cache_resource
 def get_wizard():
     """Get the energy wizard object."""
-    corpus = get_corpus()
+
+    # Getting Corpus of data. If no corpus throw error for user.
+    try:
+        corpus = get_corpus()
+    except Exception:
+        print("Error: Have you run 'retrieve_docs.py'?")
+        st.header("Error")
+        st.write("Error: Have you run 'retrieve_docs.py'?")
+        sys.exit(0)
+
     wizard = EnergyWizard(corpus, ref_col='ref', model=model)
     return wizard