File size: 589 Bytes
8bb8f7a 2ef81cb b040e77 2ef81cb 7b739bd 2ef81cb a1eb19d 2ef81cb 867dc2a 2ef81cb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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)
|