o ۷i @szddlZddlmZddlmZddlmZeeZ dZ dZ da de dee egefffd d Zdd d Zgd ZdS)N)Callable)Anyzvllm_omni.general_pluginszvllm_omni.platform_pluginsFgroupreturnc Csddlm}tj}||d}t|dkrtd|iS|tk}|r%tjntj}|d||D] }|d|j |j q/|durB|di}|D]1}|dusQ|j |vrw|dur[|d|j z | }|||j <WqFt yvt d |j YqFwqF|S) Nr) entry_pointsrzNo plugins for group %s found.zAvailable plugins for group %s:z - %s -> %sz^All plugins in this group will be loaded. Set `VLLM_PLUGINS` to control which plugins to load.zLoading plugin %szFailed to load plugin %s)importlib.metadatarenvs VLLM_PLUGINSlenloggerdebugOMNI_DEFAULT_PLUGINS_GROUPinfonamevalueload Exception exception) rrallowed_pluginsdiscovered_pluginsis_default_group log_levelpluginpluginsfuncrP/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm_omni/plugins/__init__.pyload_omni_plugins_by_groups4        rcCs.trdSdattd}|D]}|qdS)zWARNING: plugins can be loaded for multiple times in different processes. They should be designed in a way that they can be loaded multiple times without causing issues. NTr)omni_plugins_loadedrrvalues)rrrrrload_omni_general_plugins=s  r!)r!rOMNI_PLATFORM_PLUGINS_GROUP)rN)loggingcollections.abcrtypingr vllm.envsr getLogger__name__r rr"rstrdictrr!__all__rrrrs    " %