agno 2.2.13__py3-none-any.whl

agno/db/surrealdb/utils.py

@@ -0,0 +1,147 @@
+import dataclasses
+from typing import Any, Dict, Optional, Sequence, TypeVar, Union, cast
+
+from surrealdb import BlockingHttpSurrealConnection, BlockingWsSurrealConnection, Surreal
+
+from agno.db.schemas.culture import CulturalKnowledge
+from agno.utils.log import logger
+
+RecordType = TypeVar("RecordType")
+
+
+def build_client(
+    url: str, creds: dict[str, str], ns: str, db: str
+) -> Union[BlockingWsSurrealConnection, BlockingHttpSurrealConnection]:
+    client = Surreal(url=url)
+    client.signin(creds)
+    client.use(namespace=ns, database=db)
+    return client
+
+
+def _query_aux(
+    client: Union[BlockingWsSurrealConnection, BlockingHttpSurrealConnection],
+    query: str,
+    vars: dict[str, Any],
+) -> Union[list, dict, str, int]:
+    try:
+        response = client.query(query, vars)
+    except Exception as e:
+        msg = f"!! Query execution error: {query} with {vars}, Error: {e}"
+        logger.error(msg)
+        raise RuntimeError(msg)
+    return response
+
+
+def query(
+    client: Union[BlockingWsSurrealConnection, BlockingHttpSurrealConnection],
+    query: str,
+    vars: dict[str, Any],
+    record_type: type[RecordType],
+) -> Sequence[RecordType]:
+    response = _query_aux(client, query, vars)
+    if isinstance(response, list):
+        if dataclasses.is_dataclass(record_type) and hasattr(record_type, "from_dict"):
+            return [getattr(record_type, "from_dict").__call__(x) for x in response]
+        else:
+            result: list[RecordType] = []
+            for x in response:
+                if isinstance(x, dict):
+                    result.append(record_type(**x))
+                else:
+                    result.append(record_type.__call__(x))
+            return result
+    else:
+        raise ValueError(f"Unexpected response type: {type(response)}")
+
+
+def query_one(
+    client: Union[BlockingWsSurrealConnection, BlockingHttpSurrealConnection],
+    query: str,
+    vars: dict[str, Any],
+    record_type: type[RecordType],
+) -> Optional[RecordType]:
+    response = _query_aux(client, query, vars)
+    if response is None:
+        return None
+    elif not isinstance(response, list):
+        if dataclasses.is_dataclass(record_type) and hasattr(record_type, "from_dict"):
+            return getattr(record_type, "from_dict").__call__(response)
+        elif isinstance(response, dict):
+            return record_type(**response)
+        else:
+            return record_type.__call__(response)
+    elif isinstance(response, list):
+        # Handle list responses - SurrealDB might return a list with a single element
+        if len(response) == 1 and isinstance(response[0], dict):
+            result = response[0]
+            if dataclasses.is_dataclass(record_type) and hasattr(record_type, "from_dict"):
+                return getattr(record_type, "from_dict").__call__(result)
+            elif record_type is dict:
+                return cast(RecordType, result)
+            else:
+                return record_type(**result)
+        elif len(response) == 0:
+            return None
+        else:
+            raise ValueError(f"Expected single record, got {len(response)} records: {response}")
+    else:
+        raise ValueError(f"Unexpected response type: {type(response)}")
+
+
+# -- Cultural Knowledge util methods --
+
+
+def serialize_cultural_knowledge_for_db(cultural_knowledge: CulturalKnowledge) -> Dict[str, Any]:
+    """Serialize a CulturalKnowledge object for database storage.
+
+    Converts the model's separate content, categories, and notes fields
+    into a single dict for the database content field.
+
+    Args:
+        cultural_knowledge (CulturalKnowledge): The cultural knowledge object to serialize.
+
+    Returns:
+        Dict[str, Any]: A dictionary with content, categories, and notes.
+    """
+    content_dict: Dict[str, Any] = {}
+    if cultural_knowledge.content is not None:
+        content_dict["content"] = cultural_knowledge.content
+    if cultural_knowledge.categories is not None:
+        content_dict["categories"] = cultural_knowledge.categories
+    if cultural_knowledge.notes is not None:
+        content_dict["notes"] = cultural_knowledge.notes
+
+    return content_dict if content_dict else {}
+
+
+def deserialize_cultural_knowledge_from_db(db_row: Dict[str, Any]) -> CulturalKnowledge:
+    """Deserialize a database row to a CulturalKnowledge object.
+
+    The database stores content as a dict containing content, categories, and notes.
+    This method extracts those fields and converts them back to the model format.
+
+    Args:
+        db_row (Dict[str, Any]): The database row as a dictionary.
+
+    Returns:
+        CulturalKnowledge: The cultural knowledge object.
+    """
+    # Extract content, categories, and notes from the content field
+    content_json = db_row.get("content", {}) or {}
+
+    return CulturalKnowledge.from_dict(
+        {
+            "id": db_row.get("id"),
+            "name": db_row.get("name"),
+            "summary": db_row.get("summary"),
+            "content": content_json.get("content"),
+            "categories": content_json.get("categories"),
+            "notes": content_json.get("notes"),
+            "metadata": db_row.get("metadata"),
+            "input": db_row.get("input"),
+            "created_at": db_row.get("created_at"),
+            "updated_at": db_row.get("updated_at"),
+            "agent_id": db_row.get("agent_id"),
+            "team_id": db_row.get("team_id"),
+        }
+    )

agno/db/utils.py

@@ -0,0 +1,116 @@
+"""Logic shared across different database implementations"""
+
+import json
+from datetime import date, datetime
+from uuid import UUID
+
+from agno.models.message import Message
+from agno.models.metrics import Metrics
+
+
+class CustomJSONEncoder(json.JSONEncoder):
+    """Custom encoder to handle non JSON serializable types."""
+
+    def default(self, obj):
+        if isinstance(obj, UUID):
+            return str(obj)
+        elif isinstance(obj, (date, datetime)):
+            return obj.isoformat()
+        elif isinstance(obj, Message):
+            return obj.to_dict()
+        elif isinstance(obj, Metrics):
+            return obj.to_dict()
+
+        return super().default(obj)
+
+
+def serialize_session_json_fields(session: dict) -> dict:
+    """Serialize all JSON fields in the given Session dictionary.
+
+    Args:
+        data (dict): The dictionary to serialize JSON fields in.
+
+    Returns:
+        dict: The dictionary with JSON fields serialized.
+    """
+    if session.get("session_data") is not None:
+        session["session_data"] = json.dumps(session["session_data"])
+    if session.get("agent_data") is not None:
+        session["agent_data"] = json.dumps(session["agent_data"])
+    if session.get("team_data") is not None:
+        session["team_data"] = json.dumps(session["team_data"])
+    if session.get("workflow_data") is not None:
+        session["workflow_data"] = json.dumps(session["workflow_data"])
+    if session.get("metadata") is not None:
+        session["metadata"] = json.dumps(session["metadata"])
+    if session.get("chat_history") is not None:
+        session["chat_history"] = json.dumps(session["chat_history"])
+    if session.get("summary") is not None:
+        session["summary"] = json.dumps(session["summary"], cls=CustomJSONEncoder)
+    if session.get("runs") is not None:
+        session["runs"] = json.dumps(session["runs"], cls=CustomJSONEncoder)
+
+    return session
+
+
+def deserialize_session_json_fields(session: dict) -> dict:
+    """Deserialize JSON fields in the given Session dictionary.
+
+    Args:
+        session (dict): The dictionary to deserialize.
+
+    Returns:
+        dict: The dictionary with JSON string fields deserialized to objects.
+    """
+    from agno.utils.log import log_warning
+
+    if session.get("agent_data") is not None and isinstance(session["agent_data"], str):
+        try:
+            session["agent_data"] = json.loads(session["agent_data"])
+        except (json.JSONDecodeError, TypeError) as e:
+            log_warning(f"Warning: Could not parse agent_data as JSON, keeping as string: {e}")
+
+    if session.get("team_data") is not None and isinstance(session["team_data"], str):
+        try:
+            session["team_data"] = json.loads(session["team_data"])
+        except (json.JSONDecodeError, TypeError) as e:
+            log_warning(f"Warning: Could not parse team_data as JSON, keeping as string: {e}")
+
+    if session.get("workflow_data") is not None and isinstance(session["workflow_data"], str):
+        try:
+            session["workflow_data"] = json.loads(session["workflow_data"])
+        except (json.JSONDecodeError, TypeError) as e:
+            log_warning(f"Warning: Could not parse workflow_data as JSON, keeping as string: {e}")
+
+    if session.get("metadata") is not None and isinstance(session["metadata"], str):
+        try:
+            session["metadata"] = json.loads(session["metadata"])
+        except (json.JSONDecodeError, TypeError) as e:
+            log_warning(f"Warning: Could not parse metadata as JSON, keeping as string: {e}")
+
+    if session.get("chat_history") is not None and isinstance(session["chat_history"], str):
+        try:
+            session["chat_history"] = json.loads(session["chat_history"])
+        except (json.JSONDecodeError, TypeError) as e:
+            log_warning(f"Warning: Could not parse chat_history as JSON, keeping as string: {e}")
+
+    if session.get("summary") is not None and isinstance(session["summary"], str):
+        try:
+            session["summary"] = json.loads(session["summary"])
+        except (json.JSONDecodeError, TypeError) as e:
+            log_warning(f"Warning: Could not parse summary as JSON, keeping as string: {e}")
+
+    if session.get("session_data") is not None and isinstance(session["session_data"], str):
+        try:
+            session["session_data"] = json.loads(session["session_data"])
+        except (json.JSONDecodeError, TypeError) as e:
+            log_warning(f"Warning: Could not parse session_data as JSON, keeping as string: {e}")
+
+    # Handle runs field with session type checking
+    if session.get("runs") is not None and isinstance(session["runs"], str):
+        try:
+            session["runs"] = json.loads(session["runs"])
+        except (json.JSONDecodeError, TypeError) as e:
+            log_warning(f"Warning: Could not parse runs as JSON, keeping as string: {e}")
+
+    return session

agno/debug.py

@@ -0,0 +1,18 @@
+def enable_debug_mode() -> None:
+    """Enable debug mode for the agno library.
+
+    This function sets the logging level to DEBUG
+    """
+    from agno.utils.log import set_log_level_to_debug
+
+    set_log_level_to_debug()
+
+
+def disable_debug_mode() -> None:
+    """Disable debug mode for the agno library.
+
+    This function resets the logging level to INFO
+    """
+    from agno.utils.log import set_log_level_to_info
+
+    set_log_level_to_info()

agno/eval/__init__.py

@@ -0,0 +1,14 @@
+from agno.eval.accuracy import AccuracyAgentResponse, AccuracyEval, AccuracyEvaluation, AccuracyResult
+from agno.eval.performance import PerformanceEval, PerformanceResult
+from agno.eval.reliability import ReliabilityEval, ReliabilityResult
+
+__all__ = [
+    "AccuracyAgentResponse",
+    "AccuracyEvaluation",
+    "AccuracyResult",
+    "AccuracyEval",
+    "PerformanceEval",
+    "PerformanceResult",
+    "ReliabilityEval",
+    "ReliabilityResult",
+]