vrr/open-notebook
2
0
Fork 0
mirror of https://github.com/lfnovo/open-notebook.git synced 2026-04-29 03:50:04 +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/graphs/utils.py
View file

@ -2,7 +2,7 @@ from langchain.output_parsers import OutputFixingParser
from loguru import logger
from open_notebook.domain.models import DefaultModels
from open_notebook.models import get_model
from open_notebook.models import model_manager
from open_notebook.prompter import Prompter
from open_notebook.utils import token_count
@ -33,12 +33,12 @@ def run_pattern(
or DEFAULT_MODELS.default_chat_model
)
chain = get_model(model_id)
chain = model_manager.get_default_model("transformation")
if parser:
chain = chain | parser
if output_fixing_model_id and parser:
output_fix_model = get_model(output_fixing_model_id)
output_fix_model = model_manager.get_model(output_fixing_model_id)
chain = chain | OutputFixingParser.from_llm(
parser=parser,
llm=output_fix_model,