Skip to content

Commit f180337

Browse files
committed
fix nvidia provider
1 parent f51cb91 commit f180337

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

llms/extensions/providers/nvidia.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ def __init__(self, **kwargs):
1919
self.mode = kwargs.get("mode", "base")
2020
self.gen_url = kwargs.get("api", "https://ai.api.nvidia.com/v1/genai")
2121

22-
def to_response(self, response, chat, started_at):
22+
def to_response(self, response, chat, started_at, context=None):
2323
if "artifacts" in response:
2424
for artifact in response["artifacts"]:
2525
base64 = artifact.get("base64")

0 commit comments

Comments
 (0)