Ig ddlmZddlmZmZmZddlmZmZddlm Z m Z ddl m Z Gdde Z eeZGdd e eZy ) )abstractmethod)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 b/var/www/html/answerous/venv/lib/python3.12/site-packages/chromadb/segment/distributed/__init__.pyget_segment_endpointz%SegmentDirectory.get_segment_endpoint scallbackNcy)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 strrrrrrrr r s]UBGBBB  'D1   rr 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)rr+s rset_memberlist_namez&MemberlistProvider.set_memberlist_name,r*rcg|_y)zStops watching the memberlistN)r!r(s rstopzMemberlistProvider.stop1s 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_callback6s 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_callback>s( t~~ % NN ! !( +r)rN)rrrrrr Memberlistr__annotations__r r%rr)rr-rr/r2boolr5 __classcell__)r&s@rr r s"Hj\3./00!v!    c d  ( *s!23( ( *s!23 rr N)abcrtypingrrrrrchromadb.configr r chromadb.typesr r rr6r rrrr>s=&&1-"  y  #Y *$4*r