gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix lint

Browse Source
This commit is contained in:
zrguo
2025-04-03 14:44:56 +08:00
parent 9648300b18
commit e17e61f58e
1 changed files with 23 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
lightrag/llm/openai.py
View File

@@ -73,7 +73,11 @@ def create_openai_async_client(
client_configs = {} client_configs = {}
# Create a merged config dict with precedence: explicit params > client_configs > defaults # Create a merged config dict with precedence: explicit params > client_configs > defaults
merged_configs = {**client_configs, "default_headers": default_headers, "api_key": api_key} merged_configs = {
**client_configs,
"default_headers": default_headers,
"api_key": api_key,
}
if base_url is not None: if base_url is not None:
merged_configs["base_url"] = base_url merged_configs["base_url"] = base_url
@@ -136,9 +140,7 @@ async def openai_complete_if_cache(
# Create the OpenAI client # Create the OpenAI client
openai_async_client = create_openai_async_client( openai_async_client = create_openai_async_client(
api_key=api_key, api_key=api_key, base_url=base_url, client_configs=client_configs
base_url=base_url,
client_configs=client_configs
) )
# Remove special kwargs that shouldn't be passed to OpenAI # Remove special kwargs that shouldn't be passed to OpenAI
@@ -357,9 +359,7 @@ async def openai_embed(
""" """
# Create the OpenAI client # Create the OpenAI client
openai_async_client = create_openai_async_client( openai_async_client = create_openai_async_client(
api_key=api_key, api_key=api_key, base_url=base_url, client_configs=client_configs
base_url=base_url,
client_configs=client_configs
) )
response = await openai_async_client.embeddings.create( response = await openai_async_client.embeddings.create(