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

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
1 changed files with 9 additions and 4 deletions
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 import numpy as np
from typing import Union from typing import Union
class InvalidResponseError(Exception): class InvalidResponseError(Exception):
"""Custom exception class for triggering retry mechanism""" """Custom exception class for triggering retry mechanism"""
pass pass
@retry( @retry(
stop=stop_after_attempt(3), stop=stop_after_attempt(3),
wait=wait_exponential(multiplier=1, min=4, max=10), wait=wait_exponential(multiplier=1, min=4, max=10),
@@ -105,10 +108,11 @@ async def openai_complete_if_cache(
default_headers = { default_headers = {
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_8) LightRAG/{__api_version__}", "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 = ( 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) else AsyncOpenAI(base_url=base_url, default_headers=default_headers)
) )
kwargs.pop("hashing_kv", None) kwargs.pop("hashing_kv", None)
@@ -295,10 +299,11 @@ async def openai_embed(
default_headers = { default_headers = {
"User-Agent": f"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_8) LightRAG/{__api_version__}", "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 = ( 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) else AsyncOpenAI(base_url=base_url, default_headers=default_headers)
) )
response = await openai_async_client.embeddings.create( response = await openai_async_client.embeddings.create(