Soulter
54 lines
1.2 KiB
Python
54 lines
1.2 KiB
Python
from model.provider.provider import Provider as LLMProvider
|
|
from model.platform._platfrom import Platform
|
|
from type.plugin import *
|
|
from typing import List
|
|
from types import ModuleType
|
|
from dataclasses import dataclass
|
|
|
|
@dataclass
|
|
class RegisteredPlugin:
|
|
'''
|
|
注册在 AstrBot 中的插件。
|
|
'''
|
|
metadata: PluginMetadata
|
|
plugin_instance: object
|
|
module_path: str
|
|
module: ModuleType
|
|
root_dir_name: str
|
|
trig_cnt: int = 0
|
|
|
|
def reset_trig_cnt(self):
|
|
self.trig_cnt = 0
|
|
|
|
def trig(self):
|
|
self.trig_cnt += 1
|
|
|
|
def __str__(self) -> str:
|
|
return f"RegisteredPlugin({self.metadata}, {self.module_path}, {self.root_dir_name})"
|
|
|
|
|
|
RegisteredPlugins = List[RegisteredPlugin]
|
|
|
|
|
|
@dataclass
|
|
class RegisteredPlatform:
|
|
'''
|
|
注册在 AstrBot 中的平台。平台应当实现 Platform 接口。
|
|
'''
|
|
platform_name: str
|
|
platform_instance: Platform
|
|
origin: str = None # 注册来源
|
|
|
|
def __str__(self) -> str:
|
|
return self.platform_name
|
|
|
|
|
|
@dataclass
|
|
class RegisteredLLM:
|
|
'''
|
|
注册在 AstrBot 中的大语言模型调用。大语言模型应当实现 LLMProvider 接口。
|
|
'''
|
|
llm_name: str
|
|
llm_instance: LLMProvider
|
|
origin: str = None # 注册来源
|