ugddlmZddlmZmZmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZmZmZGddeZy ) )abstractmethod)OptionalSequenceTuple)UUID)CollectionConfigurationInternal) CollectionCollectionAndSegmentsDatabaseTenantMetadataSegment SegmentScopeOptionalArgument UnspecifiedUpdateMetadata)DEFAULT_DATABASEDEFAULT_TENANT Componentc eZdZdZeefdedededdfdZeefdedede fdZ eefdededdfd Z eddefd e e d e e dedee fd Zededdfd ZededefdZededdfdZedededdfdZe d$dede ede ede edeef dZeefdededee eddfdZedddeefdedededeede ede e dededede e!effdZ"eeefdedededdfdZ#eddeeddfde ede ededed e e d e e dee!fd Z$ed!ede%fd"Z&eeeefdedeedee e dee eddf d#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 G/var/www/openai/venv/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.Nrrrrs r get_databasezSysDB.get_database r!cy)zDelete a database.Nrr#s rdelete_databasezSysDB.delete_database% r!limitoffsetcy)z List all databases for a tenant.Nr)rr)r*rs rlist_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.Nrrrs r create_tenantzSysDB.create_tenant4r%r!cy)zCGet a tenant by name. Raises an Error if the Tenant does not exist.Nrr.s 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)rr2s rcreate_segmentzSysDB.create_segmentAr%r! collectioncy)z*Delete a segment from the System database.Nr)rr5rs rdelete_segmentzSysDB.delete_segmentGr(r!typescopecy)z/Find segments by id, type, scope or collection.Nr)rr5rr8r9s r get_segmentszSysDB.get_segmentsL 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)rr5rr=s rupdate_segmentzSysDB.update_segmentWs r!F 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) rrrr@rAr=rBrCrrDs rcreate_collectionzSysDB.create_collectioncs, r!cy)zDelete a collection, all associated segments and any associate resources (log stream) from the SysDB and the system at large.Nr)rrrrDs rdelete_collectionzSysDB.delete_collection{r<r!cy)z_Find collections by id or name. If name is provided, tenant and database must also be provided.Nr)rrrrrDr)r*s rget_collectionszSysDB.get_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)rrLs rget_collection_with_segmentsz"SysDB.get_collection_with_segmentss r!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)rrrrBr=s rupdate_collectionzSysDB.update_collectionrKr!)NNN)(__name__ __module__ __qualname____doc__rrrstrr r r$r'rintrr,r/r r1rr4r7rr;rrrr?rrr boolrr rFrHrJr rNrPrr!rrrs51?  ! +.   4B  c x  7E C  $   $ $$  }     (     # $   s v   g $    4 D  ""(,    TN sm   %  '     @K}       #8N#;<      (,#'#$(   7  7#  8$  C=     z4   .%(         ""$(# $  TN  sm      }      *      !   '2m5@]?J}    s#  $HSM2  #8N#;<      r!rN)abcrtypingrrruuidrchromadb.api.configurationrchromadb.typesr r r r r rrrrrchromadb.configrrrrrr!rr^s:,,F   HGS IS r!