SHAP-plot-interpreter / src /tools_loader.py
samyakshrestha's picture
Fixed typo
139cc06
raw
history blame contribute delete
630 Bytes
import os
from pathlib import Path
from .tools.shap_vision_tool import ShapVisionTool # Import SHAP vision tool
def get_tools():
"""Create and return all configured tools"""
# Get paths and API keys
groq_key = os.getenv("GROQ_API_KEY") # Retrieve GROQ API key from environment
gemini_key = os.getenv("GEMINI_API_KEY") # Retrieve Gemini API key from environment
# Create tool
shap_tool = ShapVisionTool(metadata={"GEMINI_API_KEY": gemini_key}) # Initialize SHAP vision tool with Gemini API key
# Return all tools in a dictionary
return {"shap_tool": shap_tool}