from smolagents import CodeAgent, InferenceClientModel
from pdf_qa_tool import PDFQATool
from mind_map_tool import MindMapTool
from image_analysis_tool import ImageAnalysisTool
from data_graph_tool import DataGraphTool
from web_search_tool import WebSearchTool
import os
hf_token = os.environ.get("HF_TOKEN")


tools = [
    PDFQATool(),
    MindMapTool(),
    ImageAnalysisTool(),
    DataGraphTool(),
    WebSearchTool()
]

model = InferenceClientModel(model_id="mistralai/Mixtral-8x7B-Instruct-v0.1", token=hf_token)

agent = CodeAgent(tools=tools, model=model, add_base_tools=False)