From 6508379d7b2615d8740f7b23ef98d149085992ff Mon Sep 17 00:00:00 2001 From: Saba Date: Sun, 4 Jun 2023 00:57:00 -0700 Subject: [PATCH] Use api_key keyword argument to set the openai_api_key parameter for GPT --- src/khoj/processor/conversation/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/khoj/processor/conversation/utils.py b/src/khoj/processor/conversation/utils.py index 5e3d5ef4..8f0aa2ce 100644 --- a/src/khoj/processor/conversation/utils.py +++ b/src/khoj/processor/conversation/utils.py @@ -41,7 +41,9 @@ max_prompt_size = {"gpt-3.5-turbo": 4096, "gpt-4": 8192} ) def completion_with_backoff(**kwargs): prompt = kwargs.pop("prompt") - if "openai_api_key" not in kwargs: + if "api_key" in kwargs: + kwargs["openai_api_key"] = kwargs.get("api_key") + elif "openai_api_key" not in kwargs: kwargs["openai_api_key"] = os.getenv("OPENAI_API_KEY") llm = OpenAI(**kwargs, request_timeout=10, max_retries=1) return llm(prompt)