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

@ -13,3 +13,4 @@ class Nvidia(OpenaiTemplate):
needs_auth = True
models_needs_auth = True
default_model = DEFAULT_MODEL
add_user = False

View file

@ -26,6 +26,7 @@ class OpenaiTemplate(AsyncGeneratorProvider, ProviderModelMixin, RaiseErrorMixin
models_needs_auth = False
use_model_names = False
ssl = None
add_user = True
@classmethod
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,
stop=stop,
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_body
)