22 lines
807 B
Python
22 lines
807 B
Python
from langchain_aws import AmazonKnowledgeBasesRetriever
|
|
from langchain_core.tools.retriever import create_retriever_tool
|
|
|
|
from .config import REGION, KNOWLEDGE_BASE_ID
|
|
|
|
|
|
def build_knowledge_base_tool():
|
|
retriever = AmazonKnowledgeBasesRetriever(
|
|
knowledge_base_id=KNOWLEDGE_BASE_ID,
|
|
retrieval_config={"vectorSearchConfiguration": {"numberOfResults": 5}},
|
|
region_name=REGION,
|
|
)
|
|
return create_retriever_tool(
|
|
retriever,
|
|
name="consultar_base_conhecimento",
|
|
description=(
|
|
"Consulta a base de conhecimento com informações sobre procedimentos internos de matrícula "
|
|
"para cursos técnicos e superiores do campus Capivari do IFSP. "
|
|
"Use esta ferramenta para responder dúvidas sobre matrícula."
|
|
),
|
|
)
|