From f96fc580b385073a75c0c85fc1dba4bd2292f432 Mon Sep 17 00:00:00 2001 From: LUIS NOVO Date: Tue, 22 Oct 2024 16:45:08 -0300 Subject: [PATCH] make doc_query tool name consistent with prompt --- open_notebook/graphs/doc_query.py | 2 +- prompts/{ask_content.jinja => doc_query.jinja} | 0 2 files changed, 1 insertion(+), 1 deletion(-) rename prompts/{ask_content.jinja => doc_query.jinja} (100%) diff --git a/open_notebook/graphs/doc_query.py b/open_notebook/graphs/doc_query.py index 8b22e6f..068d8a4 100644 --- a/open_notebook/graphs/doc_query.py +++ b/open_notebook/graphs/doc_query.py @@ -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} diff --git a/prompts/ask_content.jinja b/prompts/doc_query.jinja similarity index 100% rename from prompts/ask_content.jinja rename to prompts/doc_query.jinja