don't validate openai key in debug mode

junison17 · Sep 11, 2023 · 54bb4c7 · 54bb4c7
1 parent 249679a
commit 54bb4c7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/knowledge_gpt/main.py b/knowledge_gpt/main.py
@@ -22,7 +22,7 @@
 MODEL = "openai"
 
 # For testing
-# EMBEDDING, VECTOR_STORE, MODEL = ["debug"] * 3
+EMBEDDING, VECTOR_STORE, MODEL = ["debug"] * 3
 
 st.set_page_config(page_title="KnowledgeGPT", page_icon="📖", layout="wide")
 st.header("📖KnowledgeGPT")
@@ -61,7 +61,7 @@
 if not is_file_valid(file):
     st.stop()
 
-if not is_open_ai_key_valid(openai_api_key):
+if MODEL != "debug" and not is_open_ai_key_valid(openai_api_key):
     st.stop()