ui JddlmZmZmZmZddlmZmZddlm Z m Z ddl m cm Z ddlmZddlmZddlmZmZmZddlZddlmZddlZdd lmZmZmZmZmZdd l m!Z!dd l"m#Z#ddl$Z$dd l%m&Z&dd l'm(Z(ddl)m*Z*ddl+m,Z,Gdde j-Z-GddeeeZ.dS)) ConnectionLockPool PerThreadPoolPool) MigratableDB Migration)SystemSettingsN)SqlEmbeddingsQueue)SqlSysDB)OpenTelemetryClientOpenTelemetryGranularity trace_method)override)SequencecastOptionalTypeAny)Literal) TracebackTypeUUID)local)files) Traversablec eZdZUeed<eed<dedefdZede j fdZ ede e ed e ed e eded fd Zd S) TxWrapper_conn_pool conn_poolstackcT||_||_||_dSN) _tx_stackconnectrr )selfr!r"s J/var/www/icac/venv/lib/python3.11/site-packages/chromadb/db/impl/sqlite.py__init__zTxWrapper.__init__s'&&((  returnct|jjdkr4|jd|jd|jj||jS)NrPRAGMA case_sensitive_like = ONzBEGIN;)lenr%r"rexecuteappendcursorr's r( __enter__zTxWrapper.__enter__"st t~# $ $ ) ) J  @ A A A J  x ( ( ( ##D)))z  """r*exc_type exc_value tracebackFcz|jjt|jjdkr5||jn|j|j|j |jdS)NrF) r%r"popr.rcommitrollbackr1closer return_to_pool)r'r4r5r6s r(__exit__zTxWrapper.__exit__*s   """ t~# $ $ ) ) !!#### ##%%% !!### !!$*---ur*N)__name__ __module__ __qualname__r__annotations__rrr)rbaseCursorr3rr BaseExceptionrrr=r*r(rrs  KKK$u #4;###X#4 ./M*M*   Xr*rc eZdZUeed<eed<eeed<eed<e ed<e ed<de ffd Z e d ejed%fd Ze d ejed%fd Zeed eejfdZeed efdZeed efdZed eefdZed efdZe dejed%fd Ze dejed%dZe dejed e fdZe dejeded ee fdZ!ede"j#de d d fdZ$eed e%e&d e%e'fd!Z(eed"e%e'd e%e&fd#Z)eed ee*fd$Z+xZ,S)&SqliteDB _conn_pool _settings_migration_imports_db_filer%_is_persistentsystemc|j|_tdtdtdg|_|jd|_|t |_|js#d|_t|jd|_ n|jdd z|_tj |js8tjtj |jd t!|j|_ t#|_t'|dS) Nz$chromadb.migrations.embeddings_queuezchromadb.migrations.sysdbzchromadb.migrations.metadb is_persistentzfile::memory:?cache=sharedT)is_uripersist_directoryz/chroma.sqlite3)exist_ok)settingsrIrrJrequirerLr _opentelemetry_clientrKrrHospathexistsmakedirsdirnamerrr%superr))r'rM __class__s r(r)zSqliteDB.__init__Ds2 8 9 9 - . . . / /#  #n44_EE%+^^4G%H%H"" ; 9DM&t}TBBBDOO&&':;;>OO M7>>$-00 K BGOODM::TJJJJ+DM::DO      r*zSqliteDB.startr+Nct|5}|d|ddddn #1swxYwY|dS)NzPRAGMA foreign_keys = ONr-)r[starttxr/initialize_migrations)r'curr\s r(r^zSqliteDB.start^s   WWYY ;# KK2 3 3 3 KK9 : : : ; ; ; ; ; ; ; ; ; ; ; ; ; ; ; ""$$$$$s+A--A14A1z SqliteDB.stopczt|jdSr$)r[stoprHr;)r'r\s r(rcz SqliteDB.stopgs0   r*ctjSr$)pypikaQueryrEr*r( querybuilderzSqliteDB.querybuilderms |r*cdS)N?rErEr*r(parameter_formatzSqliteDB.parameter_formatrs sr*cdS)NsqliterErEr*r(migration_scopezSqliteDB.migration_scopews xr*c|jSr$)rJr2s r(migration_dirszSqliteDB.migration_dirs|s &&r*czt|jds g|j_t|j|jS)Nr")r")hasattrr%r"rrHr2s r(r_z SqliteDB.txs7t~w// &#%DN ????r*zSqliteDB.reset_statec|jdstd|5}|d|D] }|d|d! dddn #1swxYwY|j|t dS)N allow_resetzResetting the database is not allowed. Set `allow_reset` to true in the config in tests or other non-production environments where reset should be permitted.zo SELECT name FROM sqlite_master WHERE type='table' zDROP TABLE IF EXISTS r) rIrT ValueErrorr_r/fetchallrHr;r^r[ reset_state)r'rarowr\s r(rvzSqliteDB.reset_states1~%%m44 p WWYY ># KK    ||~~ > >  > > > > > > > > > > > > > > >   sA BBBzSqliteDB.setup_migrationsc|5}|dddddS#1swxYwYdS)Na~ CREATE TABLE IF NOT EXISTS migrations ( dir TEXT NOT NULL, version INTEGER NOT NULL, filename TEXT NOT NULL, sql TEXT NOT NULL, hash TEXT NOT NULL, PRIMARY KEY (dir, version) ) )r_r/r'ras r(setup_migrationszSqliteDB.setup_migrationssWWYY # KK                      s 8<<zSqliteDB.migrations_initializedc|5}|d|ddkr ddddS ddddS#1swxYwYdS)Nz[SELECT count(*) FROM sqlite_master WHERE type='table' AND name='migrations'rFT)r_r/fetchonerys r(migrations_initializedzSqliteDB.migrations_initializedsWWYY # KK<    ||~~a A%%                          s5A%A%%A),A)zSqliteDB.db_migrationsdirc N|5}|d|jfg}|D]}t t |d}t t |d}t t |d}t t |d}t t |d} |t||||| | |cdddS#1swxYwYdS)Nz SELECT dir, version, filename, sql, hash FROM migrations WHERE dir = ? ORDER BY version ASC r)r~versionfilenamesqlhashscope) r_r/namerurstrintr0rrm) r'r~ra migrationsrw found_dir found_versionfound_filename found_sql found_hashs r( db_migrationszSqliteDB.db_migrationssfWWYY # KK     J||~~   c!f-- $S#a& 1 1 !%c3q6!2!2 c!f-- !#s1v.. !!% -!/%'"2244     9                  sC8DD!Dra migrationc ||d|d|d|d|d|d|dfdS)Nrzw INSERT INTO migrations (dir, version, filename, sql, hash) VALUES (?, ?, ?, ?, ?) r~rrr) executescriptr/)r'rars r(apply_migrationzSqliteDB.apply_migrationso )E*+++   % )$*%% &!   r*valuec(|t|ndSr$r)rs r( uuid_from_dbzSqliteDB.uuid_from_dbs$/tE{{{T9r*uuidc(|t|ndSr$)r)rs r( uuid_to_dbzSqliteDB.uuid_to_dbs!,s4yyy$6r*ctjSr$)sqlite3IntegrityErrorrEr*r(unique_constraint_errorz SqliteDB.unique_constraint_errors %%r*)r+N)-r>r?r@rrAr rrrrboolr r)rrALLrr^rc staticmethodrrerfrgrjrmrorr_rvrzr}rrrBrCrrrrrrrDr __classcell__)r\s@r(rGrG<s ----MMM!v!!!!!!4\"$<$@AA %%%%%XBA%\/#;#?@@      XA@  $v|,X\ cX\ SX\' 5'''X'@I@@@X@ \(*B*FGG XHG&\-/G/KLL    XML \35M5QRR     XSR \*,D,HII )1DXJI> 4; 9    X  :HSM:htn:::X\: 7$7HSM777X\7 &T-%8&&&X\&&&&&r*rG)/chromadb.db.impl.sqlite_poolrrrrchromadb.db.migrationsrrchromadb.configr r chromadb.db.basedbrB#chromadb.db.mixins.embeddings_queuer chromadb.db.mixins.sysdbr chromadb.telemetry.opentelemetryr rrr overridesrretypingrrrrrtyping_extensionsrtypesrrVrr threadingrimportlib_resourcesrimportlib_resources.abcrrrGrEr*r(rsRRRRRRRRRRRR::::::::,,,,,,,,BBBBBB------  66666666666666%%%%%% %%%%%%//////        F}&}&}&}&}&|/}&}&}&}&}&r*