o h'@sdddlmZddlZddlZddlZddlZddlmZddlm Z ddl m Z Gdd d e Z dS) ) annotationsN)Path)TResponseInputItem) SessionABCc@sjeZdZdZ   d%d&d d Zd'ddZd(ddZd)d*ddZd+ddZd,dd Z d-d!d"Z d-d#d$Z dS). SQLiteSessionzSQLite-based implementation of session storage. This implementation stores conversation history in a SQLite database. By default, uses an in-memory database that is lost when the process ends. For persistent storage, provide a file path. :memory:agent_sessionsagent_messages session_idstrdb_path str | Pathsessions_tablemessages_tablecCs||_||_||_||_t|_t|_t |dk|_ |j r6t j ddd|_ |j d||j dSt j t |jdd}|d|||dS)aInitialize the SQLite session. Args: session_id: Unique identifier for the conversation session db_path: Path to the SQLite database file. Defaults to ':memory:' (in-memory database) sessions_table: Name of the table to store session metadata. Defaults to 'agent_sessions' messages_table: Name of the table to store message data. Defaults to 'agent_messages' r Fcheck_same_threadPRAGMA journal_mode=WALN)r rrr threadinglocal_localLock_lockr _is_memory_dbsqlite3connect_shared_connectionexecute_init_db_for_connectionclose)selfr rrr init_connr#]/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/memory/sqlite_session.py__init__s      zSQLiteSession.__init__returnsqlite3.ConnectioncCsl|jr|jSt|jdstjt|jdd|j_|jj dt |jjtj s2Jdt |jj|jjS)zGet a database connection. connectionFrrz!Expected sqlite3.Connection, got ) rrhasattrrrrr rr(r isinstance Connectiontyper!r#r#r$_get_connection:s  zSQLiteSession._get_connectionconnNonecCsX|d|jd|d|jd|jd|d|jd|jd|dS) z9Initialize the database schema for a specific connection.z( CREATE TABLE IF NOT EXISTS z ( session_id TEXT PRIMARY KEY, created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP, updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP ) a ( id INTEGER PRIMARY KEY AUTOINCREMENT, session_id TEXT NOT NULL, message_data TEXT NOT NULL, created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP, FOREIGN KEY (session_id) REFERENCES zJ (session_id) ON DELETE CASCADE ) z, CREATE INDEX IF NOT EXISTS idx_z_session_id ON z" (session_id, created_at) N)rrrcommit)r!r/r#r#r$rLs(   z%SQLiteSession._init_db_for_connectionNlimit int | Nonelist[TResponseInputItem]cs fdd}t|IdHS)aORetrieve the conversation history for this session. Args: limit: Maximum number of items to retrieve. If None, retrieves all items. When specified, returns the latest N items in chronological order. Returns: List of input items representing the conversation history c s}jr jntVdur!|djdjf}n|djdjf}|}dur=t t |}g}|D]\}z t |}| |WqAt jyZYqAw|WdS1sgwYdS)Nz2 SELECT message_data FROM zr WHERE session_id = ? ORDER BY created_at ASC z WHERE session_id = ? ORDER BY created_at DESC LIMIT ? )r.rrrrrrr fetchalllistreversedjsonloadsappendJSONDecodeError)r/cursorrowsitems message_dataitemr2r!r#r$_get_items_syncys8     $z0SQLiteSession.get_items.._get_items_syncNasyncio to_thread)r!r2rBr#rAr$ get_itemsns *zSQLiteSession.get_itemsr>cs,sdSfdd}t|IdHdS)zAdd new items to the conversation history. Args: items: List of input items to add to the history Ncs}jr jnt;|djdjffddD}|dj d||djdjf| WddS1sLwYdS) Nz+ INSERT OR IGNORE INTO z) (session_id) VALUES (?) csg|] }jt|fqSr#)r r8dumps).0r@r-r#r$ szDSQLiteSession.add_items.._add_items_sync..z! INSERT INTO z: (session_id, message_data) VALUES (?, ?) z UPDATE zq SET updated_at = CURRENT_TIMESTAMP WHERE session_id = ? ) r.rrrrrrr executemanyrr1)r/r?r>r!r#r$_add_items_syncs. "z0SQLiteSession.add_items.._add_items_syncrC)r!r>rLr#rKr$ add_itemss !zSQLiteSession.add_itemsTResponseInputItem | Nonecsfdd}t|IdHS)zRemove and return the most recent item from the session. Returns: The most recent item if it exists, None if the session is empty c s}jr jntJ|djdjdjf}|}| |rO|d}zt |}|WWdSt j yNYWddSw WddS1s[wYdS)Nz! DELETE FROM zI WHERE id = ( SELECT id FROM z WHERE session_id = ? ORDER BY created_at DESC LIMIT 1 ) RETURNING message_data r) r.rrrrrrr fetchoner1r8r9r;)r/r<resultr?r@r-r#r$_pop_item_syncs0  $z.SQLiteSession.pop_item.._pop_item_syncNrC)r!rQr#r-r$pop_items  zSQLiteSession.pop_itemcs"fdd}t|IdHdS)z!Clear all items for this session.cs~}jr jnt'|djdjf|djdjf| WddS1s8wYdS)Nz DELETE FROM z WHERE session_id = ?) r.rrrrrrr rr1)r/r-r#r$_clear_session_syncs   "z8SQLiteSession.clear_session.._clear_session_syncNrC)r!rSr#r-r$ clear_sessions  zSQLiteSession.clear_sessioncCsB|jrt|dr|jdSdSt|jdr|jjdSdS)zClose the database connection.rr(N)rr)rr rr(r-r#r#r$r s  zSQLiteSession.close)r r r )r r rrrr rr )r&r')r/r'r&r0)N)r2r3r&r4)r>r4r&r0)r&rN)r&r0) __name__ __module__ __qualname____doc__r%r.rrFrMrRrTr r#r#r#r$r s  %  " 7 , )r) __future__rrDr8rrpathlibrr>rsessionrrr#r#r#r$s