Bvh ddlmZddlmZddlmZmZmZddlm Z mZddl m Z m Z ddl mZGdde ZeGd d ZeeZGd d e e Zy ))abstractmethod) dataclass)AnyCallableList)EnforceOverrides overrides) ComponentSystem)SegmentcZeZdZdZedededeefdZ ede egdfddfdZ y) 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.segmentnreturncy)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 O/RAG/venv/lib/python3.12/site-packages/chromadb/segment/distributed/__init__.pyget_segment_endpointsz&SegmentDirectory.get_segment_endpointsscallbackNcy)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 sdW W  c    'D1   rrc,eZdZUeed<eed<eed<y)MemberidipnodeN)rrrr!__annotations__rrrr#r#s G G Irr#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. callbackssystemc2g|_t| |y)N)r*super__init__)rr+ __class__s rr.zMemberlistProvider.__init__-s  rrcy)zReturns the latest memberlistNrrs rget_memberlistz!MemberlistProvider.get_memberlist1 r memberlistNcy)z1Sets the memberlist that this provider will watchNr)rr4s rset_memberlist_namez&MemberlistProvider.set_memberlist_name6r3rcg|_y)zStops watching the memberlistN)r*r1s rstopzMemberlistProvider.stop;s rrc:|jj|y)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'rcX||jvr|jj|yy)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_callbackHs( t~~ % NN ! !( +r)rN)rrrrrr Memberlistrr'r r.rr2r!r6r r8r;boolr> __classcell__)r/s@rr)r)&s"Hj\3./00!v!    c d  ( *s!23( ( *s!23 rr)N)abcr dataclassesrtypingrrrr rchromadb.configr r chromadb.typesr rr#r?r)rrrrGsY!&&1-" y $   &\ *$4*r