Merge pull request #57 from mt7180/56-fix-marvin_openai_api_key-error

set MARVIN_OPENAI_API_KEY separately in environment
mt7180 · Mar 28, 2024 · de6beec · de6beec
2 parents c8ffff9 + 6991fb4
commit de6beec
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/backend/script_RAG.py b/backend/script_RAG.py
@@ -39,7 +39,9 @@
 from .document_categories import CATEGORY_LABELS
 from .models import QuestionModel
 
-marvin_settings.openai.api_key = os.getenv("OPENAI_API_KEY")
+if openai_api_key := os.getenv("OPENAI_API_KEY"):
+    marvin_settings.openai.api_key = os.getenv("OPENAI_API_KEY")
+    os.environ["MARVIN_OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")  # type: ignore
 
 
 class AITextDocument: