agno 1.7.2__py3-none-any.whl → 1.7.4__py3-none-any.whl

agno/tools/scrapegraph.py

@@ -45,7 +45,7 @@ class ScrapeGraphTools(Toolkit):
         super().__init__(name="scrapegraph_tools", tools=tools, **kwargs)
 
     def smartscraper(self, url: str, prompt: str) -> str:
-        """Use this function to extract structured data from a webpage using LLM.
+        """Extract structured data from a webpage using LLM.
         Args:
             url (str): The URL to scrape
             prompt (str): Natural language prompt describing what to extract
@@ -59,7 +59,7 @@ class ScrapeGraphTools(Toolkit):
             return json.dumps({"error": str(e)})
 
     def markdownify(self, url: str) -> str:
-        """Use this function to convert a webpage to markdown format.
+        """Convert a webpage to markdown format.
         Args:
             url (str): The URL to convert
         Returns:
@@ -82,7 +82,7 @@ class ScrapeGraphTools(Toolkit):
         same_domain_only: bool = True,
         batch_size: int = 1,
     ) -> str:
-        """Use this function to crawl a website and extract structured data using a schema.
+        """Crawl a website and extract structured data
         Args:
             url (str): The URL to crawl
             prompt (str): Natural language prompt describing what to extract
@@ -110,17 +110,15 @@ class ScrapeGraphTools(Toolkit):
         except Exception as e:
             return json.dumps({"error": str(e)})
 
-    def searchscraper(self, url: str, prompt: str) -> str:
-        """Use this function to search and extract information from a webpage using LLM.
+    def searchscraper(self, prompt: str) -> str:
+        """Search the web and extract information from the web.
         Args:
-            url (str): The URL to search
-            prompt (str): Natural language prompt describing what to search for
+            prompt (str): Search query
         Returns:
-            The search results extracted from the webpage
+            JSON of the search results
         """
         try:
-            response = self.client.searchscraper(website_url=url, user_prompt=prompt)
-            # If response has a 'result' attribute, return it, else return the whole response
+            response = self.client.searchscraper(user_prompt=prompt)
             if hasattr(response, "result"):
                 return json.dumps(response.result)
             elif isinstance(response, dict) and "result" in response:

agno/utils/log.py

@@ -7,6 +7,7 @@ from rich.text import Text
 
 LOGGER_NAME = "agno"
 TEAM_LOGGER_NAME = f"{LOGGER_NAME}-team"
+WORKFLOW_LOGGER_NAME = f"{LOGGER_NAME}-workflow"
 
 # Define custom styles for different log sources
 LOG_STYLES = {
@@ -18,6 +19,10 @@ LOG_STYLES = {
         "debug": "magenta",
         "info": "steel_blue1",
     },
+    "workflow": {
+        "debug": "sandy_brown",
+        "info": "orange3",
+    },
 }
 
 
@@ -92,6 +97,7 @@ def build_logger(logger_name: str, source_type: Optional[str] = None) -> Any:
 
 agent_logger: AgnoLogger = build_logger(LOGGER_NAME, source_type="agent")
 team_logger: AgnoLogger = build_logger(TEAM_LOGGER_NAME, source_type="team")
+workflow_logger: AgnoLogger = build_logger(WORKFLOW_LOGGER_NAME, source_type="workflow")
 
 # Set the default logger to the agent logger
 logger: AgnoLogger = agent_logger
@@ -146,6 +152,12 @@ def use_agent_logger():
     logger = agent_logger
 
 
+def use_workflow_logger():
+    """Switch the default logger to use workflow_logger"""
+    global logger
+    logger = workflow_logger
+
+
 def log_debug(msg, center: bool = False, symbol: str = "*", log_level: Literal[1, 2] = 1, *args, **kwargs):
     global logger
     global debug_on

agno/utils/message.py

@@ -1,13 +1,17 @@
 from typing import Dict, List, Union
 
+from pydantic import BaseModel
+
 from agno.models.message import Message
 
 
-def get_text_from_message(message: Union[List, Dict, str, Message]) -> str:
+def get_text_from_message(message: Union[List, Dict, str, Message, BaseModel]) -> str:
     """Return the user texts from the message"""
 
     if isinstance(message, str):
         return message
+    if isinstance(message, BaseModel):
+        return message.model_dump_json(indent=2, exclude_none=True)
     if isinstance(message, list):
         text_messages = []
         if len(message) == 0:

agno/utils/openai.py

@@ -93,11 +93,24 @@ def audio_to_message(audio: Sequence[Audio]) -> List[Dict[str, Any]]:
     return audio_messages
 
 
-def _process_bytes_image(image: bytes) -> Dict[str, Any]:
+def _process_bytes_image(image: bytes, image_format: Optional[str] = None) -> Dict[str, Any]:
     """Process bytes image data."""
     base64_image = base64.b64encode(image).decode("utf-8")
-    # Assuming JPEG if type not specified, could attempt detection
-    image_url = f"data:image/jpeg;base64,{base64_image}"
+
+    # Use provided format or attempt detection, defaulting to JPEG
+    if image_format:
+        mime_type = f"image/{image_format.lower()}"
+    else:
+        # Try to detect the image format from the bytes
+        try:
+            import imghdr
+
+            detected_format = imghdr.what(None, h=image)
+            mime_type = f"image/{detected_format}" if detected_format else "image/jpeg"
+        except Exception:
+            mime_type = "image/jpeg"
+
+    image_url = f"data:{mime_type};base64,{base64_image}"
     return {"type": "image_url", "image_url": {"url": image_url}}
 
 
@@ -141,7 +154,8 @@ def process_image(image: Image) -> Optional[Dict[str, Any]]:
             image_payload = _process_image_path(image.filepath)
 
         elif image.content is not None:
-            image_payload = _process_bytes_image(image.content)
+            # Pass the format from the Image object
+            image_payload = _process_bytes_image(image.content, image.format)
 
         else:
             log_warning(f"Unsupported image format or no data provided: {image}")
@@ -150,7 +164,8 @@ def process_image(image: Image) -> Optional[Dict[str, Any]]:
         if image_payload and image.detail:  # Check if payload was created before adding detail
             # Ensure image_url key exists before trying to access its sub-dictionary
             if "image_url" not in image_payload:
-                image_payload["image_url"] = {}  # Initialize if missing (though unlikely based on helper funcs)
+                # Initialize if missing (though unlikely based on helper funcs)
+                image_payload["image_url"] = {}
             image_payload["image_url"]["detail"] = image.detail
 
         return image_payload

agno/utils/pprint.py

@@ -5,13 +5,20 @@ from pydantic import BaseModel
 
 from agno.run.response import RunResponse, RunResponseEvent
 from agno.run.team import TeamRunResponse, TeamRunResponseEvent
-from agno.run.workflow import WorkflowRunResponseEvent
+from agno.run.v2.workflow import WorkflowRunResponse, WorkflowRunResponseEvent
 from agno.utils.log import logger
 from agno.utils.timer import Timer
 
 
 def pprint_run_response(
-    run_response: Union[RunResponse, Iterable[RunResponseEvent], TeamRunResponse, Iterable[TeamRunResponseEvent]],
+    run_response: Union[
+        RunResponse,
+        Iterable[RunResponseEvent],
+        TeamRunResponse,
+        Iterable[TeamRunResponseEvent],
+        WorkflowRunResponse,
+        Iterable[WorkflowRunResponseEvent],
+    ],
     markdown: bool = False,
     show_time: bool = False,
 ) -> None:
@@ -25,7 +32,11 @@ def pprint_run_response(
     from agno.cli.console import console
 
     # If run_response is a single RunResponse, wrap it in a list to make it iterable
-    if isinstance(run_response, RunResponse) or isinstance(run_response, TeamRunResponse):
+    if (
+        isinstance(run_response, RunResponse)
+        or isinstance(run_response, TeamRunResponse)
+        or isinstance(run_response, WorkflowRunResponse)
+    ):
         single_response_content: Union[str, JSON, Markdown] = ""
         if isinstance(run_response.content, str):
             single_response_content = (
@@ -64,11 +75,13 @@ def pprint_run_response(
                 ):
                     if isinstance(resp.content, BaseModel):
                         try:
-                            streaming_response_content = JSON(resp.content.model_dump_json(exclude_none=True), indent=2)  # type: ignore
+                            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:
-                        streaming_response_content += resp.content
+                        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)
@@ -81,7 +94,14 @@ def pprint_run_response(
 
 
 async def apprint_run_response(
-    run_response: Union[RunResponse, AsyncIterable[RunResponse], TeamRunResponse, AsyncIterable[TeamRunResponse]],
+    run_response: Union[
+        RunResponse,
+        AsyncIterable[RunResponse],
+        TeamRunResponse,
+        AsyncIterable[TeamRunResponse],
+        WorkflowRunResponse,
+        AsyncIterable[WorkflowRunResponseEvent],
+    ],
     markdown: bool = False,
     show_time: bool = False,
 ) -> None:
@@ -95,7 +115,11 @@ async def apprint_run_response(
     from agno.cli.console import console
 
     # If run_response is a single RunResponse, wrap it in a list to make it iterable
-    if isinstance(run_response, RunResponse) or isinstance(run_response, TeamRunResponse):
+    if (
+        isinstance(run_response, RunResponse)
+        or isinstance(run_response, TeamRunResponse)
+        or isinstance(run_response, WorkflowRunResponse)
+    ):
         single_response_content: Union[str, JSON, Markdown] = ""
         if isinstance(run_response.content, str):
             single_response_content = (
@@ -139,7 +163,7 @@ async def apprint_run_response(
                         except Exception as e:
                             logger.warning(f"Failed to convert response to Markdown: {e}")
                     else:
-                        streaming_response_content += resp.content
+                        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)

agno/workflow/v2/__init__.py

@@ -0,0 +1,21 @@
+from agno.workflow.v2.condition import Condition
+from agno.workflow.v2.loop import Loop
+from agno.workflow.v2.parallel import Parallel
+from agno.workflow.v2.router import Router
+from agno.workflow.v2.step import Step
+from agno.workflow.v2.steps import Steps
+from agno.workflow.v2.types import StepInput, StepOutput, WorkflowExecutionInput
+from agno.workflow.v2.workflow import Workflow
+
+__all__ = [
+    "Workflow",
+    "Steps",
+    "Step",
+    "Loop",
+    "Parallel",
+    "Condition",
+    "Router",
+    "WorkflowExecutionInput",
+    "StepInput",
+    "StepOutput",
+]