from pilottai import Pilott
from pilottai.tools import Tool
from pilottai.agent import Agent
from duckduckgo_search import DDGS
from pilottai.core import AgentConfig, AgentType, LLMConfig

# Configure LLM
llm_config = LLMConfig(
  model_name="gpt-4",
  provider="openai",
  api_key="your-api-key"
)

def duckduckgo_search(query, max_results=5):
    """Perform a DuckDuckGo search and return top results."""
    with DDGS() as ddgs:
        results = ddgs.text(query, max_results=max_results)
        return [{"title": r["title"], "link": r["href"], "snippet": r["body"]} for r in results]

search_tool = Tool(
                name="duckduckgo_search",
                description="Search DuckDuckGo for relevant information on any topic",
                function=duckduckgo_search,
                parameters={
                    "query": "str - The search query",
                    "num_results": "int - Number of results to return (max 10)"
                }
            )

query = "Type your question here"

search_agent = Agent(
                title="search_specialist",
                goal="Find the most relevant and credible sources for any given query",
                description="An expert at formulating search queries and identifying high-quality, relevant sources",
                jobs=f"Search for information about: '{query}' using DuckDuckGo and rank the results by relevance and credibility. Return the top 5 most relevant sources.",
                tools=[search_tool],
                llm_config=llm_config
              )


synthesis_results = await Pilott(agents=[search_agent], name="Search Bot", llm_config=llm_config).serve()