diff options
Diffstat (limited to 'g4f')
-rw-r--r-- | g4f/Provider/nexra/NexraBlackbox.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/g4f/Provider/nexra/NexraBlackbox.py b/g4f/Provider/nexra/NexraBlackbox.py index 87eea8e2..732593fe 100644 --- a/g4f/Provider/nexra/NexraBlackbox.py +++ b/g4f/Provider/nexra/NexraBlackbox.py @@ -39,8 +39,8 @@ class NexraBlackbox(AbstractProvider, ProviderModelMixin): stream: bool, **kwargs ) -> CreateResult: - model = model or cls.default_model - + model = cls.get_model(model) + headers = { 'Content-Type': 'application/json' } @@ -59,7 +59,7 @@ class NexraBlackbox(AbstractProvider, ProviderModelMixin): } response = requests.post(cls.url, headers=headers, json=data, stream=stream) - + if stream: return cls.process_streaming_response(response) else: |