agno 1.8.1__py3-none-any.whl → 2.0.0a1__py3-none-any.whl

agno/app/fastapi/sync_router.py

@@ -1,448 +0,0 @@
-import json
-from dataclasses import asdict
-from io import BytesIO
-from typing import Any, Dict, Generator, List, Optional, Union, cast
-from uuid import uuid4
-
-from fastapi import APIRouter, File, Form, HTTPException, Query, UploadFile
-from fastapi.responses import StreamingResponse
-
-from agno.agent.agent import Agent, RunResponse
-from agno.app.playground.utils import process_audio, process_document, process_image, process_video
-from agno.media import Audio, Image, Video
-from agno.media import File as FileMedia
-from agno.run.base import RunStatus
-from agno.run.response import RunResponseEvent
-from agno.run.team import RunResponseErrorEvent as TeamRunResponseErrorEvent
-from agno.run.team import TeamRunResponseEvent
-from agno.run.v2.workflow import WorkflowErrorEvent
-from agno.team.team import Team
-from agno.utils.log import logger
-from agno.workflow.v2.workflow import Workflow as WorkflowV2
-from agno.workflow.workflow import Workflow
-
-
-def agent_chat_response_streamer(
-    agent: Agent,
-    message: str,
-    session_id: Optional[str] = None,
-    user_id: Optional[str] = None,
-    images: Optional[List[Image]] = None,
-    audio: Optional[List[Audio]] = None,
-    videos: Optional[List[Video]] = None,
-) -> Generator:
-    try:
-        run_response = agent.run(
-            message,
-            session_id=session_id,
-            user_id=user_id,
-            images=images,
-            audio=audio,
-            videos=videos,
-            stream=True,
-            stream_intermediate_steps=True,
-        )
-        for run_response_chunk in run_response:
-            run_response_chunk = cast(RunResponseEvent, run_response_chunk)
-            yield run_response_chunk.to_json()
-    except Exception as e:
-        error_response = RunResponse(content=str(e), status=RunStatus.error)
-        yield error_response.to_json()
-        return
-
-
-def team_chat_response_streamer(
-    team: Team,
-    message: str,
-    session_id: Optional[str] = None,
-    user_id: Optional[str] = None,
-    images: Optional[List[Image]] = None,
-    audio: Optional[List[Audio]] = None,
-    videos: Optional[List[Video]] = None,
-    files: Optional[List[FileMedia]] = None,
-) -> Generator:
-    try:
-        run_response = team.run(
-            message,
-            session_id=session_id,
-            user_id=user_id,
-            images=images,
-            audio=audio,
-            videos=videos,
-            files=files,
-            stream=True,
-            stream_intermediate_steps=True,
-        )
-        for run_response_chunk in run_response:
-            run_response_chunk = cast(TeamRunResponseEvent, run_response_chunk)
-            yield run_response_chunk.to_json()
-    except Exception as e:
-        error_response = TeamRunResponseErrorEvent(
-            content=str(e),
-        )
-        yield error_response.to_json()
-        return
-
-
-def workflow_response_streamer(
-    workflow: WorkflowV2,
-    body: Union[Dict[str, Any], str],
-    session_id: Optional[str] = None,
-    user_id: Optional[str] = None,
-) -> Generator:
-    try:
-        if isinstance(body, dict):
-            run_response = workflow.run(
-                **body,
-                user_id=user_id,
-                session_id=session_id,
-                stream=True,
-                stream_intermediate_steps=True,
-            )
-        else:
-            run_response = workflow.run(
-                body,
-                user_id=user_id,
-                session_id=session_id,
-                stream=True,
-                stream_intermediate_steps=True,
-            )
-        for run_response_chunk in run_response:
-            yield run_response_chunk.to_json()
-    except Exception as e:
-        import traceback
-
-        traceback.print_exc(limit=3)
-        error_response = WorkflowErrorEvent(
-            error=str(e),
-        )
-        yield error_response.to_json()
-        return
-
-
-def get_sync_router(
-    agents: Optional[List[Agent]] = None, teams: Optional[List[Team]] = None, workflows: Optional[List[Workflow]] = None
-) -> APIRouter:
-    router = APIRouter()
-
-    if agents is None and teams is None and workflows is None:
-        raise ValueError("Either agents, teams or workflows must be provided.")
-
-    @router.get("/status")
-    def status():
-        return {"status": "available"}
-
-    def agent_process_file(
-        files: List[UploadFile],
-        agent: Agent,
-    ):
-        base64_images: List[Image] = []
-        base64_audios: List[Audio] = []
-        base64_videos: List[Video] = []
-        for file in files:
-            logger.info(f"Processing file: {file.content_type}")
-            if file.content_type in ["image/png", "image/jpeg", "image/jpg", "image/webp"]:
-                try:
-                    base64_image = process_image(file)
-                    base64_images.append(base64_image)
-                except Exception as e:
-                    logger.error(f"Error processing image {file.filename}: {e}")
-                    continue
-            elif file.content_type in ["audio/wav", "audio/mp3", "audio/mpeg"]:
-                try:
-                    base64_audio = process_audio(file)
-                    base64_audios.append(base64_audio)
-                except Exception as e:
-                    logger.error(f"Error processing audio {file.filename}: {e}")
-                    continue
-            elif file.content_type in [
-                "video/x-flv",
-                "video/quicktime",
-                "video/mpeg",
-                "video/mpegs",
-                "video/mpgs",
-                "video/mpg",
-                "video/mpg",
-                "video/mp4",
-                "video/webm",
-                "video/wmv",
-                "video/3gpp",
-            ]:
-                try:
-                    base64_video = process_video(file)
-                    base64_videos.append(base64_video)
-                except Exception as e:
-                    logger.error(f"Error processing video {file.filename}: {e}")
-                    continue
-            else:
-                # Check for knowledge base before processing documents
-                if agent.knowledge is None:
-                    raise HTTPException(status_code=404, detail="KnowledgeBase not found")
-
-                if file.content_type == "application/pdf":
-                    from agno.document.reader.pdf_reader import PDFReader
-
-                    contents = file.file.read()
-                    pdf_file = BytesIO(contents)
-                    pdf_file.name = file.filename
-                    file_content = PDFReader().read(pdf_file)
-                    if agent.knowledge is not None:
-                        agent.knowledge.load_documents(file_content)
-                elif file.content_type == "text/csv":
-                    from agno.document.reader.csv_reader import CSVReader
-
-                    contents = file.file.read()
-                    csv_file = BytesIO(contents)
-                    csv_file.name = file.filename
-                    file_content = CSVReader().read(csv_file)
-                    if agent.knowledge is not None:
-                        agent.knowledge.load_documents(file_content)
-                elif file.content_type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
-                    from agno.document.reader.docx_reader import DocxReader
-
-                    contents = file.file.read()
-                    docx_file = BytesIO(contents)
-                    docx_file.name = file.filename
-                    file_content = DocxReader().read(docx_file)
-                    if agent.knowledge is not None:
-                        agent.knowledge.load_documents(file_content)
-                elif file.content_type == "text/plain":
-                    from agno.document.reader.text_reader import TextReader
-
-                    contents = file.file.read()
-                    text_file = BytesIO(contents)
-                    text_file.name = file.filename
-                    file_content = TextReader().read(text_file)
-                    if agent.knowledge is not None:
-                        agent.knowledge.load_documents(file_content)
-
-                elif file.content_type == "application/json":
-                    from agno.document.reader.json_reader import JSONReader
-
-                    contents = file.file.read()
-                    json_file = BytesIO(contents)
-                    json_file.name = file.filename
-                    file_content = JSONReader().read(json_file)
-                    if agent.knowledge is not None:
-                        agent.knowledge.load_documents(file_content)
-                else:
-                    raise HTTPException(status_code=400, detail="Unsupported file type")
-
-        return base64_images, base64_audios, base64_videos
-
-    def team_process_file(
-        files: List[UploadFile],
-    ):
-        base64_images: List[Image] = []
-        base64_audios: List[Audio] = []
-        base64_videos: List[Video] = []
-        document_files: List[FileMedia] = []
-        for file in files:
-            if file.content_type in ["image/png", "image/jpeg", "image/jpg", "image/webp"]:
-                try:
-                    base64_image = process_image(file)
-                    base64_images.append(base64_image)
-                except Exception as e:
-                    logger.error(f"Error processing image {file.filename}: {e}")
-                    continue
-            elif file.content_type in ["audio/wav", "audio/mp3", "audio/mpeg"]:
-                try:
-                    base64_audio = process_audio(file)
-                    base64_audios.append(base64_audio)
-                except Exception as e:
-                    logger.error(f"Error processing audio {file.filename}: {e}")
-                    continue
-            elif file.content_type in [
-                "video/x-flv",
-                "video/quicktime",
-                "video/mpeg",
-                "video/mpegs",
-                "video/mpgs",
-                "video/mpg",
-                "video/mpg",
-                "video/mp4",
-                "video/webm",
-                "video/wmv",
-                "video/3gpp",
-            ]:
-                try:
-                    base64_video = process_video(file)
-                    base64_videos.append(base64_video)
-                except Exception as e:
-                    logger.error(f"Error processing video {file.filename}: {e}")
-                    continue
-            elif file.content_type in [
-                "application/pdf",
-                "text/csv",
-                "application/vnd.openxmlformats-officedocument.wordprocessingml.document",
-                "text/plain",
-                "application/json",
-            ]:
-                document_file = process_document(file)
-                if document_file is not None:
-                    document_files.append(document_file)
-            else:
-                raise HTTPException(status_code=400, detail="Unsupported file type")
-
-        return base64_images, base64_audios, base64_videos, document_files
-
-    @router.post("/runs")
-    def run_agent_or_team_or_workflow(
-        message: str = Form(None),
-        stream: bool = Form(False),
-        monitor: bool = Form(False),
-        agent_id: Optional[str] = Query(None),
-        team_id: Optional[str] = Query(None),
-        workflow_id: Optional[str] = Query(None),
-        workflow_input: Optional[str] = Form(None),
-        session_id: Optional[str] = Form(None),
-        user_id: Optional[str] = Form(None),
-        files: Optional[List[UploadFile]] = File(None),
-    ):
-        if session_id is not None and session_id != "":
-            logger.debug(f"Continuing session: {session_id}")
-        else:
-            logger.debug("Creating new session")
-            session_id = str(uuid4())
-
-        # Only one of agent_id, team_id or workflow_id can be provided
-        if agent_id and team_id or agent_id and workflow_id or team_id and workflow_id:
-            raise HTTPException(status_code=400, detail="Only one of agent_id, team_id or workflow_id can be provided")
-
-        if not agent_id and not team_id and not workflow_id:
-            raise HTTPException(status_code=400, detail="One of agent_id, team_id or workflow_id must be provided")
-
-        agent = None
-        team = None
-        workflow = None
-
-        if agent_id and agents:
-            agent = next((agent for agent in agents if agent.agent_id == agent_id), None)
-            if agent is None:
-                raise HTTPException(status_code=404, detail="Agent not found")
-            if not message:
-                raise HTTPException(status_code=400, detail="Message is required")
-        if team_id and teams:
-            team = next((team for team in teams if team.team_id == team_id), None)
-            if team is None:
-                raise HTTPException(status_code=404, detail="Team not found")
-            if not message:
-                raise HTTPException(status_code=400, detail="Message is required")
-        if workflow_id and workflows:
-            workflow = next((workflow for workflow in workflows if workflow.workflow_id == workflow_id), None)
-            if workflow is None:
-                raise HTTPException(status_code=404, detail="Workflow not found")
-            if not workflow_input:
-                raise HTTPException(status_code=400, detail="Workflow input is required")
-
-            # Parse workflow_input into a dict if it is a valid JSON
-            try:
-                parsed_workflow_input = json.loads(workflow_input)
-                workflow_input = parsed_workflow_input
-            except json.JSONDecodeError:
-                pass
-
-        if agent:
-            agent.monitoring = bool(monitor)
-        elif team:
-            team.monitoring = bool(monitor)
-        elif workflow:
-            workflow.monitoring = bool(monitor)
-
-        if files:
-            if agent:
-                base64_images, base64_audios, base64_videos = agent_process_file(files, agent)
-            elif team:
-                base64_images, base64_audios, base64_videos, document_files = team_process_file(files)
-
-        if stream:
-            if agent:
-                return StreamingResponse(
-                    agent_chat_response_streamer(
-                        agent,
-                        message,
-                        session_id=session_id,
-                        user_id=user_id,
-                        images=base64_images if base64_images else None,
-                        audio=base64_audios if base64_audios else None,
-                        videos=base64_videos if base64_videos else None,
-                    ),
-                    media_type="text/event-stream",
-                )
-            elif team:
-                return StreamingResponse(
-                    team_chat_response_streamer(
-                        team,
-                        message,
-                        session_id=session_id,
-                        user_id=user_id,
-                        images=base64_images if base64_images else None,
-                        audio=base64_audios if base64_audios else None,
-                        videos=base64_videos if base64_videos else None,
-                        files=document_files if document_files else None,
-                    ),
-                    media_type="text/event-stream",
-                )
-            elif workflow:
-                if isinstance(workflow, Workflow):
-                    workflow_instance = workflow.deep_copy(update={"workflow_id": workflow_id})
-                    workflow_instance.user_id = user_id
-                    workflow_instance.session_name = None
-                    if isinstance(workflow_input, dict):
-                        return StreamingResponse(
-                            (json.dumps(asdict(result)) for result in workflow_instance.run(**workflow_input)),
-                            media_type="text/event-stream",
-                        )
-                    else:
-                        return StreamingResponse(
-                            (json.dumps(asdict(result)) for result in workflow_instance.run(workflow_input)),  # type: ignore
-                            media_type="text/event-stream",
-                        )
-                else:
-                    return StreamingResponse(
-                        workflow_response_streamer(workflow, workflow_input, session_id=session_id, user_id=user_id),
-                        media_type="text/event-stream",
-                    )
-        else:
-            if agent:
-                run_response = cast(
-                    RunResponse,
-                    agent.run(
-                        message=message,
-                        session_id=session_id,
-                        user_id=user_id,
-                        images=base64_images if base64_images else None,
-                        audio=base64_audios if base64_audios else None,
-                        videos=base64_videos if base64_videos else None,
-                        stream=False,
-                    ),
-                )
-                return run_response.to_dict()
-            elif team:
-                team_run_response = team.run(
-                    message=message,
-                    session_id=session_id,
-                    user_id=user_id,
-                    images=base64_images if base64_images else None,
-                    audio=base64_audios if base64_audios else None,
-                    videos=base64_videos if base64_videos else None,
-                    files=document_files if document_files else None,
-                    stream=False,
-                )
-                return team_run_response.to_dict()
-            elif workflow:
-                if isinstance(workflow, Workflow):
-                    workflow_instance = workflow.deep_copy(update={"workflow_id": workflow_id})
-                    workflow_instance.user_id = user_id
-                    workflow_instance.session_name = None
-                    if isinstance(workflow_input, dict):
-                        return workflow_instance.run(**workflow_input).to_dict()
-                    else:
-                        return workflow_instance.run(workflow_input).to_dict()  # type: ignore
-                else:
-                    if isinstance(workflow_input, dict):
-                        return workflow.run(**workflow_input, session_id=session_id, user_id=user_id).to_dict()
-                    else:
-                        return workflow.run(workflow_input, session_id=session_id, user_id=user_id).to_dict()
-
-    return router

agno/app/playground/app.py

@@ -1,228 +0,0 @@
-from os import getenv
-from typing import Any, Callable, Dict, List, Optional, Union
-from urllib.parse import quote
-from uuid import uuid4
-
-from fastapi import FastAPI, HTTPException
-from fastapi.responses import JSONResponse
-from fastapi.routing import APIRouter
-from rich import box
-from rich.panel import Panel
-from starlette.middleware.cors import CORSMiddleware
-from starlette.requests import Request
-
-from agno.agent.agent import Agent
-from agno.api.playground import PlaygroundEndpointCreate
-from agno.app.playground.async_router import get_async_playground_router
-from agno.app.playground.sync_router import get_sync_playground_router
-from agno.app.utils import generate_id
-from agno.cli.console import console
-from agno.cli.settings import agno_cli_settings
-from agno.playground.settings import PlaygroundSettings
-from agno.team.team import Team
-from agno.utils.log import log_debug, logger
-from agno.workflow.workflow import Workflow
-
-
-class Playground:
-    def __init__(
-        self,
-        agents: Optional[List[Agent]] = None,
-        teams: Optional[List[Team]] = None,
-        workflows: Optional[List[Workflow]] = None,
-        settings: Optional[PlaygroundSettings] = None,
-        api_app: Optional[FastAPI] = None,
-        router: Optional[APIRouter] = None,
-        app_id: Optional[str] = None,
-        name: Optional[str] = None,
-        description: Optional[str] = None,
-        monitoring: bool = True,
-    ):
-        if not agents and not workflows and not teams:
-            raise ValueError("Either agents, teams or workflows must be provided.")
-
-        self.agents: Optional[List[Agent]] = agents
-        self.workflows: Optional[List[Workflow]] = workflows
-        self.teams: Optional[List[Team]] = teams
-
-        self.settings: PlaygroundSettings = settings or PlaygroundSettings()
-        self.api_app: Optional[FastAPI] = api_app
-        self.router: Optional[APIRouter] = router
-
-        self.endpoints_created: Optional[PlaygroundEndpointCreate] = None
-
-        self.app_id: Optional[str] = app_id
-        self.name: Optional[str] = name
-        self.monitoring = monitoring
-        self.description = description
-        self.set_app_id()
-        if self.agents:
-            for agent in self.agents:
-                if not agent.app_id:
-                    agent.app_id = self.app_id
-                agent.initialize_agent()
-                # Required for playground to work
-                agent.store_events = True
-
-        if self.teams:
-            for team in self.teams:
-                if not team.app_id:
-                    team.app_id = self.app_id
-                team.initialize_team()
-                # Required for playground to work
-                team.store_events = True
-                for member in team.members:
-                    if isinstance(member, Agent):
-                        if not member.app_id:
-                            member.app_id = self.app_id
-
-                        member.team_id = None
-                        member.initialize_agent()
-                    elif isinstance(member, Team):
-                        member.initialize_team()
-
-        if self.workflows:
-            for workflow in self.workflows:
-                if hasattr(workflow, "app_id") and not workflow.app_id:
-                    workflow.app_id = self.app_id
-                if not workflow.workflow_id:
-                    workflow.workflow_id = generate_id(workflow.name)
-                workflow.initialize_workflow()
-
-    def set_app_id(self) -> str:
-        # If app_id is already set, keep it instead of overriding with UUID
-        if self.app_id is None:
-            self.app_id = str(uuid4())
-
-        # Don't override existing app_id
-        return self.app_id
-
-    def _set_monitoring(self) -> None:
-        """Override monitoring and telemetry settings based on environment variables."""
-
-        # Only override if the environment variable is set
-        monitor_env = getenv("AGNO_MONITOR")
-        if monitor_env is not None:
-            self.monitoring = monitor_env.lower() == "true"
-
-    def get_router(self) -> APIRouter:
-        return get_sync_playground_router(self.agents, self.workflows, self.teams, self.app_id)
-
-    def get_async_router(self) -> APIRouter:
-        return get_async_playground_router(self.agents, self.workflows, self.teams, self.app_id)
-
-    def get_app(self, use_async: bool = True, prefix: str = "/v1", lifespan: Optional[Callable] = None) -> FastAPI:
-        if not self.api_app:
-            self.api_app = FastAPI(
-                title=self.settings.title,
-                docs_url="/docs" if self.settings.docs_enabled else None,
-                redoc_url="/redoc" if self.settings.docs_enabled else None,
-                openapi_url="/openapi.json" if self.settings.docs_enabled else None,
-                lifespan=lifespan,
-            )
-
-        if not self.api_app:
-            raise Exception("API App could not be created.")
-
-        @self.api_app.exception_handler(HTTPException)
-        async def http_exception_handler(request: Request, exc: HTTPException) -> JSONResponse:
-            return JSONResponse(
-                status_code=exc.status_code,
-                content={"detail": str(exc.detail)},
-            )
-
-        async def general_exception_handler(request: Request, call_next):
-            try:
-                return await call_next(request)
-            except Exception as e:
-                return JSONResponse(
-                    status_code=e.status_code if hasattr(e, "status_code") else 500,
-                    content={"detail": str(e)},
-                )
-
-        self.api_app.middleware("http")(general_exception_handler)
-
-        if not self.router:
-            self.router = APIRouter(prefix=prefix)
-
-        if not self.router:
-            raise Exception("API Router could not be created.")
-
-        if use_async:
-            self.router.include_router(self.get_async_router())
-        else:
-            self.router.include_router(self.get_router())
-        self.api_app.include_router(self.router)
-
-        self.api_app.add_middleware(
-            CORSMiddleware,
-            allow_origins=self.settings.cors_origin_list,
-            allow_credentials=True,
-            allow_methods=["*"],
-            allow_headers=["*"],
-            expose_headers=["*"],
-        )
-
-        return self.api_app
-
-    def serve(
-        self,
-        app: Union[str, FastAPI],
-        *,
-        scheme: str = "http",
-        host: str = "localhost",
-        port: int = 7777,
-        reload: bool = False,
-        prefix="/v1",
-        **kwargs,
-    ):
-        import uvicorn
-
-        logger.info(f"Starting playground on {scheme}://{host}:{port}")
-        # Encode the full endpoint (host:port)
-        encoded_endpoint = quote(f"{host}:{port}{prefix}")
-        self.endpoints_created = PlaygroundEndpointCreate(
-            endpoint=f"{scheme}://{host}:{port}", playground_data={"prefix": prefix}
-        )
-
-        # Create a panel with the playground URL
-        url = f"{agno_cli_settings.playground_url}?endpoint={encoded_endpoint}"
-        panel = Panel(
-            f"[bold green]Playground URL:[/bold green] [link={url}]{url}[/link]",
-            title="Agent Playground",
-            expand=False,
-            border_style="cyan",
-            box=box.HEAVY,
-            padding=(2, 2),
-        )
-
-        # Print the panel
-        console.print(panel)
-        self.set_app_id()
-
-        self.register_app_on_platform()
-
-        uvicorn.run(app=app, host=host, port=port, reload=reload, **kwargs)
-
-    def register_app_on_platform(self) -> None:
-        self._set_monitoring()
-        if not self.monitoring:
-            return
-
-        from agno.api.app import AppCreate, create_app
-
-        try:
-            log_debug(f"Creating app on Platform: {self.name}, {self.app_id}")
-            create_app(app=AppCreate(name=self.name, app_id=self.app_id, config=self.to_dict()))
-        except Exception as e:
-            log_debug(f"Could not create Agent app: {e}")
-        log_debug(f"Agent app created: {self.name}, {self.app_id}")
-
-    def to_dict(self) -> Dict[str, Any]:
-        payload = {
-            "endpointData": self.endpoints_created.model_dump(exclude_none=True) if self.endpoints_created else {},
-            "type": "playground",
-            "description": self.description,
-        }
-        payload = {k: v for k, v in payload.items() if v is not None}
-        return payload