import os

from dotenv import load_dotenv

from agno.os import AgentOS

from src.agent_runner import get_agent
from src.observability import init_phoenix_tracing
from src.workflow_runner import get_news_source_workflow

load_dotenv()
_tracing_enabled = init_phoenix_tracing()

_agent = get_agent()
_workflow = get_news_source_workflow()
_agent_os = AgentOS(agents=[_agent], workflows=[_workflow], tracing=_tracing_enabled)
app = _agent_os.get_app()


def serve_agent_os() -> None:
    host = os.getenv("AGENT_OS_HOST", "127.0.0.1")
    port = int(os.getenv("AGENT_OS_PORT", "7777"))
    _agent_os.serve(app=app, host=host, port=port, reload=False)


if __name__ == "__main__":
    serve_agent_os()