sij/khoj
1
0
Fork 0
mirror of https://github.com/khoj-ai/khoj.git synced 2024-11-27 09:25:06 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

No need to set OpenAI API key from environment variable explicitly

Browse source 
It is unnecessary as the OpenAI client automatically tries to use API
key from OPENAI_API_KEY env var when the api_key field is unset
This commit is contained in:
Debanjum Singh Solanky 2024-06-22 12:18:43 +05:30
parent a038e4911b
commit 553beae848
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
src/khoj/processor/conversation/openai/utils.py
View file

@ -1,5 +1,4 @@
import logging
import os
from threading import Thread
from typing import Dict
@ -40,7 +39,7 @@ def completion_with_backoff(
client: openai.OpenAI = openai_clients.get(client_key)
if not client:
client = openai.OpenAI(
api_key=openai_api_key or os.getenv("OPENAI_API_KEY"),
api_key=openai_api_key,
base_url=api_base_url,
)
openai_clients[client_key] = client
@ -102,7 +101,7 @@ def llm_thread(g, messages, model_name, temperature, openai_api_key=None, api_ba
client_key = f"{openai_api_key}--{api_base_url}"
if client_key not in openai_clients:
client: openai.OpenAI = openai.OpenAI(
api_key=openai_api_key or os.getenv("OPENAI_API_KEY"),
api_key=openai_api_key,
base_url=api_base_url,
)
openai_clients[client_key] = client