import os
from dotenv import load_dotenv

from smolagents import (
    CodeAgent,
    OpenAIServerModel,
    DuckDuckGoSearchTool,
)

from tools.rag_tool import search_guest_story
from tools.weather_tool import weather_info_tool
from tools.hub_stats_tool import hub_stats_tool

# Load environment variables
load_dotenv()

# Create Groq model
model = OpenAIServerModel(
    model_id="llama-3.3-70b-versatile",
    api_key=os.getenv("GROQ_API_KEY"),
    api_base="https://api.groq.com/openai/v1",
)

# Create DuckDuckGo search tool
search_tool = DuckDuckGoSearchTool()

# Create Alfred Agent
agent = CodeAgent(
    tools=[
        search_guest_story,
        search_tool,
        weather_info_tool,
        hub_stats_tool,
    ],
    model=model,
)