Spaces:
Build error
Build error
File size: 640 Bytes
0ede848 03d676c 33d3dfe 03d676c 9b09327 03d676c 9b09327 03d676c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
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(query: str) -> str:
"""Delegate the retrieve query context to the agent.
Args:
query: the full query of th user
"""
result = agent.invoke({
"messages": [{"role": "user", "content": query}]
})
return result["messages"][-1].text
return retrieve_book_context_rag_agent
|