cleanup logs
This commit is contained in:
parent
a2123d03b0
commit
42b9b2da07
2 changed files with 0 additions and 4 deletions
|
|
@ -6,7 +6,6 @@ from langchain_core.runnables import (
|
||||||
RunnableConfig,
|
RunnableConfig,
|
||||||
)
|
)
|
||||||
from langgraph.graph import END, START, StateGraph
|
from langgraph.graph import END, START, StateGraph
|
||||||
from loguru import logger
|
|
||||||
from typing_extensions import Annotated, TypedDict
|
from typing_extensions import Annotated, TypedDict
|
||||||
|
|
||||||
from open_notebook.graphs.utils import run_pattern
|
from open_notebook.graphs.utils import run_pattern
|
||||||
|
|
@ -33,7 +32,6 @@ def call_model(state: dict, config: RunnableConfig) -> dict:
|
||||||
}
|
}
|
||||||
current_transformation = "patterns/custom"
|
current_transformation = "patterns/custom"
|
||||||
|
|
||||||
logger.debug(f"Using input: {input_args}")
|
|
||||||
transformation_result = run_pattern(
|
transformation_result = run_pattern(
|
||||||
pattern_name=current_transformation,
|
pattern_name=current_transformation,
|
||||||
model_name=model_name,
|
model_name=model_name,
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,6 @@
|
||||||
import os
|
import os
|
||||||
|
|
||||||
from langchain.output_parsers import OutputFixingParser
|
from langchain.output_parsers import OutputFixingParser
|
||||||
from loguru import logger
|
|
||||||
|
|
||||||
from open_notebook.llm_router import get_langchain_model
|
from open_notebook.llm_router import get_langchain_model
|
||||||
from open_notebook.prompter import Prompter
|
from open_notebook.prompter import Prompter
|
||||||
|
|
@ -32,7 +31,6 @@ def run_pattern(
|
||||||
system_prompt = Prompter(prompt_template=pattern_name, parser=parser).render(
|
system_prompt = Prompter(prompt_template=pattern_name, parser=parser).render(
|
||||||
data=state
|
data=state
|
||||||
)
|
)
|
||||||
logger.debug(f"System prompt: {system_prompt}")
|
|
||||||
|
|
||||||
if len(messages) > 0:
|
if len(messages) > 0:
|
||||||
response = chain.invoke([system_prompt] + messages)
|
response = chain.invoke([system_prompt] + messages)
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue