Research Agent Seltz

"""Run `pip install groq seltz agno` to install dependencies."""

from pathlib import Path
from textwrap import dedent

from agno.agent import Agent
from agno.models.groq import Groq
from agno.tools.seltz import SeltzTools

# ---------------------------------------------------------------------------
# Create Agent
# ---------------------------------------------------------------------------

cwd = Path(__file__).parent.resolve()
tmp = cwd.joinpath("tmp")
if not tmp.exists():
    tmp.mkdir(exist_ok=True, parents=True)

agent = Agent(
    model=Groq(id="openai/gpt-oss-120b"),
    tools=[SeltzTools(max_results=10, show_results=True)],
    description="You are an advanced AI researcher writing a report on a topic.",
    instructions=[
        "For the provided topic, run 3 different searches.",
        "Read the results carefully and prepare a report.",
        "Focus on facts and make sure to provide references.",
    ],
    expected_output=dedent(
        """\
        An engaging, informative, and well-structured report in markdown format:

        ## Engaging Report Title

        ### Overview
        {give a brief introduction of the report and why the user should read this report}
        {make this section engaging and create a hook for the reader}

        ### Section 1
        {break the report into sections}
        {provide details/facts/processes in this section}

        ... more sections as necessary...

        ### Takeaways
        {provide key takeaways from the article}

        ### References
        - [Reference 1](link)
        - [Reference 2](link)
        - [Reference 3](link)

        ### About the Author
        {write a made up for yourself, give yourself a cyberpunk name and a title}

        - published on {date} in dd/mm/yyyy
        """
    ),
    markdown=True,
    add_datetime_to_context=True,
    save_response_to_file=str(tmp.joinpath("{message}.md")),
)
agent.print_response("Recent advances in AI safety", stream=True)

# ---------------------------------------------------------------------------
# Run Agent
# ---------------------------------------------------------------------------

if __name__ == "__main__":
    pass