From 3fc1347bba45a26ba963cc7137180b5b5cff2d40 Mon Sep 17 00:00:00 2001 From: Soulter <905617992@qq.com> Date: Tue, 12 Mar 2024 20:00:02 +0800 Subject: [PATCH] fix: plugin register management --- util/plugin_dev/api/v1/register.py | 14 -------------- 1 file changed, 14 deletions(-) diff --git a/util/plugin_dev/api/v1/register.py b/util/plugin_dev/api/v1/register.py index ee1983450..9c731f6c5 100644 --- a/util/plugin_dev/api/v1/register.py +++ b/util/plugin_dev/api/v1/register.py @@ -21,13 +21,6 @@ def register_platform(platform_name: str, platform_instance: Platform, context: if platform.platform_name == platform_name: raise ValueError(f"Platform {platform_name} has been registered.") - # check - should_attrs = Platform.__dir__() - has_attrs = platform_instance.__dir__() - - if not all([attr in has_attrs for attr in should_attrs]): - raise ValueError(f"Platform {platform_name} should implement all methods in LLMProvider.") - context.platforms.append(RegisteredPlatform(platform_name, platform_instance)) def register_llm(llm_name: str, llm_instance: LLMProvider, context: GlobalObject) -> None: @@ -43,13 +36,6 @@ def register_llm(llm_name: str, llm_instance: LLMProvider, context: GlobalObject if llm.llm_name == llm_name: raise ValueError(f"LLMProvider {llm_name} has been registered.") - # check - should_attrs = LLMProvider.__dir__() - has_attrs = llm_instance.__dir__() - - if not all([attr in has_attrs for attr in should_attrs]): - raise ValueError(f"LLMProvider {llm_name} should implement all methods in LLMProvider.") - context.llms.append(RegisteredLLM(llm_name, llm_instance)) def unregister_platform(platform_name: str, context: GlobalObject) -> None: