agents/knowledge_agent.py

from smolagents import CodeAgent
import sys
import os

sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from tools import search_schemes, SchemeRetrieverTool

def create_knowledge_agent(model):
    retriever_tool = SchemeRetrieverTool()
    return CodeAgent(
        tools=[retriever_tool, search_schemes],
        model=model,
        name="knowledge_researcher",
        description="Performs deep research into government schemes using the knowledge base. Use for detailed questions about scheme rules, recent changes, application procedures, or when other agents can't find the answer."
    )