import os

from crewai import Agent, Task, Crew
from crewai_tools import (
    SerperDevTool,
    WebsiteSearchTool
)
from phoenix.otel import register

PHOENIX_API_KEY = os.environ["PHOENIX_API_KEY"]

os.environ["PHOENIX_CLIENT_HEADERS"] = f"api_key={PHOENIX_API_KEY}"
os.environ["PHOENIX_COLLECTOR_ENDPOINT"] = "https://app.phoenix.arize.com"

tracer_provider = register(
    auto_instrument=True,
    project_name="gaia"
)

def get_crew():
    search_tool = SerperDevTool()
    web_rag_tool = WebsiteSearchTool()
    
    researcher = Agent(
        role='Web Research Agent.',
        goal='Search the web and scrape web pages.',
        backstory='An expert researcher that can search the web and scrape web page content.',
        tools=[search_tool, web_rag_tool],
        verbose=True
    )
    
    research = Task(
        description='Search the web and scrape web pages.',
        expected_output='Scraped web page content.',
        agent=researcher
    )

    return Crew(
        agents=[researcher],
        tasks=[research],
        verbose=True,
        planning=True
    )