from __future__ import annotations

from langchain_core.tools import BaseTool, tool

from memory_agent.knowledge import KnowledgeService


class KnowledgeToolFactory:
    def __init__(self, service: KnowledgeService) -> None:
        self._service = service

    def build(self, namespace: str) -> list[BaseTool]:
        @tool
        def answer_from_knowledge(question: str) -> str:
            """Answer a user question using ONLY previously stored knowledge."""

            return self._service.answer_from_knowledge(namespace=namespace, question=question)

        @tool
        def store_knowledge(
            knowledge_text: str,
            fact_key: str | None = None,
            fact_value: str | None = None,
        ) -> str:
            """Store or update knowledge in FAISS memory. Use when user shares new facts."""

            return self._service.store_knowledge(
                namespace=namespace,
                knowledge_text=knowledge_text,
                fact_key=fact_key,
                fact_value=fact_value,
            )

        return [answer_from_knowledge, store_knowledge]