_j ddlmZddlmZddlmZmZmZddlm Z mZddl m Z m Z ddl mZGdde ZeGd d ZeeZGd d e e Zd S))abstractmethod) dataclass)AnyCallableList)EnforceOverrides overrides) ComponentSystem)SegmentcteZdZdZedededeefdZ ede egdfddfdZ dS) SegmentDirectoryzA segment directory is a data interface that manages the location of segments. Concretely, this means that for distributed chroma, it provides the grpc endpoint for a segment.segmentnreturncdS)zReturn the segment residences for a given segment ID. Will return at most n residences. Should only return less than n residences if there are less than n residences available. N)selfrrs b/var/www/FlaskApp/flask-venv/lib/python3.11/site-packages/chromadb/segment/distributed/__init__.pyget_segment_endpointsz&SegmentDirectory.get_segment_endpointsscallbackNcdS)zARegister a callback that will be called when a segment is updatedNrrrs r!register_updated_segment_callbackz2SegmentDirectory.register_updated_segment_callbacks r) __name__ __module__ __qualname____doc__rr intrstrrrrrrrrr sWW W  c   ^   'D1    ^   rrc.eZdZUeed<eed<eed<dS)MemberidipnodeN)rrrr!__annotations__rrrr#r#s+ GGG GGG IIIIIrr#ceZdZUdZeeegefed<de ffd Z e defdZ e de ddfd Zedd Zd eegefddfd Zd eegefdefd ZxZS)MemberlistProviderzReturns the latest memberlist and provdes a callback for when it changes. This callback may be called from a different thread than the one that called. Callers should ensure that they are thread-safe. callbackssystemcXg|_t|dS)N)r*super__init__)rr+ __class__s rr.zMemberlistProvider.__init__-s(      rrcdS)zReturns the latest memberlistNrrs rget_memberlistz!MemberlistProvider.get_memberlist1  r memberlistNcdS)z1Sets the memberlist that this provider will watchNr)rr4s rset_memberlist_namez&MemberlistProvider.set_memberlist_name6r3rcg|_dS)zStops watching the memberlistN)r*r1s rstopzMemberlistProvider.stop;srrc:|j|dS)zRegisters a callback that will be called when the memberlist changes. May be called many times with the same memberlist, so callers should be idempotent. May be called from a different thread. N)r*appendrs r$register_updated_memberlist_callbackz7MemberlistProvider.register_updated_memberlist_callback@s h'''''rcP||jvr|j|dSdS)zUnregisters a callback that was previously registered. Returns True if the callback was successfully unregistered, False if it was not ever registered.TF)r*removers r&unregister_updated_memberlist_callbackz9MemberlistProvider.unregister_updated_memberlist_callbackHs1 t~ % % N ! !( + + +4ur)rN)rrrrrr Memberlistrr'r r.rr2r!r6r r8r;boolr> __classcell__)r/s@rr)r)&sD""Hj\3./0000!v!!!!!!    ^  c d   ^ Y( *s!23( (((( *s!23 rr)N)abcr dataclassesrtypingrrrr rchromadb.configr r chromadb.typesr rr#r?r)rrrrGs)!!!!!!&&&&&&&&&&11111111--------""""""     y   $   &\ *****$4*****r