o 1Xxi@sddlmZddlmZddlmZddlmZmZddl m Z ddl m Z m Z GdddeZGd d d eZGd d d e ZGd dde ZdS))abstractmethod)Enum)Optional) Documents Embeddings) Metadatas) ComponentSystemc@s(eZdZdZdZdZdZdZdZdZ dS)ResourceMETADATA_KEY_LENGTHMETADATA_VALUE_LENGTH DOCUMENT_SIZEADD_PER_MINUTEQUERY_PER_MINUTEEMBEDDINGS_DIMENSIONN) __name__ __module__ __qualname__r r r rrGET_PER_MINUTErrrV/home/air/biblejyuku/back/venv/lib/python3.10/site-packages/chromadb/quota/__init__.pyr sr cs*eZdZdededeffdd ZZS) QuotaErrorresourcequotaactualcs4td|d|d|||_||_||_dS)Nzquota error. resource: z quota: z actual: )super__init__rrr)selfrrr __class__rrrs  zQuotaError.__init__)rrrr intr __classcell__rrrrrs"rc sZeZdZdZdeddffdd Ze  d dedee d ee dee fd d Z Z S) QuotaProviderz Retrieves quotas for resources within a system. Methods: get_for_subject(resource, subject=None, tier=None): Returns the quota for a given resource, optionally considering the tier and subject. systemreturnNcst|||_dSN)rrr#rr#rrrr*s  zQuotaProvider.__init__rsubjecttiercCsdSr%r)rrr'r(rrrget_for_subject.szQuotaProvider.get_for_subject)NN) rrr__doc__r rrr rstrr r)r!rrrrr"!sr"c s^eZdZdZdeddffdd Z    d deedeed ee d ee fd d Z Z S) QuotaEnforcera? Enforces quota restrictions on various resources using quota provider. Methods: static_check(metadatas=None, documents=None, embeddings=None, collection_id=None): Performs static checks against quotas for metadatas, documents, and embeddings. Raises QuotaError if limits are exceeded. r#r$Ncs6t|d|_|jjr|t|_d|_||_dS)NFT) rrshould_enforcesettingschroma_quota_provider_implrequirer"_quota_providerr#r&rrrrAs   zQuotaEnforcer.__init__ metadatas documents embeddings collection_idc Cs@|jsdS|jjtj|d}|jjtj|d}|rV|s|rV|D]6}|D]1}|r7t||kr7ttjt||d|rTt||t rTt|||krTttjt|||dq#q|jjtj |d} | rx|rx|D]} t| | krwttj t| | dqe|jjtj |d} | r|r|D]} t| | krttj t| | dqdSdSdS)N)rr')rrr) r-r1r)r r r lenr isinstancer+r r) rr2r3r4r5metadata_key_length_quotametadata_value_length_quotametadatakeydocument_size_quotadocumentembedding_dimension_quota embeddingrrr static_checkIsr      zQuotaEnforcer.static_check)NNNN) rrrr*r rrrrrr+r@r!rrrrr,8s  r,N)abcrenumrtypingrchromadbrr chromadb.apirchromadb.configrr r Exceptionrr"r,rrrrs