agno 2.2.13__py3-none-any.whl

agno/utils/pickle.py

@@ -0,0 +1,32 @@
+from pathlib import Path
+from typing import Any, Optional
+
+from agno.utils.log import logger
+
+
+def pickle_object_to_file(obj: Any, file_path: Path) -> Any:
+    """Pickles and saves object to file_path"""
+    import pickle
+
+    _obj_parent = file_path.parent
+    if not _obj_parent.exists():
+        _obj_parent.mkdir(parents=True, exist_ok=True)
+    pickle.dump(obj, file_path.open("wb"))
+
+
+def unpickle_object_from_file(file_path: Path, verify_class: Optional[Any] = None) -> Any:
+    """Reads the contents of file_path and unpickles the binary content into an object.
+    If verify_class is provided, checks if the object is an instance of that class.
+    """
+    import pickle
+
+    _obj = None
+    # log_debug(f"Reading {file_path}")
+    if file_path.exists() and file_path.is_file():
+        _obj = pickle.load(file_path.open("rb"))
+
+    if _obj and verify_class and not isinstance(_obj, verify_class):
+        logger.warning(f"Object does not match {verify_class}")
+        _obj = None
+
+    return _obj

agno/utils/pprint.py

@@ -0,0 +1,178 @@
+import json
+from typing import AsyncIterable, Iterable, Union, get_args
+
+from pydantic import BaseModel
+
+from agno.run.agent import RunOutput, RunOutputEvent
+from agno.run.team import TeamRunOutput, TeamRunOutputEvent
+from agno.run.workflow import WorkflowRunOutput, WorkflowRunOutputEvent
+from agno.utils.log import logger
+from agno.utils.timer import Timer
+
+
+def pprint_run_response(
+    run_response: Union[
+        RunOutput,
+        Iterable[RunOutputEvent],
+        TeamRunOutput,
+        Iterable[TeamRunOutputEvent],
+        WorkflowRunOutput,
+        Iterable[WorkflowRunOutputEvent],
+    ],
+    markdown: bool = False,
+    show_time: bool = False,
+) -> None:
+    from rich.box import ROUNDED
+    from rich.console import Console
+    from rich.json import JSON
+    from rich.live import Live
+    from rich.markdown import Markdown
+    from rich.status import Status
+    from rich.table import Table
+
+    console = Console()
+
+    # If run_response is a single RunOutput, wrap it in a list to make it iterable
+    if (
+        isinstance(run_response, RunOutput)
+        or isinstance(run_response, TeamRunOutput)
+        or isinstance(run_response, WorkflowRunOutput)
+    ):
+        single_response_content: Union[str, JSON, Markdown] = ""
+        if isinstance(run_response.content, str):
+            single_response_content = (
+                Markdown(run_response.content) if markdown else run_response.get_content_as_string(indent=4)
+            )
+        elif isinstance(run_response.content, BaseModel):
+            try:
+                single_response_content = JSON(run_response.content.model_dump_json(exclude_none=True), indent=2)
+            except Exception as e:
+                logger.warning(f"Failed to convert response to Markdown: {e}")
+        else:
+            try:
+                single_response_content = JSON(json.dumps(run_response.content), indent=4)
+            except Exception as e:
+                logger.warning(f"Failed to convert response to string: {e}")
+
+        table = Table(box=ROUNDED, border_style="blue", show_header=False)
+        table.add_row(single_response_content)
+        console.print(table)
+    else:
+        streaming_response_content: str = ""
+        with Live(console=console) as live_log:
+            status = Status("Working...", spinner="dots")
+            live_log.update(status)
+            response_timer = Timer()
+            response_timer.start()
+            for resp in run_response:
+                if (
+                    (
+                        isinstance(resp, tuple(get_args(RunOutputEvent)))
+                        or isinstance(resp, tuple(get_args(TeamRunOutputEvent)))
+                        or isinstance(resp, tuple(get_args(WorkflowRunOutputEvent)))
+                    )
+                    and hasattr(resp, "content")
+                    and resp.content is not None
+                ):
+                    if isinstance(resp.content, BaseModel):
+                        try:
+                            JSON(resp.content.model_dump_json(exclude_none=True), indent=2)  # type: ignore
+                        except Exception as e:
+                            logger.warning(f"Failed to convert response to Markdown: {e}")
+                    else:
+                        if isinstance(streaming_response_content, JSON):
+                            streaming_response_content = streaming_response_content.text + "\n"  # type: ignore
+                        streaming_response_content += resp.content  # type: ignore
+
+                formatted_response = Markdown(streaming_response_content) if markdown else streaming_response_content  # type: ignore
+                table = Table(box=ROUNDED, border_style="blue", show_header=False)
+                if show_time:
+                    table.add_row(f"Response\n({response_timer.elapsed:.1f}s)", formatted_response)  # type: ignore
+                else:
+                    table.add_row(formatted_response)  # type: ignore
+                live_log.update(table)
+            response_timer.stop()
+
+
+async def apprint_run_response(
+    run_response: Union[
+        RunOutput,
+        AsyncIterable[RunOutput],
+        TeamRunOutput,
+        AsyncIterable[TeamRunOutput],
+        WorkflowRunOutput,
+        AsyncIterable[WorkflowRunOutputEvent],
+    ],
+    markdown: bool = False,
+    show_time: bool = False,
+) -> None:
+    from rich.box import ROUNDED
+    from rich.console import Console
+    from rich.json import JSON
+    from rich.live import Live
+    from rich.markdown import Markdown
+    from rich.status import Status
+    from rich.table import Table
+
+    console = Console()
+    # If run_response is a single RunOutput, wrap it in a list to make it iterable
+    if (
+        isinstance(run_response, RunOutput)
+        or isinstance(run_response, TeamRunOutput)
+        or isinstance(run_response, WorkflowRunOutput)
+    ):
+        single_response_content: Union[str, JSON, Markdown] = ""
+        if isinstance(run_response.content, str):
+            single_response_content = (
+                Markdown(run_response.content) if markdown else run_response.get_content_as_string(indent=4)
+            )
+        elif isinstance(run_response.content, BaseModel):
+            try:
+                single_response_content = JSON(run_response.content.model_dump_json(exclude_none=True), indent=2)
+            except Exception as e:
+                logger.warning(f"Failed to convert response to Markdown: {e}")
+        else:
+            try:
+                single_response_content = JSON(json.dumps(run_response.content), indent=4)
+            except Exception as e:
+                logger.warning(f"Failed to convert response to string: {e}")
+
+        table = Table(box=ROUNDED, border_style="blue", show_header=False)
+        table.add_row(single_response_content)
+        console.print(table)
+    else:
+        streaming_response_content: str = ""
+        with Live(console=console) as live_log:
+            status = Status("Working...", spinner="dots")
+            live_log.update(status)
+            response_timer = Timer()
+            response_timer.start()
+
+            async for resp in run_response:
+                if (
+                    (
+                        isinstance(resp, tuple(get_args(RunOutputEvent)))
+                        or isinstance(resp, tuple(get_args(TeamRunOutputEvent)))
+                        or isinstance(resp, tuple(get_args(WorkflowRunOutputEvent)))
+                    )
+                    and hasattr(resp, "content")
+                    and resp.content is not None
+                ):
+                    if isinstance(resp.content, BaseModel):
+                        try:
+                            streaming_response_content = JSON(resp.content.model_dump_json(exclude_none=True), indent=2)  # type: ignore
+                        except Exception as e:
+                            logger.warning(f"Failed to convert response to Markdown: {e}")
+                    else:
+                        if isinstance(streaming_response_content, JSON):
+                            streaming_response_content = streaming_response_content.text + "\n"  # type: ignore
+                        streaming_response_content += resp.content  # type: ignore
+
+                formatted_response = Markdown(streaming_response_content) if markdown else streaming_response_content  # type: ignore
+                table = Table(box=ROUNDED, border_style="blue", show_header=False)
+                if show_time:
+                    table.add_row(f"Response\n({response_timer.elapsed:.1f}s)", formatted_response)  # type: ignore
+                else:
+                    table.add_row(formatted_response)  # type: ignore
+                live_log.update(table)
+            response_timer.stop()