vrr/open-notebook
2
0
Fork 0
mirror of https://github.com/lfnovo/open-notebook.git synced 2026-04-29 20:10:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

model manager

Browse source
This commit is contained in:
LUIS NOVO 2024-11-01 20:37:23 -03:00
parent 3b7dd5f25f
commit a9ac4a6dc8
6 changed files with 141 additions and 43 deletions
Download patch file Download diff file Expand all files Collapse all files

6
open_notebook/domain/base.py
View file

@ -68,11 +68,9 @@ class ObjectModel(BaseModel):
return None
def save(self) -> None:
from open_notebook.domain.models import DefaultModels
from open_notebook.models import get_model
from open_notebook.models import model_manager
DEFAULT_MODELS = DefaultModels.load()
EMBEDDING_MODEL = get_model(DEFAULT_MODELS.default_embedding_model)
EMBEDDING_MODEL = model_manager.get_default_model("embedding")
try:
logger.debug(f"Validating {self.__class__.__name__}")