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

make doc_query tool name consistent with prompt

Browse source
This commit is contained in:
LUIS NOVO 2024-10-22 16:45:08 -03:00
parent ab51a92ecd
commit f96fc580b3
2 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
open_notebook/graphs/doc_query.py
View file

@ -27,7 +27,7 @@ def call_model_with_messages(state: DocQueryState, config: RunnableConfig) -> di
model_name = os.environ.get("RETRIEVAL_MODEL", os.environ["DEFAULT_MODEL"])
model = get_langchain_model(model_name)
system_prompt = Prompter(prompt_template="ask_content").render(data=state)
system_prompt = Prompter(prompt_template="doc_query").render(data=state)
logger.debug(f"System prompt: {system_prompt}")
ai_message = model.invoke(system_prompt)
return {"answer": ai_message}