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

rename model_name to model_id

Browse source
This commit is contained in:
LUIS NOVO 2024-11-01 12:07:00 -03:00
parent af315a0bab
commit c65bf8ba12
7 changed files with 40 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

6
open_notebook/graphs/multipattern.py
View file

@ -18,8 +18,8 @@ class PatternChainState(TypedDict):
def call_model(state: dict, config: RunnableConfig) -> dict:
model_name = config.get("configurable", {}).get(
"model_name", DEFAULT_MODELS.default_transformation_model
model_id = config.get("configurable", {}).get(
"model_id", DEFAULT_MODELS.default_transformation_model
)
transformations = state["transformations"]
current_transformation = transformations.pop(0)
@ -34,7 +34,7 @@ def call_model(state: dict, config: RunnableConfig) -> dict:
transformation_result = run_pattern(
pattern_name=current_transformation,
model_name=model_name,
model_id=model_id,
state=input_args,
)
return {