@@ -36,10 +36,10 @@ class NOVITA(OPENAI):
3636 """Novita AI provider"""
3737
3838 def __init__ (self , * args , ** kwargs ):
39- kwargs .setdefault ("model" , model )
40- if not model in available_models :
39+ model_choice = kwargs .setdefault ("model" , model )
40+ if not model_choice in available_models :
4141 raise UnsupportedModelError (
42- f"Model '{ model } ' is not yet supported. Choose from { available_models } "
42+ f"Model '{ model_choice } ' is not yet supported. Choose from { available_models } "
4343 )
4444 super ().__init__ (* args , ** kwargs )
4545 self .chat_endpoint = "https://api.novita.ai/v3/openai/chat/completions"
@@ -49,10 +49,10 @@ class AsyncNOVITA(AsyncOPENAI):
4949 """Async Novita AI provider"""
5050
5151 def __init__ (self , * args , ** kwargs ):
52- kwargs .setdefault ("model" , model )
53- if not model in available_models :
52+ model_choice = kwargs .setdefault ("model" , model )
53+ if not model_choice in available_models :
5454 raise UnsupportedModelError (
55- f"Model '{ model } ' is not yet supported choose from { available_models } "
55+ f"Model '{ model_choice } ' is not yet supported. Choose from { available_models } "
5656 )
5757 super ().__init__ (* args , ** kwargs )
5858 self .chat_endpoint = "https://api.novita.ai/v3/openai/chat/completions"
0 commit comments