vrr/open-notebook
2
0
Fork 0
mirror of https://github.com/lfnovo/open-notebook.git synced 2026-05-05 15:24:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: implement the new model management based on esperanto framework

Browse source
This commit is contained in:
LUIS NOVO 2025-06-08 19:38:43 -03:00
parent 10049342cb
commit bea43f3ce7
4 changed files with 58 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

3
open_notebook/graphs/utils.py
View file

@ -1,8 +1,8 @@
from esperanto import LanguageModel
from langchain_core.language_models.chat_models import BaseChatModel
from loguru import logger
from open_notebook.domain.models import model_manager
from open_notebook.models.llms import LanguageModel
from open_notebook.utils import token_count
@ -27,5 +27,6 @@ def provision_langchain_model(
else:
model = model_manager.get_default_model(default_type, **kwargs)
logger.debug(f"Using model: {model}")
assert isinstance(model, LanguageModel), f"Model is not a LanguageModel: {model}"
return model.to_langchain()