Zi?ddlmZddlmZmZmZddlmZddlm Z m Z ddl m Z ddl mZmZmZmZmZmZmZmZmZmZddlmZmZmZGdd eZd S) )abstractmethod)OptionalSequenceTuple)UUID)CreateCollectionConfigurationUpdateCollectionConfiguration)Schema) CollectionCollectionAndSegmentsDatabaseTenantMetadataSegment SegmentScopeOptionalArgument UnspecifiedUpdateMetadata)DEFAULT_DATABASEDEFAULT_TENANT ComponentceZdZdZeefdedededdfdZeefdedede 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+dS)(SysDBz+Data interface for Chroma's System databaseidnametenantreturnNcdS)zeCreate a new database in the System database. Raises an Error if the Database already exists.N)selfrrrs O/root/projects/butler/venv/lib64/python3.11/site-packages/chromadb/db/system.pycreate_databasezSysDB.create_databases cdS)zZGet a database by name and tenant. Raises an Error if the Database does not exist.Nrr rrs r! get_databasezSysDB.get_database#  r#cdS)zDelete a database.Nrr%s r!delete_databasezSysDB.delete_database)  r#limitoffsetcdS)z List all databases for a tenant.Nr)r r+r,rs r!list_databaseszSysDB.list_databases.s  r#cdS)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#cdS)zCGet a tenant by name. Raises an Error if the Tenant does not exist.Nrr0s r! get_tenantzSysDB.get_tenant>r*r#segmentcdS)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# collectioncdS)z*Delete a segment from the System database.Nr)r r7rs r!delete_segmentzSysDB.delete_segmentKr*r#typescopecdS)z/Find segments by id, type, scope or collection.Nr)r r7rr:r;s r! get_segmentszSysDB.get_segmentsP  r#metadatacdS)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 dS)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#cdS)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#cdS)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#cdS)zFGets the number of collections for the (tenant, database) combination.Nr)r rrGs r!count_collectionszSysDB.count_collections  r# collection_idcdS)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#cdS)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#cdS)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!rrs551?   ! +.    ^ 4B   c x   ^ 7E  C  $   ^  $ $$   }     (     ^  # $   ^  s v   ^  g $   ^   4 D   ^ ""(,     TN sm   %  '     ^  @K{}}        #8N#;<      ^  (,#'#$(      5  7#  8$ C=     z4    ^ 0%(           ^ ""$(# $   TN  sm      }      *     ^  %"&   3-     ^  !    ^ '2kmm5@[]]?J{}} KMM     s#  $HSM2  #8N#;<  ( 2 3       ^   d s   ^   r#rN)abcrtypingrrruuidr%chromadb.api.collection_configurationrr chromadb.api.typesr chromadb.typesr r r rrrrrrrchromadb.configrrrrrr#r!rfsZ,,,,,,,,,,&%%%%%                        HGGGGGGGGGe e e e e Ie e e e e r#