ug$rddlZddlmZmZmZmZddlmZmZddl m Z m Z ddl m cmZddlmZddlmZddlmZmZmZddlZddlmZddlZdd lmZmZmZmZm Z dd l!m"Z"dd l#m$Z$ddl%Z%dd l&m'Z'dd l(m)Z)ddl*m+Z+ddl,m-Z-ej\e/Z0GddejbZ1GddeeeZ2y)N) ConnectionLockPool PerThreadPoolPool) MigratableDB Migration)SystemSettings)SqlEmbeddingsQueue)SqlSysDB)OpenTelemetryClientOpenTelemetryGranularity trace_method)override)SequencecastOptionalTypeAny)Literal) TracebackTypeUUID)local)files) Traversablec eZdZUeed<eed<dedefdZede jfdZ ede e ed e ed e eded fd Zy ) TxWrapper_conn_pool conn_poolstackcJ||_|j|_||_yN) _tx_stackconnectrr )selfr!r"s L/var/www/openai/venv/lib/python3.12/site-packages/chromadb/db/impl/sqlite.py__init__zTxWrapper.__init__ s &&(  returnc0t|jjdk(r6|jj d|jj d|jjj ||jj S)NrPRAGMA case_sensitive_like = ONzBEGIN;)lenr%r"rexecuteappendcursorr's r( __enter__zTxWrapper.__enter__%si t~~## $ ) JJ  @ A JJ  x ( ##D)zz  ""r*exc_type exc_value tracebackFc|jjjt|jjdk(r7||jj n|jj |jjj|jj|jy)NrF) r%r"popr.rcommitrollbackr1closer return_to_pool)r'r4r5r6s r(__exit__zTxWrapper.__exit__-s   " t~~## $ ) !!# ##% !!# !!$**-r*N)__name__ __module__ __qualname__r__annotations__rrr)rbaseCursorr3rr BaseExceptionrrr=r*r(rrs  K$u #4;;##4 ./M*M*   r*rcpeZdZUeed<eed<eeed<eed<e ed<e ed<de ffd Z e d ejed'fd Ze d ejed'fd Zeed eej,fdZeed efdZeed efdZed eefdZed efdZe dejed'fd Ze dejed'dZe dejed e fdZe dejeded ee fdZ!ede"jFde d d fdZ$eed e%e&d e%e'fd!Z(eed"e%e'd e%e&fd#Z)eed ee*fd$Z+d(d%e,d d fd&Z-xZ.S))SqliteDB _conn_pool _settings_migration_imports_db_filer%_is_persistentsystemc|j|_tdtdtdg|_|jj d|_|j t |_|j s$d|_t|jd|_ n|jj dd z|_tjj|js>tjtjj|jd t!|j|_ t#|_t&|Q|y) 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__Gs 8 9 - . . /#  #nn44_E%+^^4G%H""" 9DM&t}}TBDO&&':;>OO M77>>$--0 BGGOODMM:TJ+DMM:DO  r*zSqliteDB.startr+Nct||j5}|jd|jdddd|j |j j ddk(r>|jjdjdurtjdyyy#1swYvxYw)NzPRAGMA foreign_keys = ONr- migrationsapplyautomatically_purgeFu⚠️ It looks like you upgraded from a version below 0.5.6 and could benefit from vacuuming your database. Run chromadb utils vacuum --help for more information.) r[starttxr/initialize_migrationsrIrTconfig get_parametervalueloggerwarn)r'curr\s r(razSqliteDB.startas   WWY# KK2 3 KK9 : ""$ NN " "< 0G ; ))*?@FF%O KKv P <Ys #B88Cz SqliteDB.stopcVt||jjyr$)r[stoprHr;)r'r\s r(rkz SqliteDB.stopss   r*c"tjSr$)pypikaQueryrEr*r( querybuilderzSqliteDB.querybuilderys||r*cy)N?rErEr*r(parameter_formatzSqliteDB.parameter_format~sr*cy)NsqliterErEr*r(migration_scopezSqliteDB.migration_scopesr*c|jSr$)rJr2s r(migration_dirszSqliteDB.migration_dirss&&&r*ct|jdsg|j_t|j|jS)Nr")r")hasattrr%r"rrHr2s r(rbz SqliteDB.txs1t~~w/#%DNN ??r*zSqliteDB.reset_statec|jjds td|j5}|j d|j D]}|j d|d ddd|j j|jt|)y#1swYBxYw)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 ValueErrorrbr/fetchallrHr;rar[ reset_state)r'rirowr\s r(r~zSqliteDB.reset_states~~%%m4p WWY# KK  ||~ 3CF8<=&   Ys >B88CzSqliteDB.setup_migrationscp|j5}|jddddy#1swYyxYw)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) ) )rbr/r'ris r(setup_migrationszSqliteDB.setup_migrationss*WWY# KK  YYs,5zSqliteDB.migrations_initializedc|j5}|jd|jddk(r dddy dddy#1swYyxYw)Nz[SELECT count(*) FROM sqlite_master WHERE type='table' AND name='migrations'rFT)rbr/fetchoners r(migrations_initializedzSqliteDB.migrations_initializedsNWWY# KK<  ||~a A%YYYs)A A  AzSqliteDB.db_migrationsdirc |j5}|jd|jfg}|jD]}t t |d}t t |d}t t |d}t t |d}t t |d} |jt||||| |j|cdddS#1swYyxYw)Nz SELECT dir, version, filename, sql, hash FROM migrations WHERE dir = ? ORDER BY version ASC r)rversionfilenamesqlhashscope) rbr/namer}rstrintr0rru) r'rrir^r found_dir found_versionfound_filename found_sql found_hashs r( db_migrationszSqliteDB.db_migrationssWWY# KK   J||~ c!f- $S#a& 1 !%c3q6!2 c!f- !#s1v. !!% -!/%'"224   & 9YYs CCC'ri migrationc x|j|d|jd|d|d|d|d|dfy)Nrzw INSERT INTO migrations (dir, version, filename, sql, hash) VALUES (?, ?, ?, ?, ?) rrrr) executescriptr/)r'rirs r(apply_migrationzSqliteDB.apply_migrationsU )E*+   % )$*%% &!   r*rfc | t|SdSr$r)rfs r( uuid_from_dbzSqliteDB.uuid_from_dbs$/tE{9T9r*uuidc | t|SdSr$)r)rs r( uuid_to_dbzSqliteDB.uuid_to_dbs!,s4y6$6r*c"tjSr$)sqlite3IntegrityErrorrEr*r(unique_constraint_errorz SqliteDB.unique_constraint_errors%%%r*timeoutc|jj}|jdt|dz|jd|jdy)zdRuns VACUUM on the database. `timeout` is the maximum time to wait for an exclusive lock in seconds.zPRAGMA busy_timeout = iVACUUMz} INSERT INTO maintenance_log (operation, timestamp) VALUES ('vacuum', CURRENT_TIMESTAMP) N)rHr&r/r)r'rconns r(vacuumzSqliteDB.vacuumsO&&( -c'lT.A-BCD X   r*)r+N))/r>r?r@rrAr rrrrboolr r)rrALLrrark staticmethodrrmrnrorrrurwrrbr~rrrrrBrCrrrrrrrDrrr __classcell__)r\s@r(rGrG?s --M!v!4"$<$@$@A B /#;#?#?@  A  $v||, c S' 5''@I@@ (*B*F*FG H&-/G/K/KL  M 35M5Q5QR   S *,D,H,HI )1DJ> 4;; 9    :HSM:htn:: 7$7HSM77 &T-%8&&  c  $  r*rG)3loggingchromadb.db.impl.sqlite_poolrrrrchromadb.db.migrationsrrchromadb.configr r chromadb.db.basedbrB#chromadb.db.mixins.embeddings_queuer chromadb.db.mixins.sysdbr chromadb.telemetry.opentelemetryr rrr overridesrrmtypingrrrrrtyping_extensionsrtypesrrVrr threadingrimportlib_resourcesrimportlib_resources.abcr getLoggerr>rgrrGrEr*r(rsRR:,B-  66% %/   8 $  FR |/R r*