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

agno/storage/postgres.py

@@ -1,682 +0,0 @@
-import time
-from typing import List, Literal, Optional
-
-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, log_warning, logger
-
-try:
-    from sqlalchemy.dialects import postgresql
-    from sqlalchemy.engine import Engine, create_engine
-    from sqlalchemy.inspection import inspect
-    from sqlalchemy.orm import scoped_session, sessionmaker
-    from sqlalchemy.schema import Column, MetaData, Table
-    from sqlalchemy.sql.expression import select, text
-    from sqlalchemy.types import BigInteger, String
-except ImportError:
-    raise ImportError("`sqlalchemy` not installed. Please install it using `pip install sqlalchemy`")
-
-
-class PostgresStorage(Storage):
-    def __init__(
-        self,
-        table_name: str,
-        schema: Optional[str] = "ai",
-        db_url: Optional[str] = None,
-        db_engine: Optional[Engine] = None,
-        schema_version: int = 1,
-        auto_upgrade_schema: bool = False,
-        mode: Optional[Literal["agent", "team", "workflow", "workflow_v2"]] = "agent",
-    ):
-        """
-        This class provides agent storage using a PostgreSQL table.
-
-        The following order is used to determine the database connection:
-            1. Use the db_engine if provided
-            2. Use the db_url
-            3. Raise an error if neither is provided
-
-        Args:
-            table_name (str): Name of the table to store Agent sessions.
-            schema (Optional[str]): The schema to use for the table. Defaults to "ai".
-            db_url (Optional[str]): The database URL to connect to.
-            db_engine (Optional[Engine]): The SQLAlchemy database engine to use.
-            schema_version (int): Version of the schema. Defaults to 1.
-            auto_upgrade_schema (bool): Whether to automatically upgrade the schema.
-            mode (Optional[Literal["agent", "team", "workflow"]]): The mode of the storage.
-        Raises:
-            ValueError: If neither db_url nor db_engine is provided.
-        """
-        super().__init__(mode)
-        _engine: Optional[Engine] = db_engine
-        if _engine is None and db_url is not None:
-            _engine = create_engine(db_url)
-
-        if _engine is None:
-            raise ValueError("Must provide either db_url or db_engine")
-
-        # Database attributes
-        self.table_name: str = table_name
-        self.schema: Optional[str] = schema
-        self.db_url: Optional[str] = db_url
-        self.db_engine: Engine = _engine
-        self.metadata: MetaData = MetaData(schema=self.schema)
-        self.inspector = inspect(self.db_engine)
-
-        # Table schema version
-        self.schema_version: int = schema_version
-        # Automatically upgrade schema if True
-        self.auto_upgrade_schema: bool = auto_upgrade_schema
-        self._schema_up_to_date: bool = False
-
-        # Database session
-        self.Session: scoped_session = scoped_session(sessionmaker(bind=self.db_engine))
-        # Database table for storage
-        self.table: Table = self.get_table()
-        log_debug(f"Created PostgresStorage: '{self.schema}.{self.table_name}'")
-
-    @property
-    def mode(self) -> Literal["agent", "team", "workflow", "workflow_v2"]:
-        """Get the mode of the storage."""
-        return super().mode
-
-    @mode.setter
-    def mode(self, value: Optional[Literal["agent", "team", "workflow", "workflow_v2"]]) -> None:
-        """Set the mode and refresh the table if mode changes."""
-        super(PostgresStorage, type(self)).mode.fset(self, value)  # type: ignore
-        if value is not None:
-            self.table = self.get_table()
-
-    def get_table_v1(self) -> Table:
-        """
-        Define the table schema for version 1.
-
-        Returns:
-            Table: SQLAlchemy Table object representing the schema.
-        """
-        # Common columns for both agent and workflow modes
-        common_columns = [
-            Column("session_id", String, primary_key=True),
-            Column("user_id", String, index=True),
-            Column("memory", postgresql.JSONB),
-            Column("session_data", postgresql.JSONB),
-            Column("extra_data", postgresql.JSONB),
-            Column("created_at", BigInteger, server_default=text("(extract(epoch from now()))::bigint")),
-            Column("updated_at", BigInteger, server_onupdate=text("(extract(epoch from now()))::bigint")),
-        ]
-
-        # Mode-specific columns
-        specific_columns = []
-        if self.mode == "agent":
-            specific_columns = [
-                Column("agent_id", String, index=True),
-                Column("team_session_id", String, index=True, nullable=True),
-                Column("agent_data", postgresql.JSONB),
-            ]
-        elif self.mode == "team":
-            specific_columns = [
-                Column("team_id", String, index=True),
-                Column("team_session_id", String, index=True, nullable=True),
-                Column("team_data", postgresql.JSONB),
-            ]
-        elif self.mode == "workflow":
-            specific_columns = [
-                Column("workflow_id", String, index=True),
-                Column("workflow_data", postgresql.JSONB),
-            ]
-        elif self.mode == "workflow_v2":
-            specific_columns = [
-                Column("workflow_id", String, index=True),
-                Column("workflow_name", String, index=True),
-                Column("workflow_data", postgresql.JSONB),
-                Column("runs", postgresql.JSONB),
-            ]
-
-        # Create table with all columns
-        table = Table(
-            self.table_name,
-            self.metadata,
-            *common_columns,
-            *specific_columns,
-            extend_existing=True,
-            schema=self.schema,  # type: ignore
-        )
-
-        return table
-
-    def get_table(self) -> Table:
-        """
-        Get the table schema based on the schema version.
-
-        Returns:
-            Table: SQLAlchemy Table object for the current schema version.
-
-        Raises:
-            ValueError: If an unsupported schema version is specified.
-        """
-        if self.schema_version == 1:
-            return self.get_table_v1()
-        else:
-            raise ValueError(f"Unsupported schema version: {self.schema_version}")
-
-    def table_exists(self) -> bool:
-        """
-        Check if the table exists in the database.
-
-        Returns:
-            bool: True if the table exists, False otherwise.
-        """
-        try:
-            # Use a direct SQL query to check if the table exists
-            with self.Session() as sess:
-                if self.schema is not None:
-                    exists_query = text(
-                        "SELECT 1 FROM information_schema.tables WHERE table_schema = :schema AND table_name = :table"
-                    )
-                    exists = (
-                        sess.execute(exists_query, {"schema": self.schema, "table": self.table_name}).scalar()
-                        is not None
-                    )
-                else:
-                    exists_query = text("SELECT 1 FROM information_schema.tables WHERE table_name = :table")
-                    exists = sess.execute(exists_query, {"table": self.table_name}).scalar() is not None
-
-            log_debug(f"Table '{self.table.fullname}' does{' not ' if not exists else ' '}exist")
-            return exists
-
-        except Exception as e:
-            logger.error(f"Error checking if table exists: {e}")
-            return False
-
-    def create(self) -> None:
-        """
-        Create the table if it does not exist.
-        """
-        self.table = self.get_table()
-        if not self.table_exists():
-            try:
-                with self.Session() as sess, sess.begin():
-                    if self.schema is not None:
-                        log_debug(f"Creating schema: {self.schema}")
-                        sess.execute(text(f"CREATE SCHEMA IF NOT EXISTS {self.schema};"))
-
-                log_debug(f"Creating table: {self.table_name}")
-
-                # First create the table without indexes
-                table_without_indexes = Table(
-                    self.table_name,
-                    MetaData(schema=self.schema),
-                    *[c.copy() for c in self.table.columns],
-                    schema=self.schema,
-                )
-                table_without_indexes.create(self.db_engine, checkfirst=True)
-
-                # Then create each index individually with error handling
-                for idx in self.table.indexes:
-                    try:
-                        idx_name = idx.name
-                        log_debug(f"Creating index: {idx_name}")
-
-                        # Check if index already exists
-                        with self.Session() as sess:
-                            if self.schema:
-                                exists_query = text(
-                                    "SELECT 1 FROM pg_indexes WHERE schemaname = :schema AND indexname = :index_name"
-                                )
-                                exists = (
-                                    sess.execute(exists_query, {"schema": self.schema, "index_name": idx_name}).scalar()
-                                    is not None
-                                )
-                            else:
-                                exists_query = text("SELECT 1 FROM pg_indexes WHERE indexname = :index_name")
-                                exists = sess.execute(exists_query, {"index_name": idx_name}).scalar() is not None
-
-                        if not exists:
-                            idx.create(self.db_engine)
-                        else:
-                            log_debug(f"Index {idx_name} already exists, skipping creation")
-
-                    except Exception as e:
-                        # Log the error but continue with other indexes
-                        logger.warning(f"Error creating index {idx.name}: {e}")
-
-            except Exception as e:
-                logger.error(f"Could not create table: '{self.table.fullname}': {e}")
-                raise
-
-    def read(self, session_id: str, user_id: Optional[str] = None) -> Optional[Session]:
-        """
-        Read an Session from the database.
-
-        Args:
-            session_id (str): ID of the session to read.
-            user_id (Optional[str]): User ID to filter by. Defaults to None.
-
-        Returns:
-            Optional[Session]: Session object if found, None otherwise.
-        """
-        try:
-            with self.Session() as sess:
-                stmt = select(self.table).where(self.table.c.session_id == session_id)
-                if user_id:
-                    stmt = stmt.where(self.table.c.user_id == user_id)
-                result = sess.execute(stmt).fetchone()
-                if self.mode == "agent":
-                    return AgentSession.from_dict(result._mapping) if result is not None else None
-                elif self.mode == "team":
-                    return TeamSession.from_dict(result._mapping) if result is not None else None
-                elif self.mode == "workflow":
-                    return WorkflowSession.from_dict(result._mapping) if result is not None else None
-                elif self.mode == "workflow_v2":
-                    return WorkflowSessionV2.from_dict(result._mapping) if result is not None else None
-        except Exception as e:
-            if "does not exist" in str(e):
-                log_debug(f"Table does not exist: {self.table.name}")
-                log_debug("Creating table for future transactions")
-                self.create()
-            else:
-                log_debug(f"Exception reading from table: {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.
-
-        Args:
-            user_id (Optional[str]): The ID of the user to filter by.
-            entity_id (Optional[str]): The ID of the agent / workflow to filter by.
-
-        Returns:
-            List[str]: List of session IDs matching the criteria.
-        """
-        try:
-            with self.Session() as sess, sess.begin():
-                # get all session_ids
-                stmt = select(self.table.c.session_id)
-                if user_id is not None:
-                    stmt = stmt.where(self.table.c.user_id == user_id)
-                if entity_id is not None:
-                    if self.mode == "agent":
-                        stmt = stmt.where(self.table.c.agent_id == entity_id)
-                    elif self.mode == "team":
-                        stmt = stmt.where(self.table.c.team_id == entity_id)
-                    elif self.mode == "workflow":
-                        stmt = stmt.where(self.table.c.workflow_id == entity_id)
-                    elif self.mode == "workflow_v2":
-                        stmt = stmt.where(self.table.c.workflow_id == entity_id)
-                # order by created_at desc
-                stmt = stmt.order_by(self.table.c.created_at.desc())
-                # execute query
-                rows = sess.execute(stmt).fetchall()
-                return [row[0] for row in rows] if rows is not None else []
-        except Exception as e:
-            log_debug(f"Exception reading from table: {e}")
-            log_debug(f"Table does not exist: {self.table.name}")
-            log_debug("Creating table for future transactions")
-            self.create()
-        return []
-
-    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.
-
-        Args:
-            user_id (Optional[str]): The ID of the user to filter by.
-            entity_id (Optional[str]): The ID of the agent / workflow to filter by.
-
-        Returns:
-            List[Session]: List of Session objects matching the criteria.
-        """
-        try:
-            with self.Session() as sess, sess.begin():
-                # get all sessions
-                stmt = select(self.table)
-                if user_id is not None:
-                    stmt = stmt.where(self.table.c.user_id == user_id)
-                if entity_id is not None:
-                    if self.mode == "agent":
-                        stmt = stmt.where(self.table.c.agent_id == entity_id)
-                    elif self.mode == "team":
-                        stmt = stmt.where(self.table.c.team_id == entity_id)
-                    elif self.mode == "workflow":
-                        stmt = stmt.where(self.table.c.workflow_id == entity_id)
-                    elif self.mode == "workflow_v2":
-                        stmt = stmt.where(self.table.c.workflow_id == entity_id)
-                # order by created_at desc
-                stmt = stmt.order_by(self.table.c.created_at.desc())
-                # execute query
-                rows = sess.execute(stmt).fetchall()
-                if rows is not None:
-                    if self.mode == "agent":
-                        return [AgentSession.from_dict(row._mapping) for row in rows]  # type: ignore
-                    elif self.mode == "team":
-                        return [TeamSession.from_dict(row._mapping) for row in rows]  # type: ignore
-                    else:
-                        return [WorkflowSession.from_dict(row._mapping) for row in rows]  # type: ignore
-                else:
-                    return []
-        except Exception as e:
-            log_debug(f"Exception reading from table: {e}")
-            log_debug(f"Table does not exist: {self.table.name}")
-            log_debug("Creating table for future transactions")
-            self.create()
-        return []
-
-    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
-        """
-        try:
-            with self.Session() as sess, sess.begin():
-                # Build the base query
-                stmt = select(self.table)
-
-                # Add filters
-                if user_id is not None:
-                    stmt = stmt.where(self.table.c.user_id == user_id)
-                if entity_id is not None:
-                    if self.mode == "agent":
-                        stmt = stmt.where(self.table.c.agent_id == entity_id)
-                    elif self.mode == "team":
-                        stmt = stmt.where(self.table.c.team_id == entity_id)
-                    elif self.mode == "workflow":
-                        stmt = stmt.where(self.table.c.workflow_id == entity_id)
-                    elif self.mode == "workflow_v2":
-                        stmt = stmt.where(self.table.c.workflow_id == entity_id)
-                # Order by created_at desc and limit results
-                stmt = stmt.order_by(self.table.c.created_at.desc())
-                if limit is not None:
-                    stmt = stmt.limit(limit)
-
-                # Execute query
-                rows = sess.execute(stmt).fetchall()
-                if rows is not None:
-                    sessions: List[Session] = []
-                    for row in rows:
-                        session: Optional[Session] = None
-                        if self.mode == "agent":
-                            session = AgentSession.from_dict(row._mapping)  # type: ignore
-                        elif self.mode == "team":
-                            session = TeamSession.from_dict(row._mapping)  # type: ignore
-                        elif self.mode == "workflow":
-                            session = WorkflowSession.from_dict(row._mapping)  # type: ignore
-                        elif self.mode == "workflow_v2":
-                            session = WorkflowSessionV2.from_dict(row._mapping)  # type: ignore
-                        if session is not None:
-                            sessions.append(session)
-                    return sessions
-                return []
-
-        except Exception as e:
-            if "does not exist" in str(e):
-                log_debug(f"Table does not exist: {self.table.name}")
-                log_debug("Creating table for future transactions")
-                self.create()
-            else:
-                log_debug(f"Exception reading from table: {e}")
-            return []
-
-    def upgrade_schema(self) -> None:
-        """
-        Upgrade the schema to the latest version.
-        Currently handles adding the team_session_id column for agent mode.
-        """
-        if not self.auto_upgrade_schema:
-            log_debug("Auto schema upgrade disabled. Skipping upgrade.")
-            return
-
-        try:
-            if self.mode == "agent" and self.table_exists():
-                with self.Session() as sess:
-                    # Check if team_session_id column exists
-                    column_exists_query = text(
-                        """
-                        SELECT 1 FROM information_schema.columns
-                        WHERE table_schema = :schema AND table_name = :table
-                        AND column_name = 'team_session_id'
-                        """
-                    )
-                    column_exists = (
-                        sess.execute(column_exists_query, {"schema": self.schema, "table": self.table_name}).scalar()
-                        is not None
-                    )
-
-                    if not column_exists:
-                        log_info(f"Adding 'team_session_id' column to {self.schema}.{self.table_name}")
-                        alter_table_query = text(
-                            f"ALTER TABLE {self.schema}.{self.table_name} ADD COLUMN team_session_id TEXT"
-                        )
-                        sess.execute(alter_table_query)
-                        sess.commit()
-                        self._schema_up_to_date = True
-                        log_info("Schema upgrade completed successfully")
-        except Exception as e:
-            logger.error(f"Error during schema upgrade: {e}")
-            raise
-
-    def upsert(self, session: Session, create_and_retry: bool = True) -> Optional[Session]:
-        """
-        Insert or update an Session in the database.
-
-        Args:
-            session (Session): The session data to upsert.
-            create_and_retry (bool): Retry upsert if table does not exist.
-
-        Returns:
-            Optional[Session]: The upserted Session, or None if operation failed.
-        """
-        # Perform schema upgrade if auto_upgrade_schema is enabled
-        if self.auto_upgrade_schema and not self._schema_up_to_date:
-            self.upgrade_schema()
-
-        try:
-            with self.Session() as sess, sess.begin():
-                # Create an insert statement
-                if self.mode == "agent":
-                    stmt = postgresql.insert(self.table).values(
-                        session_id=session.session_id,
-                        agent_id=session.agent_id,  # type: ignore
-                        team_session_id=session.team_session_id,  # type: ignore
-                        user_id=session.user_id,
-                        memory=getattr(session, "memory", None),
-                        agent_data=session.agent_data,  # type: ignore
-                        session_data=session.session_data,
-                        extra_data=session.extra_data,
-                    )
-                    # Define the upsert if the session_id already exists
-                    # See: https://docs.sqlalchemy.org/en/20/dialects/postgresql.html#postgresql-insert-on-conflict
-                    stmt = stmt.on_conflict_do_update(
-                        index_elements=["session_id"],
-                        set_=dict(
-                            agent_id=session.agent_id,  # type: ignore
-                            team_session_id=session.team_session_id,  # type: ignore
-                            user_id=session.user_id,
-                            memory=getattr(session, "memory", None),
-                            agent_data=session.agent_data,  # type: ignore
-                            session_data=session.session_data,
-                            extra_data=session.extra_data,
-                            updated_at=int(time.time()),
-                        ),  # The updated value for each column
-                    )
-                elif self.mode == "team":
-                    stmt = postgresql.insert(self.table).values(
-                        session_id=session.session_id,
-                        team_id=session.team_id,  # type: ignore
-                        user_id=session.user_id,
-                        team_session_id=session.team_session_id,  # type: ignore
-                        memory=getattr(session, "memory", None),
-                        team_data=session.team_data,  # type: ignore
-                        session_data=session.session_data,
-                        extra_data=session.extra_data,
-                    )
-                    # Define the upsert if the session_id already exists
-                    # See: https://docs.sqlalchemy.org/en/20/dialects/postgresql.html#postgresql-insert-on-conflict
-                    stmt = stmt.on_conflict_do_update(
-                        index_elements=["session_id"],
-                        set_=dict(
-                            team_id=session.team_id,  # type: ignore
-                            user_id=session.user_id,
-                            team_session_id=session.team_session_id,  # type: ignore
-                            memory=getattr(session, "memory", None),
-                            team_data=session.team_data,  # type: ignore
-                            session_data=session.session_data,
-                            extra_data=session.extra_data,
-                            updated_at=int(time.time()),
-                        ),  # The updated value for each column
-                    )
-                elif self.mode == "workflow":
-                    stmt = postgresql.insert(self.table).values(
-                        session_id=session.session_id,
-                        workflow_id=session.workflow_id,  # type: ignore
-                        user_id=session.user_id,
-                        memory=getattr(session, "memory", None),
-                        workflow_data=session.workflow_data,  # type: ignore
-                        session_data=session.session_data,
-                        extra_data=session.extra_data,
-                    )
-                    # Define the upsert if the session_id already exists
-                    # See: https://docs.sqlalchemy.org/en/20/dialects/postgresql.html#postgresql-insert-on-conflict
-                    stmt = stmt.on_conflict_do_update(
-                        index_elements=["session_id"],
-                        set_=dict(
-                            workflow_id=session.workflow_id,  # type: ignore
-                            user_id=session.user_id,
-                            memory=getattr(session, "memory", None),
-                            workflow_data=session.workflow_data,  # type: ignore
-                            session_data=session.session_data,
-                            extra_data=session.extra_data,
-                            updated_at=int(time.time()),
-                        ),  # The updated value for each column
-                    )
-                elif self.mode == "workflow_v2":
-                    # Convert session to dict to ensure proper serialization
-                    session_dict = session.to_dict()
-
-                    stmt = postgresql.insert(self.table).values(
-                        session_id=session.session_id,
-                        workflow_id=session.workflow_id,  # type: ignore
-                        user_id=session.user_id,
-                        runs=session_dict.get("runs"),
-                        workflow_name=session.workflow_name,  # type: ignore
-                        workflow_data=session.workflow_data,  # type: ignore
-                        session_data=session.session_data,
-                        extra_data=session.extra_data,
-                    )
-                    # Define the upsert if the session_id already exists
-                    # See: https://docs.sqlalchemy.org/en/20/dialects/postgresql.html#postgresql-insert-on-conflict
-                    stmt = stmt.on_conflict_do_update(
-                        index_elements=["session_id"],
-                        set_=dict(
-                            workflow_id=session.workflow_id,  # type: ignore
-                            user_id=session.user_id,
-                            runs=session_dict.get("runs"),
-                            workflow_name=session.workflow_name,  # type: ignore
-                            workflow_data=session.workflow_data,  # type: ignore
-                            session_data=session.session_data,
-                            extra_data=session.extra_data,
-                            updated_at=int(time.time()),
-                        ),
-                    )
-
-                sess.execute(stmt)
-        except Exception as e:
-            if create_and_retry and not self.table_exists():
-                log_debug(f"Table does not exist: {self.table.name}")
-                log_debug("Creating table and retrying upsert")
-                self.create()
-                return self.upsert(session, create_and_retry=False)
-            else:
-                log_warning(f"Exception upserting into table: {e}")
-                log_warning(
-                    "A table upgrade might be required, please review these docs for more information: https://agno.link/upgrade-schema"
-                )
-                return None
-        return self.read(session_id=session.session_id)
-
-    def delete_session(self, session_id: Optional[str] = None):
-        """
-        Delete a session from the database.
-
-        Args:
-            session_id (Optional[str], optional): ID of the session to delete. Defaults to None.
-
-        Raises:
-            Exception: If an error occurs during deletion.
-        """
-        if session_id is None:
-            logger.warning("No session_id provided for deletion.")
-            return
-
-        try:
-            with self.Session() as sess, sess.begin():
-                # Delete the session with the given session_id
-                delete_stmt = self.table.delete().where(self.table.c.session_id == session_id)
-                result = sess.execute(delete_stmt)
-                if result.rowcount == 0:
-                    log_debug(f"No session found with session_id: {session_id}")
-                else:
-                    log_debug(f"Successfully deleted session with session_id: {session_id}")
-        except Exception as e:
-            logger.error(f"Error deleting session: {e}")
-
-    def drop(self) -> None:
-        """
-        Drop the table from the database if it exists.
-        """
-        if self.table_exists():
-            log_debug(f"Deleting table: {self.table_name}")
-            # Drop with checkfirst=True to avoid errors if the table doesn't exist
-            self.table.drop(self.db_engine, checkfirst=True)
-            # Clear metadata to ensure indexes are recreated properly
-            self.metadata = MetaData(schema=self.schema)
-            self.table = self.get_table()
-
-    def __deepcopy__(self, memo):
-        """
-        Create a deep copy of the PostgresStorage instance, handling unpickleable attributes.
-
-        Args:
-            memo (dict): A dictionary of objects already copied during the current copying pass.
-
-        Returns:
-            PostgresStorage: A deep-copied instance of PostgresStorage.
-        """
-        from copy import deepcopy
-
-        # Create a new instance without calling __init__
-        cls = self.__class__
-        copied_obj = cls.__new__(cls)
-        memo[id(self)] = copied_obj
-
-        # Deep copy attributes
-        for k, v in self.__dict__.items():
-            if k in {"metadata", "table", "inspector"}:
-                continue
-            # Reuse db_engine and Session without copying
-            elif k in {"db_engine", "SqlSession"}:
-                setattr(copied_obj, k, v)
-            else:
-                setattr(copied_obj, k, deepcopy(v, memo))
-
-        # Recreate metadata and table for the copied instance
-        copied_obj.metadata = MetaData(schema=copied_obj.schema)
-        copied_obj.inspector = inspect(copied_obj.db_engine)
-        copied_obj.table = copied_obj.get_table()
-
-        return copied_obj