tools/BookRetriverRagAgent.py

from langchain.tools import tool


def get_retrieve_book_context_rag_agent_tool(agent):

    @tool(
        "Computer science knowledge retriever",
        description="Used to retrieve computer science related context and answer the request."
    )
    def retrieve_book_context_rag_agent(request: str):
        result = agent.invoke({
            "messages": [{"role": "user", "content": request}]
        })
        return result["messages"][-1].text

    return retrieve_book_context_rag_agent