yangdx
commited on
Commit
·
7d1f32a
1
Parent(s):
a3f614a
Fix linting
Browse files- 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,
|
|
|
|
|
|
|
|
|
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,
|
|
|
|
|
|
|
|
|
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,
|
|
|
|
|
|
|
|
|
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,
|
|
|
|
|
|
|
|
|
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 = []
|