vrr/open-notebook
2
0
Fork 0
mirror of https://github.com/lfnovo/open-notebook.git synced 2026-04-30 12:30:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: implement ai_prompter library

Browse source
This commit is contained in:
LUIS NOVO 2025-06-01 08:09:33 -03:00
parent 858b5e0d6e
commit 2afbd36cb4
7 changed files with 21 additions and 122 deletions
Download patch file Download diff file Expand all files Collapse all files

8
open_notebook/graphs/prompt.py
View file

@ -1,15 +1,13 @@
from typing import Any, Optional
from ai_prompter import Prompter
from langchain_core.messages import HumanMessage, SystemMessage
from langchain_core.runnables import (
RunnableConfig,
)
from langchain_core.runnables import RunnableConfig
from langgraph.graph import END, START, StateGraph
from loguru import logger
from typing_extensions import TypedDict
from open_notebook.graphs.utils import provision_langchain_model
from open_notebook.prompter import Prompter
class PatternChainState(TypedDict):
@ -22,7 +20,7 @@ class PatternChainState(TypedDict):
def call_model(state: dict, config: RunnableConfig) -> dict:
content = state["input_text"]
system_prompt = Prompter(
prompt_text=state["prompt"], parser=state.get("parser")
template_text=state["prompt"], parser=state.get("parser")
).render(data=state)
logger.warning(content)
payload = [SystemMessage(content=system_prompt)] + [HumanMessage(content=content)]