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