.`i@ ddlmZmZddlmZddlmZmZddlm Z ddl m Z e e Z GddeZeGdd ZeZd S) )ABCabstractmethod)Set) dataclassfield) init_logger) LoRARequestc<eZdZdZededededzfdZdS) LoRAResolveraBase class for LoRA adapter resolvers. This class defines the interface for resolving and fetching LoRA adapters. Implementations of this class should handle the logic for locating and downloading LoRA adapters from various sources (e.g. S3, cloud storage, etc.). base_model_name lora_namereturnNc KdS)aAbstract method to resolve and fetch a LoRA model adapter. Implements logic to locate and download LoRA adapter based on the name. Implementations might fetch from a blob storage or other sources. Args: base_model_name: The name/identifier of the base model to resolve. lora_name: The name/identifier of the LoRA model to resolve. Returns: Optional[LoRARequest]: The resolved LoRA model information, or None if the LoRA model cannot be found. N)selfr r s f/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/lora/resolver.py resolve_lorazLoRAResolver.resolve_loras " )__name__ __module__ __qualname____doc__rstrr rrrrr r s[ " /2 t    ^   rr c~eZdZUeeZeeefed<de efdZ dededdfdZ dedefd Z dS) _LoRAResolverRegistry)default_factory resolversrc4|jS)z"Get all registered resolver names.)rkeys)rs rget_supported_resolversz-_LoRAResolverRegistry.get_supported_resolvers/s~""$$$r resolver_nameresolverNcd||jvrtd||||j|<dS)zRegister a LoRA resolver. Args: resolver_name: Name to register the resolver under. resolver: The LoRA resolver instance to register. z`LoRA resolver %s is already registered, and will be overwritten by the new resolver instance %s.N)rloggerwarning)rr!r"s rregister_resolverz'_LoRAResolverRegistry.register_resolver3sF DN * * NN?    )1}%%%rc ||jvr9td|dt|j|j|S)aGet a registered resolver instance by name. Args: resolver_name: Name of the resolver to get. Returns: The resolver instance. Raises: KeyError: If the resolver is not found in the registry. zLoRA resolver 'z"' not found. Available resolvers: )rKeyErrorlistr)rr!s r get_resolverz"_LoRAResolverRegistry.get_resolverGsk  . .F-FF(,T^-@-@-B-B(C(CFF ~m,,r) rrrrdictrrr __annotations__rr r&r*rrrrr+s).t)D)D)DItC%&DDD%S%%%%111  1111(-#-,------rrN)abcrrcollections.abcr dataclassesrr vllm.loggerrvllm.lora.requestr rr$r rLoRAResolverRegistryrrrr3s$#######((((((((######)))))) X       3   : )-)-)-)-)-)-)- )-X-,..r