From bdc61ca273dde2771c5bc8f93705652842f78d15 Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Sun, 7 Apr 2024 16:02:49 +0200 Subject: Add nodriver to OpenaiChat --- g4f/client/async_client.py | 3 ++- g4f/client/client.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) (limited to 'g4f/client') diff --git a/g4f/client/async_client.py b/g4f/client/async_client.py index 25de1c76..bab2fde9 100644 --- a/g4f/client/async_client.py +++ b/g4f/client/async_client.py @@ -87,7 +87,8 @@ def create_response( else: create = provider.create_completion response = create( - model, messages, stream, + model, messages, + stream=stream, **filter_none( proxy=proxy, max_tokens=max_tokens, diff --git a/g4f/client/client.py b/g4f/client/client.py index 8ce5d932..bc0bae6f 100644 --- a/g4f/client/client.py +++ b/g4f/client/client.py @@ -98,7 +98,8 @@ class Completions(): ) stop = [stop] if isinstance(stop, str) else stop response = provider.create_completion( - model, messages, stream, + model, messages, + stream=stream, **filter_none( proxy=self.client.get_proxy() if proxy is None else proxy, max_tokens=max_tokens, -- cgit v1.2.3