ug ddlmZddlmZddlmZmZmZddlm Z mZddl m Z m Z ddl mZGdde ZeGd d ZeeZGd d e e Zy ))abstractmethod) dataclass)AnyCallableList)EnforceOverrides overrides) ComponentSystem)SegmentcPeZdZdZededefdZedeegdfddfdZ y)SegmentDirectoryzA segment directory is a data interface that manages the location of segments. Concretely, this means that for clustered chroma, it provides the grpc endpoint for a segment.segmentreturncy)z3Return the segment residence for a given segment IDN)selfrs Z/var/www/openai/venv/lib/python3.12/site-packages/chromadb/segment/distributed/__init__.pyget_segment_endpointz%SegmentDirectory.get_segment_endpointscallbackNcy)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 strrrrrrrrr s]UBGBBB  'D1   rrc"eZdZUeed<eed<y)MemberidipN)rrrr__annotations__rrrr!r!s  G Grr!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_memberlist. r memberlistNcy)z1Sets the memberlist that this provider will watchNr)rr1s rset_memberlist_namez&MemberlistProvider.set_memberlist_name3r0rcg|_y)zStops watching the memberlistN)r'r.s rstopzMemberlistProvider.stop8s 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_callbackEs( t~~ % NN ! !( +r)rN)rrrrrr Memberlistrr$r r+rr/rr3r r5r8boolr; __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&rrrrDsY!&&1-"  y       &\ *$4*r