o žº¥gâ ã@s˜ddlmZddlmZddlmZmZmZddlm Z mZddl m Z m Z ddl mZGdd„de ƒZeGd d „d ƒƒZeeZGd d „d e e ƒZd S)é)Úabstractmethod)Ú dataclass)ÚAnyÚCallableÚList)ÚEnforceOverridesÚ overrides)Ú ComponentÚSystem)ÚSegmentc@sFeZdZdZededefdd„ƒZedeegdfddfdd „ƒZ dS) ÚSegmentDirectoryz±A 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.ÚsegmentÚreturncCódS)z3Return the segment residence for a given segment IDN©)Úselfr rrú\/var/www/visachat/venv/lib/python3.10/site-packages/chromadb/segment/distributed/__init__.pyÚget_segment_endpointsz%SegmentDirectory.get_segment_endpointÚcallbackNcCr)zARegister a callback that will be called when a segment is updatedNr©rrrrrÚ!register_updated_segment_callbacksz2SegmentDirectory.register_updated_segment_callback) Ú__name__Ú __module__Ú __qualname__Ú__doc__rr Ústrrrrrrrrr s ÿþr c@seZdZUeed<eed<dS)ÚMemberÚidÚipN)rrrrÚ__annotations__rrrrrs  rcs¬eZdZUdZeeegefed<de f‡fdd„ Z e defdd„ƒZ e d e dd fd d „ƒZedd d„ƒZdeegefdd fdd„Zdeegefdefdd„Z‡ZS)ÚMemberlistProviderzÐReturns 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.Ú callbacksÚsystemcsg|_tƒ |¡dS)N)r!ÚsuperÚ__init__)rr"©Ú __class__rrr$*szMemberlistProvider.__init__rcCr)zReturns the latest memberlistNr©rrrrÚget_memberlist.óz!MemberlistProvider.get_memberlistÚ memberlistNcCr)z1Sets the memberlist that this provider will watchNr)rr*rrrÚset_memberlist_name3r)z&MemberlistProvider.set_memberlist_namecCs g|_dS)zStops watching the memberlistN)r!r'rrrÚstop8s zMemberlistProvider.stoprcCs|j |¡dS)zÑRegisters 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!ÚappendrrrrÚ$register_updated_memberlist_callback=sz7MemberlistProvider.register_updated_memberlist_callbackcCs||jvr |j |¡dSdS)zŸUnregisters a callback that was previously registered. Returns True if the callback was successfully unregistered, False if it was not ever registered.TF)r!ÚremoverrrrÚ&unregister_updated_memberlist_callbackEs  z9MemberlistProvider.unregister_updated_memberlist_callback)rN)rrrrrrÚ Memberlistrrr r$rr(rr+rr,r.Úboolr0Ú __classcell__rrr%rr #s(   ÿ þ ÿþr N)ÚabcrÚ dataclassesrÚtypingrrrrrÚchromadb.configr r Úchromadb.typesr r rr1r rrrrÚs