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

simplify the model selector

Browse source
This commit is contained in:
LUIS NOVO 2024-10-30 14:30:29 -03:00
parent 8bb5db158f
commit 859b7f6e7e
6 changed files with 151 additions and 75 deletions
Download patch file Download diff file Expand all files Collapse all files

6
open_notebook/graphs/utils.py
View file

@ -1,7 +1,7 @@
from langchain.output_parsers import OutputFixingParser
from open_notebook.config import DEFAULT_MODELS
from open_notebook.models.llms import get_langchain_model
from open_notebook.models import get_model
from open_notebook.prompter import Prompter
@ -16,12 +16,12 @@ def run_pattern(
if not model_name:
model_name = DEFAULT_MODELS.default_transformation_model
chain = get_langchain_model(model_name)
chain = get_model(model_name, model_type="language")
if parser:
chain = chain | parser
if output_fixing_model_name and parser:
output_fix_model = get_langchain_model(output_fixing_model_name)
output_fix_model = get_model(output_fixing_model_name, model_type="language")
chain = chain | OutputFixingParser.from_llm(
parser=parser,
llm=output_fix_model,