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

implement model config

Browse source
This commit is contained in:
LUIS NOVO 2024-10-30 14:09:24 -03:00
parent 63a568490e
commit 8bb5db158f
19 changed files with 434 additions and 105 deletions
Download patch file Download diff file Expand all files Collapse all files

7
open_notebook/graphs/utils.py
View file

@ -1,8 +1,7 @@
import os
from langchain.output_parsers import OutputFixingParser
from open_notebook.llm_router import get_langchain_model
from open_notebook.config import DEFAULT_MODELS
from open_notebook.models.llms import get_langchain_model
from open_notebook.prompter import Prompter
@ -15,7 +14,7 @@ def run_pattern(
output_fixing_model_name=None,
) -> dict:
if not model_name:
model_name = os.environ["DEFAULT_MODEL"]
model_name = DEFAULT_MODELS.default_transformation_model
chain = get_langchain_model(model_name)
if parser: