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."
        ),
    )