yangdx commited on
Commit
7d1f32a
·
1 Parent(s): a3f614a

Fix linting

Browse files
Files changed (1) hide show
  1. lightrag/llm/openai.py +20 -4
lightrag/llm/openai.py CHANGED
@@ -148,7 +148,11 @@ async def openai_complete_if_cache(
148
 
149
 
150
  async def openai_complete(
151
- prompt, system_prompt=None, history_messages=None, keyword_extraction=False, **kwargs
 
 
 
 
152
  ) -> Union[str, AsyncIterator[str]]:
153
  if history_messages is None:
154
  history_messages = []
@@ -166,7 +170,11 @@ async def openai_complete(
166
 
167
 
168
  async def gpt_4o_complete(
169
- prompt, system_prompt=None, history_messages=None, keyword_extraction=False, **kwargs
 
 
 
 
170
  ) -> str:
171
  if history_messages is None:
172
  history_messages = []
@@ -183,7 +191,11 @@ async def gpt_4o_complete(
183
 
184
 
185
  async def gpt_4o_mini_complete(
186
- prompt, system_prompt=None, history_messages=None, keyword_extraction=False, **kwargs
 
 
 
 
187
  ) -> str:
188
  if history_messages is None:
189
  history_messages = []
@@ -200,7 +212,11 @@ async def gpt_4o_mini_complete(
200
 
201
 
202
  async def nvidia_openai_complete(
203
- prompt, system_prompt=None, history_messages=None, keyword_extraction=False, **kwargs
 
 
 
 
204
  ) -> str:
205
  if history_messages is None:
206
  history_messages = []
 
148
 
149
 
150
  async def openai_complete(
151
+ prompt,
152
+ system_prompt=None,
153
+ history_messages=None,
154
+ keyword_extraction=False,
155
+ **kwargs,
156
  ) -> Union[str, AsyncIterator[str]]:
157
  if history_messages is None:
158
  history_messages = []
 
170
 
171
 
172
  async def gpt_4o_complete(
173
+ prompt,
174
+ system_prompt=None,
175
+ history_messages=None,
176
+ keyword_extraction=False,
177
+ **kwargs,
178
  ) -> str:
179
  if history_messages is None:
180
  history_messages = []
 
191
 
192
 
193
  async def gpt_4o_mini_complete(
194
+ prompt,
195
+ system_prompt=None,
196
+ history_messages=None,
197
+ keyword_extraction=False,
198
+ **kwargs,
199
  ) -> str:
200
  if history_messages is None:
201
  history_messages = []
 
212
 
213
 
214
  async def nvidia_openai_complete(
215
+ prompt,
216
+ system_prompt=None,
217
+ history_messages=None,
218
+ keyword_extraction=False,
219
+ **kwargs,
220
  ) -> str:
221
  if history_messages is None:
222
  history_messages = []