admin-valentin/open-notebook
1
0
Fork 0
Code Issues Pull requests 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
Show stats 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_name = os.environ.get("RETRIEVAL_MODEL", os.environ["DEFAULT_MODEL"])
model = get_langchain_model(model_name) 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}") logger.debug(f"System prompt: {system_prompt}")
ai_message = model.invoke(system_prompt) ai_message = model.invoke(system_prompt)
return {"answer": ai_message} return {"answer": ai_message}

0
prompts/ask_content.jinja → prompts/doc_query.jinja
View file