sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2025-02-17 08:04:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use api_key keyword argument to set the openai_api_key parameter for GPT

Browse source
This commit is contained in:
Debanjum 2023-06-04 15:05:34 +05:30 • committed by GitHub
commit f6ceb22373
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/khoj/processor/conversation/utils.py
View file

@ -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")
else:
kwargs["openai_api_key"] = os.getenv("OPENAI_API_KEY")
llm = OpenAI(**kwargs, request_timeout=10, max_retries=1)
return llm(prompt)