o g@sddlmZddlmZmZmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZmZmZGdddeZd S) )abstractmethod)OptionalSequenceTuple)UUID)CollectionConfigurationInternal) CollectionCollectionAndSegmentsDatabaseTenantMetadataSegment SegmentScopeOptionalArgument UnspecifiedUpdateMetadata)DEFAULT_DATABASEDEFAULT_TENANT Componentc@seZdZdZeefdedededdfddZeefdedede fd d Z eefdededdfd d Z eddefd e e de e dedee fddZededdfddZededefddZededdfddZedededdfddZe   d3dede ede ede edeef ddZeefdededee eddfd d!Zeddd"eefdeded#ed$eede ed%e e d&eded'ede e!effd(d)Z"eeefdeded'eddfd*d+Z#eddeeddfde ede eded'ed e e de e dee!fd,d-Z$ed.ede%fd/d0Z&eeeefdedeed%ee e dee eddf d1d2Z'dS)4SysDBz+Data interface for Chroma's System databaseidnametenantreturnNcCdS)zeCreate a new database in the System database. Raises an Error if the Database already exists.N)selfrrrrrI/var/www/visachat/venv/lib/python3.10/site-packages/chromadb/db/system.pycreate_databaseszSysDB.create_databasecCr)zZGet a database by name and tenant. Raises an Error if the Database does not exist.Nrrrrrrr get_databasezSysDB.get_databasecCr)zDelete a database.Nrrrrrdelete_database%zSysDB.delete_databaselimitoffsetcCr)z List all databases for a tenant.Nr)rr$r%rrrrlist_databases*szSysDB.list_databasescCr)zzCreate a new tenant in the System database. The name must be unique. Raises an Error if the Tenant already exists.Nrrrrrr create_tenant4r!zSysDB.create_tenantcCr)zCGet a tenant by name. Raises an Error if the Tenant does not exist.Nrr'rrr get_tenant:r#zSysDB.get_tenantsegmentcCr)z^Create a new segment in the System database. Raises an Error if the ID already exists.Nr)rr*rrrcreate_segmentAr!zSysDB.create_segment collectioncCr)z*Delete a segment from the System database.Nr)rr,rrrrdelete_segmentGr#zSysDB.delete_segmenttypescopecCr)z/Find segments by id, type, scope or collection.Nr)rr,rr.r/rrr get_segmentsL zSysDB.get_segmentsmetadatacCr)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)rr,rr2rrrupdate_segmentWs zSysDB.update_segmentF configurationsegments dimension get_or_createdatabasec Cr)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) rrrr4r5r2r6r7rr8rrrcreate_collectioncszSysDB.create_collectioncCr)zDelete a collection, all associated segments and any associate resources (log stream) from the SysDB and the system at large.Nr)rrrr8rrrdelete_collection{r1zSysDB.delete_collectioncCr)z_Find collections by id or name. If name is provided, tenant and database must also be provided.Nr)rrrrr8r$r%rrrget_collections zSysDB.get_collections collection_idcCr)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)rr=rrrget_collection_with_segmentssz"SysDB.get_collection_with_segmentscCr)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)rrrr6r2rrrupdate_collectionr<zSysDB.update_collection)NNN)(__name__ __module__ __qualname____doc__rrrstrrr r r"rintrr&r(r r)r r+r-rr0rrrr3rrr boolrrr9r:r;r r>r?rrrrrs$                  rN)abcrtypingrrruuidrchromadb.api.configurationrchromadb.typesrr r r r r rrrrchromadb.configrrrrrrrrs   0