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

agno/storage/redis.py

@@ -1,336 +0,0 @@
-import json
-import time
-from dataclasses import asdict
-from typing import List, Literal, Optional
-from uuid import UUID
-
-from agno.storage.base import Storage
-from agno.storage.session import Session
-from agno.storage.session.agent import AgentSession
-from agno.storage.session.team import TeamSession
-from agno.storage.session.v2.workflow import WorkflowSession as WorkflowSessionV2
-from agno.storage.session.workflow import WorkflowSession
-from agno.utils.log import log_debug, log_info, logger
-
-try:
-    from redis import ConnectionError, Redis
-except ImportError:
-    raise ImportError("`redis` not installed. Please install it using `pip install redis`")
-
-
-class UUIDEncoder(json.JSONEncoder):
-    def default(self, obj):
-        if isinstance(obj, UUID):
-            return str(obj)
-        return super().default(obj)
-
-
-class RedisStorage(Storage):
-    def __init__(
-        self,
-        prefix: str,
-        host: str = "localhost",
-        port: int = 6379,
-        db: int = 0,
-        password: Optional[str] = None,
-        mode: Optional[Literal["agent", "team", "workflow", "workflow_v2"]] = "agent",
-        ssl: Optional[bool] = False,
-        expire: Optional[int] = None,
-    ):
-        """
-        Initialize Redis storage for sessions.
-
-        Args:
-            prefix (str): Prefix for Redis keys to namespace the sessions
-            host (str): Redis host address
-            port (int): Redis port number
-            db (int): Redis database number
-            password (Optional[str]): Redis password if authentication is required
-            mode (Optional[Literal["agent", "team", "workflow", "workflow_v2"]]): Storage mode
-            ssl (Optional[bool]): Whether to use SSL for Redis connection
-            expire (Optional[int]): TTL (time to live) in seconds for Redis keys. None means no expiration.
-        """
-        super().__init__(mode)
-        self.prefix = prefix
-        self.expire = expire
-        self.redis_client = Redis(
-            host=host,
-            port=port,
-            db=db,
-            password=password,
-            decode_responses=True,  # Automatically decode responses to str
-            ssl=ssl,
-        )
-        log_debug(f"Created RedisStorage with prefix: '{self.prefix}'")
-
-    def _get_key(self, session_id: str) -> str:
-        """Generate Redis key for a session."""
-        return f"{self.prefix}:{session_id}"
-
-    def serialize(self, data: dict) -> str:
-        """Serialize data to JSON string."""
-        return json.dumps(data, ensure_ascii=False, cls=UUIDEncoder)
-
-    def deserialize(self, data: str) -> dict:
-        """Deserialize JSON string to dict."""
-        return json.loads(data)
-
-    def create(self) -> None:
-        """
-        Create storage if it doesn't exist.
-        For Redis, we don't need to create anything as it's schema-less.
-        """
-        # Test connection
-        try:
-            self.redis_client.ping()
-            log_debug("Redis connection successful")
-        except ConnectionError as e:
-            logger.error(f"Could not connect to Redis: {e}")
-            raise
-
-    def read(self, session_id: str, user_id: Optional[str] = None) -> Optional[Session]:
-        """Read a Session from Redis."""
-        try:
-            data = self.redis_client.get(self._get_key(session_id))
-            if data is None:
-                return None
-
-            session_data = self.deserialize(data)  # type: ignore
-            if user_id and session_data.get("user_id") != user_id:
-                return None
-
-            if self.mode == "agent":
-                return AgentSession.from_dict(session_data)
-            elif self.mode == "team":
-                return TeamSession.from_dict(session_data)
-            elif self.mode == "workflow":
-                return WorkflowSession.from_dict(session_data)
-            elif self.mode == "workflow_v2":
-                return WorkflowSessionV2.from_dict(session_data)
-
-        except Exception as e:
-            logger.error(f"Error reading session: {e}")
-            return None
-
-    def get_all_session_ids(self, user_id: Optional[str] = None, entity_id: Optional[str] = None) -> List[str]:
-        """Get all session IDs, optionally filtered by user_id and/or entity_id."""
-        session_ids = []
-        try:
-            # Get all keys matching the prefix pattern
-            pattern = f"{self.prefix}:*"
-            for key in self.redis_client.scan_iter(match=pattern):
-                data = self.deserialize(self.redis_client.get(key))  # type: ignore
-
-                if user_id or entity_id:
-                    if user_id and entity_id:
-                        if self.mode == "agent" and data["agent_id"] == entity_id and data["user_id"] == user_id:
-                            session_ids.append(data["session_id"])
-                        elif self.mode == "team" and data["team_id"] == entity_id and data["user_id"] == user_id:
-                            session_ids.append(data["session_id"])
-                        elif (
-                            self.mode == "workflow" and data["workflow_id"] == entity_id and data["user_id"] == user_id
-                        ):
-                            session_ids.append(data["session_id"])
-                        elif self.mode == "workflow_v2" and data["workflow_id"] == entity_id:
-                            session_ids.append(data["session_id"])
-                    elif user_id and data["user_id"] == user_id:
-                        session_ids.append(data["session_id"])
-                    elif entity_id:
-                        if self.mode == "agent" and data["agent_id"] == entity_id:
-                            session_ids.append(data["session_id"])
-                        elif self.mode == "team" and data["team_id"] == entity_id:
-                            session_ids.append(data["session_id"])
-                        elif self.mode == "workflow" and data["workflow_id"] == entity_id:
-                            session_ids.append(data["session_id"])
-                        elif self.mode == "workflow_v2" and data["workflow_id"] == entity_id:
-                            session_ids.append(data["session_id"])
-                else:
-                    # No filters applied, add all session_ids
-                    session_ids.append(data["session_id"])
-
-        except Exception as e:
-            logger.error(f"Error getting session IDs: {e}")
-
-        return session_ids
-
-    def get_all_sessions(self, user_id: Optional[str] = None, entity_id: Optional[str] = None) -> List[Session]:
-        """Get all sessions, optionally filtered by user_id and/or entity_id."""
-        sessions: List[Session] = []
-        try:
-            pattern = f"{self.prefix}:*"
-            for key in self.redis_client.scan_iter(match=pattern):
-                data = self.deserialize(self.redis_client.get(key))  # type: ignore
-
-                if user_id or entity_id:
-                    _session: Optional[Session] = None
-
-                    if user_id and entity_id:
-                        if self.mode == "agent" and data["agent_id"] == entity_id and data["user_id"] == user_id:
-                            _session = AgentSession.from_dict(data)
-                        elif self.mode == "team" and data["team_id"] == entity_id and data["user_id"] == user_id:
-                            _session = TeamSession.from_dict(data)
-                        elif (
-                            self.mode == "workflow" and data["workflow_id"] == entity_id and data["user_id"] == user_id
-                        ):
-                            _session = WorkflowSession.from_dict(data)
-                        elif self.mode == "workflow_v2" and data["workflow_id"] == entity_id:
-                            _session = WorkflowSessionV2.from_dict(data)
-                    elif user_id and data["user_id"] == user_id:
-                        if self.mode == "agent":
-                            _session = AgentSession.from_dict(data)
-                        elif self.mode == "team":
-                            _session = TeamSession.from_dict(data)
-                        elif self.mode == "workflow":
-                            _session = WorkflowSession.from_dict(data)
-                        elif self.mode == "workflow_v2" and data["workflow_id"] == entity_id:
-                            _session = WorkflowSessionV2.from_dict(data)
-                    elif entity_id:
-                        if self.mode == "agent" and data["agent_id"] == entity_id:
-                            _session = AgentSession.from_dict(data)
-                        elif self.mode == "team" and data["team_id"] == entity_id:
-                            _session = TeamSession.from_dict(data)
-                        elif self.mode == "workflow" and data["workflow_id"] == entity_id:
-                            _session = WorkflowSession.from_dict(data)
-                        elif self.mode == "workflow_v2" and data["workflow_id"] == entity_id:
-                            _session = WorkflowSessionV2.from_dict(data)
-
-                    if _session:
-                        sessions.append(_session)
-                else:
-                    _session = None
-                    # No filters applied, add all sessions
-                    if self.mode == "agent":
-                        _session = AgentSession.from_dict(data)
-                    elif self.mode == "team":
-                        _session = TeamSession.from_dict(data)
-                    elif self.mode == "workflow":
-                        _session = WorkflowSession.from_dict(data)
-                    elif self.mode == "workflow_v2":
-                        _session = WorkflowSessionV2.from_dict(data)
-                    if _session:
-                        sessions.append(_session)
-
-        except Exception as e:
-            logger.error(f"Error getting all sessions: {e}")
-
-        return sessions
-
-    def get_recent_sessions(
-        self,
-        user_id: Optional[str] = None,
-        entity_id: Optional[str] = None,
-        limit: Optional[int] = 2,
-    ) -> List[Session]:
-        """Get the last N sessions, ordered by created_at descending.
-
-        Args:
-            num_history_sessions: Number of most recent sessions to return
-            user_id: Filter by user ID
-            entity_id: Filter by entity ID (agent_id, team_id, or workflow_id)
-
-        Returns:
-            List[Session]: List of most recent sessions
-        """
-        sessions: List[Session] = []
-        # List of (created_at, data) tuples for sorting
-        session_data: List[tuple[int, dict]] = []
-
-        try:
-            pattern = f"{self.prefix}:*"
-            for key in self.redis_client.scan_iter(match=pattern):
-                try:
-                    data = self.deserialize(self.redis_client.get(key))  # type: ignore
-
-                    # Apply filters
-                    if user_id and data["user_id"] != user_id:
-                        continue
-
-                    if entity_id:
-                        if self.mode == "agent" and data["agent_id"] != entity_id:
-                            continue
-                        elif self.mode == "team" and data["team_id"] != entity_id:
-                            continue
-                        elif self.mode == "workflow" and data["workflow_id"] != entity_id:
-                            continue
-                        elif self.mode == "workflow_v2" and data["workflow_id"] != entity_id:
-                            continue
-                    # Store with created_at for sorting
-                    created_at = data.get("created_at", 0)
-                    session_data.append((created_at, data))
-
-                except Exception as e:
-                    logger.error(f"Error processing session data: {e}")
-                    continue
-
-            # Sort by created_at descending and take only num_history_sessions
-            session_data.sort(key=lambda x: x[0], reverse=True)
-            if limit is not None:
-                session_data = session_data[:limit]
-
-            # Convert filtered and sorted data to Session objects
-            for _, data in session_data:
-                session: Optional[Session] = None
-                if self.mode == "agent":
-                    session = AgentSession.from_dict(data)
-                elif self.mode == "team":
-                    session = TeamSession.from_dict(data)
-                elif self.mode == "workflow":
-                    session = WorkflowSession.from_dict(data)
-                elif self.mode == "workflow_v2":
-                    session = WorkflowSessionV2.from_dict(data)
-                if session is not None:
-                    sessions.append(session)
-
-        except Exception as e:
-            logger.error(f"Error getting last {limit} sessions: {e}")
-
-        return sessions
-
-    def upsert(self, session: Session) -> Optional[Session]:
-        """Insert or update a Session in Redis."""
-        try:
-            if self.mode == "workflow_v2":
-                data = session.to_dict()
-            else:
-                data = asdict(session)
-            data["updated_at"] = int(time.time())
-            if "created_at" not in data or data["created_at"] is None:
-                data["created_at"] = data["updated_at"]
-
-            key = self._get_key(session.session_id)
-            if self.expire is not None:
-                self.redis_client.set(key, self.serialize(data), ex=self.expire)
-            else:
-                self.redis_client.set(key, self.serialize(data))
-            return session
-        except Exception as e:
-            logger.error(f"Error upserting session: {e}")
-            return None
-
-    def delete_session(self, session_id: Optional[str] = None):
-        """Delete a session from Redis."""
-        if session_id is None:
-            return
-        try:
-            key = self._get_key(session_id)
-            self.redis_client.delete(key)
-            log_debug(f"Deleted session: {session_id}")
-        except Exception as e:
-            logger.error(f"Error deleting session: {e}")
-
-    def drop(self) -> None:
-        """Drop all sessions from storage."""
-        try:
-            pattern = f"{self.prefix}:*"
-            for key in self.redis_client.scan_iter(match=pattern):
-                self.redis_client.delete(key)
-            log_info(f"Dropped all sessions with prefix: {self.prefix}")
-        except Exception as e:
-            logger.error(f"Error dropping sessions: {e}")
-
-    def upgrade_schema(self) -> None:
-        """
-        Upgrade the schema of the storage.
-        For Redis, this is a no-op as it's schema-less.
-        """
-        pass

agno/storage/session/__init__.py

@@ -1,16 +0,0 @@
-from typing import Union
-
-from agno.storage.session.agent import AgentSession
-from agno.storage.session.team import TeamSession
-from agno.storage.session.v2.workflow import WorkflowSession as WorkflowSessionV2
-from agno.storage.session.workflow import WorkflowSession
-
-Session = Union[AgentSession, TeamSession, WorkflowSession, WorkflowSessionV2]
-
-__all__ = [
-    "AgentSession",
-    "TeamSession",
-    "WorkflowSession",
-    "WorkflowSessionV2",
-    "Session",
-]

agno/storage/session/agent.py

@@ -1,64 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import asdict, dataclass
-from typing import Any, Dict, Mapping, Optional
-
-from agno.utils.log import logger
-
-
-@dataclass
-class AgentSession:
-    """Agent Session that is stored in the database"""
-
-    # Session UUID
-    session_id: str
-    # ID of the user interacting with this agent
-    user_id: Optional[str] = None
-    # ID of the team session this agent session is associated with
-    team_session_id: Optional[str] = None
-    # ID of the workflow session this agent session is associated with
-    workflow_session_id: Optional[str] = None
-    # Agent Memory
-    memory: Optional[Dict[str, Any]] = None
-    # Session Data: session_name, session_state, images, videos, audio
-    session_data: Optional[Dict[str, Any]] = None
-    # Extra Data stored with this agent
-    extra_data: Optional[Dict[str, Any]] = None
-    # The unix timestamp when this session was created
-    created_at: Optional[int] = None
-    # The unix timestamp when this session was last updated
-    updated_at: Optional[int] = None
-
-    # ID of the agent that this session is associated with
-    agent_id: Optional[str] = None
-    # Agent Data: agent_id, name and model
-    agent_data: Optional[Dict[str, Any]] = None
-
-    def to_dict(self) -> Dict[str, Any]:
-        return asdict(self)
-
-    def telemetry_data(self) -> Dict[str, Any]:
-        return {
-            "model": self.agent_data.get("model") if self.agent_data else None,
-            "created_at": self.created_at,
-            "updated_at": self.updated_at,
-        }
-
-    @classmethod
-    def from_dict(cls, data: Mapping[str, Any]) -> Optional[AgentSession]:
-        if data is None or data.get("session_id") is None:
-            logger.warning("AgentSession is missing session_id")
-            return None
-        return cls(
-            session_id=data.get("session_id"),  # type: ignore
-            agent_id=data.get("agent_id"),
-            team_session_id=data.get("team_session_id"),
-            workflow_session_id=data.get("workflow_session_id"),
-            user_id=data.get("user_id"),
-            memory=data.get("memory"),
-            agent_data=data.get("agent_data"),
-            session_data=data.get("session_data"),
-            extra_data=data.get("extra_data"),
-            created_at=data.get("created_at"),
-            updated_at=data.get("updated_at"),
-        )

agno/storage/session/team.py

@@ -1,63 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import asdict, dataclass
-from typing import Any, Dict, Mapping, Optional
-
-from agno.utils.log import logger
-
-
-@dataclass
-class TeamSession:
-    """Team Session that is stored in the database"""
-
-    # Session UUID
-    session_id: str
-    # ID of the team session this team session is associated with (so for sub-teams)
-    team_session_id: Optional[str] = None
-    # ID of the workflow session this team session is associated with
-    workflow_session_id: Optional[str] = None
-    # ID of the team that this session is associated with
-    team_id: Optional[str] = None
-    # ID of the user interacting with this team
-    user_id: Optional[str] = None
-    # Team Memory
-    memory: Optional[Dict[str, Any]] = None
-    # Team Data: agent_id, name and model
-    team_data: Optional[Dict[str, Any]] = None
-    # Session Data: session_name, session_state, images, videos, audio
-    session_data: Optional[Dict[str, Any]] = None
-    # Extra Data stored with this agent
-    extra_data: Optional[Dict[str, Any]] = None
-    # The unix timestamp when this session was created
-    created_at: Optional[int] = None
-    # The unix timestamp when this session was last updated
-    updated_at: Optional[int] = None
-
-    def to_dict(self) -> Dict[str, Any]:
-        return asdict(self)
-
-    def telemetry_data(self) -> Dict[str, Any]:
-        return {
-            "model": self.team_data.get("model") if self.team_data else None,
-            "created_at": self.created_at,
-            "updated_at": self.updated_at,
-        }
-
-    @classmethod
-    def from_dict(cls, data: Mapping[str, Any]) -> Optional[TeamSession]:
-        if data is None or data.get("session_id") is None:
-            logger.warning("TeamSession is missing session_id")
-            return None
-        return cls(
-            session_id=data.get("session_id"),  # type: ignore
-            team_id=data.get("team_id"),
-            team_session_id=data.get("team_session_id"),
-            workflow_session_id=data.get("workflow_session_id"),
-            user_id=data.get("user_id"),
-            memory=data.get("memory"),
-            team_data=data.get("team_data"),
-            session_data=data.get("session_data"),
-            extra_data=data.get("extra_data"),
-            created_at=data.get("created_at"),
-            updated_at=data.get("updated_at"),
-        )

agno/storage/session/v2/__init__.py

@@ -1,5 +0,0 @@
-from agno.storage.session.workflow import WorkflowSession
-
-__all__ = [
-    "WorkflowSession",
-]

agno/storage/session/workflow.py

@@ -1,61 +0,0 @@
-from __future__ import annotations
-
-from dataclasses import asdict, dataclass
-from typing import Any, Dict, Mapping, Optional
-
-from agno.utils.log import logger
-
-
-@dataclass
-class WorkflowSession:
-    """Workflow Session that is stored in the database"""
-
-    # Session UUID
-    session_id: str
-    # ID of the user interacting with this agent
-    user_id: Optional[str] = None
-    # Agent Memory
-    memory: Optional[Dict[str, Any]] = None
-    # Session Data: session_name, session_state, images, videos, audio
-    session_data: Optional[Dict[str, Any]] = None
-    # Extra Data stored with this agent
-    extra_data: Optional[Dict[str, Any]] = None
-    # The unix timestamp when this session was created
-    created_at: Optional[int] = None
-    # The unix timestamp when this session was last updated
-    updated_at: Optional[int] = None
-
-    # ID of the workflow that this session is associated with
-    workflow_id: Optional[str] = None
-    # Workflow Data
-    workflow_data: Optional[Dict[str, Any]] = None
-
-    def to_dict(self) -> Dict[str, Any]:
-        return asdict(self)
-
-    def monitoring_data(self) -> Dict[str, Any]:
-        return asdict(self)
-
-    def telemetry_data(self) -> Dict[str, Any]:
-        return {
-            "created_at": self.created_at,
-            "updated_at": self.updated_at,
-        }
-
-    @classmethod
-    def from_dict(cls, data: Mapping[str, Any]) -> Optional[WorkflowSession]:
-        if data is None or data.get("session_id") is None:
-            logger.warning("WorkflowSession is missing session_id")
-            return None
-
-        return cls(
-            session_id=data.get("session_id"),  # type: ignore
-            workflow_id=data.get("workflow_id"),
-            user_id=data.get("user_id"),
-            memory=data.get("memory"),
-            workflow_data=data.get("workflow_data"),
-            session_data=data.get("session_data"),
-            extra_data=data.get("extra_data"),
-            created_at=data.get("created_at"),
-            updated_at=data.get("updated_at"),
-        )