gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix linting

Browse source
This commit is contained in:
yangdx 2025-02-06 23:12:35 +08:00 • committed by ultrageopro
parent e756aca3a2
commit a61db0852a
No known key found for this signature in database
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
lightrag/llm/openai.py
View file

@ -78,10 +78,13 @@ from lightrag.api import __api_version__
import numpy as np
from typing import Union
class InvalidResponseError(Exception):
"""Custom exception class for triggering retry mechanism"""
pass
@retry(
stop=stop_after_attempt(3),
wait=wait_exponential(multiplier=1, min=4, max=10),
@ -105,10 +108,11 @@ async def openai_complete_if_cache(
default_headers = {
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_8) LightRAG/{__api_version__}",
"Content-Type": "application/json"
"Content-Type": "application/json",
}
openai_async_client = (
AsyncOpenAI(default_headers=default_headers) if base_url is None
AsyncOpenAI(default_headers=default_headers)
if base_url is None
else AsyncOpenAI(base_url=base_url, default_headers=default_headers)
)
kwargs.pop("hashing_kv", None)
@ -295,10 +299,11 @@ async def openai_embed(
default_headers = {
"User-Agent": f"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_8) LightRAG/{__api_version__}",
"Content-Type": "application/json"
"Content-Type": "application/json",
}
openai_async_client = (
AsyncOpenAI(default_headers=default_headers) if base_url is None
AsyncOpenAI(default_headers=default_headers)
if base_url is None
else AsyncOpenAI(base_url=base_url, default_headers=default_headers)
)
response = await openai_async_client.embeddings.create(