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

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 logging
import os
from threading import Thread from threading import Thread
from typing import Dict from typing import Dict
@ -40,7 +39,7 @@ def completion_with_backoff(
client: openai.OpenAI = openai_clients.get(client_key) client: openai.OpenAI = openai_clients.get(client_key)
if not client: if not client:
client = openai.OpenAI( client = openai.OpenAI(
api_key=openai_api_key or os.getenv("OPENAI_API_KEY"), api_key=openai_api_key,
base_url=api_base_url, base_url=api_base_url,
) )
openai_clients[client_key] = client 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}" client_key = f"{openai_api_key}--{api_base_url}"
if client_key not in openai_clients: if client_key not in openai_clients:
client: openai.OpenAI = openai.OpenAI( 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, base_url=api_base_url,
) )
openai_clients[client_key] = client openai_clients[client_key] = client