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

v1 of transformations

Browse source
This commit is contained in:
LUIS NOVO 2024-10-23 10:59:05 -03:00
parent e020511876
commit 02ff05b6fd
16 changed files with 389 additions and 40 deletions
Download patch file Download diff file Expand all files Collapse all files

1
open_notebook/graphs/summary.py
View file

@ -57,7 +57,6 @@ def chunk_condition(state: SummaryState) -> Literal["get_chunk", END]: # type:
return END
# todo: build a helper method for LLM communication on all graphs
def call_model(state: SummaryState, config: RunnableConfig) -> dict:
model_name = config.get("configurable", {}).get(
"model_name", os.environ.get("SUMMARIZATION_MODEL")