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

rename doc_query tool

Browse source
This commit is contained in:
LUIS NOVO 2024-10-22 16:36:21 -03:00
parent e0c3fe26c8
commit 809ecb45e1
2 changed files with 18 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

12
open_notebook/graphs/tools.py
View file

@ -6,19 +6,21 @@ from langchain.tools import tool
@tool
def get_current_timestamp() -> str:
"""
name: get_current_timestamp
Returns the current timestamp in the format YYYYMMDDHHmmss.
"""
return datetime.now().strftime("%Y%m%d%H%M%S")
@tool
def ask_the_document(doc_id: str, question: str):
def doc_query(doc_id: str, question: str):
"""
Use this tool to ask a question to the document.
Another LLM will ready the document and answer the question.
Be specific and complete in your query given the LLM that will process it is very capable.
name: doc_query
Use this tool if you need to investigate into a particular document.
Another LLM will read the document and answer the question that you might have.
Use this when the user question cannot be answered with the content you have in context.
"""
from open_notebook.graphs.ask_content import graph
from open_notebook.graphs.doc_query import graph
result = graph.invoke({"doc_id": doc_id, "question": question})
return result["answer"]