laoxong/AstrBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: plugin register management

Browse Source
This commit is contained in:
Soulter
2024-03-12 20:00:02 +08:00
parent e643eea365
commit 3fc1347bba
1 changed files with 0 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
util/plugin_dev/api/v1/register.py
-14
View File
@@ -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: