From a61db0852a158a8f6bc2c9d5b23c5b5a082beeb3 Mon Sep 17 00:00:00 2001 From: yangdx Date: Thu, 6 Feb 2025 23:12:35 +0800 Subject: [PATCH] Fix linting --- lightrag/llm/openai.py | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) diff --git a/lightrag/llm/openai.py b/lightrag/llm/openai.py index ca451bcf..535d665c 100644 --- a/lightrag/llm/openai.py +++ b/lightrag/llm/openai.py @@ -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(