set MARVIN_OPENAI_API_KEY separately in environment

mt7180 · Mar 28, 2024 · 6991fb4 · 6991fb4
1 parent c8ffff9
commit 6991fb4
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: