q 9i?ddlmZddlmZmZmZddlmZddlm Z m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZddlmZmZmZGdd eZy ) )abstractmethod)OptionalSequenceTuple)UUID)CreateCollectionConfigurationUpdateCollectionConfiguration)Schema) CollectionCollectionAndSegmentsDatabaseTenantMetadataSegment SegmentScopeOptionalArgument UnspecifiedUpdateMetadata)DEFAULT_DATABASEDEFAULT_TENANT ComponentcveZdZdZeefdedededdfdZeefdedede fdZ eefdededdfd Z eddefd e e d e e dedee fd Zededdfd ZededefdZededdfdZedededdfdZe d'dede ede ede edeef dZeefdededee eddfdZedddeefdedede ededeede ede e de dedede!e"e ffdZ#eeefdedededdfd Z$eddeeddfde ede ededed e e d e e dee"fd!Z%eedfdede ede fd"Z&ed#ede'fd$Z(eeeeefdedeedee e dee edee e)ddf d%Z*edede fd&Z+y)(SysDBz+Data interface for Chroma's System databaseidnametenantreturnNcy)zeCreate a new database in the System database. Raises an Error if the Database already exists.N)selfrrrs [/var/www/html/leadgen/airagagent/rag_env/lib/python3.12/site-packages/chromadb/db/system.pycreate_databasezSysDB.create_databases cy)zZGet a database by name and tenant. Raises an Error if the Database does not exist.Nrr rrs r! get_databasezSysDB.get_database# r#cy)zDelete a database.Nrr%s r!delete_databasezSysDB.delete_database) r#limitoffsetcy)z List all databases for a tenant.Nr)r r+r,rs r!list_databaseszSysDB.list_databases.s r#cy)zzCreate a new tenant in the System database. The name must be unique. Raises an Error if the Tenant already exists.Nrr rs r! create_tenantzSysDB.create_tenant8r'r#cy)zCGet a tenant by name. Raises an Error if the Tenant does not exist.Nrr0s r! get_tenantzSysDB.get_tenant>r*r#segmentcy)z^Create a new segment in the System database. Raises an Error if the ID already exists.Nr)r r4s r!create_segmentzSysDB.create_segmentEr'r# collectioncy)z*Delete a segment from the System database.Nr)r r7rs r!delete_segmentzSysDB.delete_segmentKr*r#typescopecy)z/Find segments by id, type, scope or collection.Nr)r r7rr:r;s r! get_segmentszSysDB.get_segmentsP r#metadatacy)zUpdate a segment. Unspecified fields will be left unchanged. For the metadata, keys with None values will be removed and keys not present in the UpdateMetadata dict will be left unchanged.Nr)r r7rr?s r!update_segmentzSysDB.update_segment[s r#Fschema configurationsegments dimension get_or_createdatabasec y)aCreate a new collection and associated resources in the SysDB. If get_or_create is True, the collection will be created if one with the same name does not exist. The metadata will be updated using the same protocol as update_collection. If get_or_create is False and the collection already exists, an error will be raised. Returns a tuple of the created collection and a boolean indicating whether the collection was created or not. Nr) r rrrBrCrDr?rErFrrGs r!create_collectionzSysDB.create_collectiongs. r#cy)zDelete a collection, all associated segments and any associate resources (log stream) from the SysDB and the system at large.Nr)r rrrGs r!delete_collectionzSysDB.delete_collectionr>r#cy)z_Find collections by id or name. If name is provided, tenant and database must also be provided.Nr)r rrrrGr+r,s r!get_collectionszSysDB.get_collectionss r#cy)zFGets the number of collections for the (tenant, database) combination.Nr)r rrGs r!count_collectionszSysDB.count_collections r# collection_idcy)zGet a consistent snapshot of a collection by id. This will return a collection with segment information that matches the collection version and log position. Nr)r rQs r!get_collection_with_segmentsz"SysDB.get_collection_with_segmentsrPr#cy)zUpdate a collection. Unspecified fields will be left unchanged. For metadata, keys with None values will be removed and keys not present in the UpdateMetadata dict will be left unchanged.Nr)r rrrEr?rCs r!update_collectionzSysDB.update_collections r#cy)z.Returns the number of records in a collection.Nr)r rs r!get_collection_sizezSysDB.get_collection_sizer*r#)NNN),__name__ __module__ __qualname____doc__rrrstrr"r r&r)rintrr.r1rr3rr6r9rr=rrrrArr rrboolrr rIrKrMrOr rSr rUrWrr#r!rrs51?  ! +.   4B  c x  7E C  $   $ $$  }     (     # $   s v   g $    4 D  ""(,    TN sm   %  '     @K}       #8N#;<      (,#'#$(     5  7#  8$ C=     z4   0%(         ""$(# $  TN  sm      }      *     %"&  3-     !   '2m5@]?J} M    s#  $HSM2  #8N#;<  ( 2 3        d s  r#rN)abcrtypingrrruuidr%chromadb.api.collection_configurationrr chromadb.api.typesr chromadb.typesr r r rrrrrrrchromadb.configrrrrrr#r!rfsA,,&   HGe Ie r#