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

model fixes

Browse source
This commit is contained in:
LUIS NOVO 2024-11-01 22:43:33 -03:00
parent 223f1bdaf5
commit 7dc37a3ac7
2 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
open_notebook/graphs/utils.py
View file

@ -22,7 +22,9 @@ def provision_model(content, config, default_type):
)
return model_manager.get_default_model("large_context").to_langchain()
elif config.get("configurable", {}).get("model_id"):
return model_manager.get_model(config.get("configurable", {}).get("model_id"))
return model_manager.get_model(
config.get("configurable", {}).get("model_id")
).to_langchain()
else:
return model_manager.get_default_model(default_type).to_langchain()