admin-valentin/open-notebook
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: remove provider check, not needed

Browse source
This commit is contained in:
LUIS NOVO 2025-06-10 11:54:15 -03:00
parent 7239f719fd
commit f4b9ccbb22
1 changed files with 9 additions and 24 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
open_notebook/domain/models.py
View file

@ -85,21 +85,6 @@ class ModelManager:
]: ]:
raise ValueError(f"Invalid model type: {model.type}") raise ValueError(f"Invalid model type: {model.type}")
# todo: change to providers in the future
if model.provider not in [
"ollama",
"openrouter",
"vertexai-anthropic",
"litellm",
"vertexai",
"anthropic",
"openai",
"xai",
]:
raise ValueError(
f"Provider {model.provider} not compatible with {model.type} models"
)
if model.type == "language": if model.type == "language":
model_instance: LanguageModel = AIFactory.create_language( model_instance: LanguageModel = AIFactory.create_language(
model_name=model.name, model_name=model.name,
@ -148,9 +133,9 @@ class ModelManager:
if not model_id: if not model_id:
return None return None
model = self.get_model(model_id, **kwargs) model = self.get_model(model_id, **kwargs)
assert model is None or isinstance( assert model is None or isinstance(model, SpeechToTextModel), (
model, SpeechToTextModel f"Expected SpeechToTextModel but got {type(model)}"
), f"Expected SpeechToTextModel but got {type(model)}" )
return model return model
@property @property
@ -160,9 +145,9 @@ class ModelManager:
if not model_id: if not model_id:
return None return None
model = self.get_model(model_id, **kwargs) model = self.get_model(model_id, **kwargs)
assert model is None or isinstance( assert model is None or isinstance(model, TextToSpeechModel), (
model, TextToSpeechModel f"Expected TextToSpeechModel but got {type(model)}"
), f"Expected TextToSpeechModel but got {type(model)}" )
return model return model
@property @property
@ -172,9 +157,9 @@ class ModelManager:
if not model_id: if not model_id:
return None return None
model = self.get_model(model_id, **kwargs) model = self.get_model(model_id, **kwargs)
assert model is None or isinstance( assert model is None or isinstance(model, EmbeddingModel), (
model, EmbeddingModel f"Expected EmbeddingModel but got {type(model)}"
), f"Expected EmbeddingModel but got {type(model)}" )
return model return model
def get_default_model(self, model_type: str, **kwargs) -> Optional[ModelType]: def get_default_model(self, model_type: str, **kwargs) -> Optional[ModelType]: