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 16:24:02 +08:00
parent 070501fdaa
commit 290a4d5ec0
1 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
lightrag/llm/openai.py
View File

@@ -148,7 +148,11 @@ async def openai_complete_if_cache(
async def openai_complete( async def openai_complete(
prompt, system_prompt=None, history_messages=None, keyword_extraction=False, **kwargs prompt,
system_prompt=None,
history_messages=None,
keyword_extraction=False,
**kwargs,
) -> Union[str, AsyncIterator[str]]: ) -> Union[str, AsyncIterator[str]]:
if history_messages is None: if history_messages is None:
history_messages = [] history_messages = []
@@ -166,7 +170,11 @@ async def openai_complete(
async def gpt_4o_complete( async def gpt_4o_complete(
prompt, system_prompt=None, history_messages=None, keyword_extraction=False, **kwargs prompt,
system_prompt=None,
history_messages=None,
keyword_extraction=False,
**kwargs,
) -> str: ) -> str:
if history_messages is None: if history_messages is None:
history_messages = [] history_messages = []
@@ -183,7 +191,11 @@ async def gpt_4o_complete(
async def gpt_4o_mini_complete( async def gpt_4o_mini_complete(
prompt, system_prompt=None, history_messages=None, keyword_extraction=False, **kwargs prompt,
system_prompt=None,
history_messages=None,
keyword_extraction=False,
**kwargs,
) -> str: ) -> str:
if history_messages is None: if history_messages is None:
history_messages = [] history_messages = []
@@ -200,7 +212,11 @@ async def gpt_4o_mini_complete(
async def nvidia_openai_complete( async def nvidia_openai_complete(
prompt, system_prompt=None, history_messages=None, keyword_extraction=False, **kwargs prompt,
system_prompt=None,
history_messages=None,
keyword_extraction=False,
**kwargs,
) -> str: ) -> str:
if history_messages is None: if history_messages is None:
history_messages = [] history_messages = []