From 6991fb45edbbe9bb81eaecbbdae9456cb4208550 Mon Sep 17 00:00:00 2001 From: Mira Date: Thu, 28 Mar 2024 19:03:45 +0100 Subject: [PATCH] set MARVIN_OPENAI_API_KEY separately in environment --- backend/script_RAG.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/backend/script_RAG.py b/backend/script_RAG.py index f528108..4551a9d 100644 --- 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: