Fix add user in Nvidia

This commit is contained in:
hlohaus 2025-08-09 01:51:44 +02:00
parent f3923f8e50
commit b6dd7b39be
2 changed files with 4 additions and 2 deletions

View file

@ -12,4 +12,5 @@ class Nvidia(OpenaiTemplate):
active_by_default = True active_by_default = True
needs_auth = True needs_auth = True
models_needs_auth = True models_needs_auth = True
default_model = DEFAULT_MODEL default_model = DEFAULT_MODEL
add_user = False

View file

@ -26,6 +26,7 @@ class OpenaiTemplate(AsyncGeneratorProvider, ProviderModelMixin, RaiseErrorMixin
models_needs_auth = False models_needs_auth = False
use_model_names = False use_model_names = False
ssl = None ssl = None
add_user = True
@classmethod @classmethod
def get_models(cls, api_key: str = None, api_base: str = None) -> list[str]: def get_models(cls, api_key: str = None, api_base: str = None) -> list[str]:
@ -129,7 +130,7 @@ class OpenaiTemplate(AsyncGeneratorProvider, ProviderModelMixin, RaiseErrorMixin
top_p=top_p, top_p=top_p,
stop=stop, stop=stop,
stream="audio" not in extra_parameters if stream is None else stream, stream="audio" not in extra_parameters if stream is None else stream,
user=user, user=user if cls.add_user else None,
**extra_parameters, **extra_parameters,
**extra_body **extra_body
) )