add: 添加命令ai model list的逻辑

maubot_llmplus/aibot.py

@@ -148,6 +148,30 @@ class AiBotPlugin(Plugin):
             return Anthropic(self.config, self.name, self.http)
         raise ValueError(f"unknown backend type {use_platform}")
 
+    """
+        父命令
+    """
+    @command.new(name="ai", require_subcommand=True)
+    async def ai_command(self, event: MessageEvent) -> None:
+        pass
+
+    """
+    """
+    @ai_command.subcommand(help="")
+    async def info(self, event: MessageEvent) -> None:
+        pass
+
+    @ai_command.subcommand(help="")
+    @command.argument("argus")
+    async def model(self, event: MessageEvent, argus: str):
+        # 如果是list表示查看当前可以使用的模型列表
+        if argus == 'list':
+            platform = self.get_ai_platform()
+            models = platform.list_models()
+            await event.reply("\n".join(models))
+
+        # 如果不是，如果是其他的名称，表示这是一个模型名
+
     @classmethod
     def get_config_class(cls) -> Type[BaseProxyConfig]:
         return Config

maubot_llmplus/local_paltform.py

@@ -1,4 +1,5 @@
 import json
+from typing import List
 
 from aiohttp import ClientSession
 from maubot import Plugin
@@ -40,6 +41,14 @@ class Ollama(Platform):
                 model=response_json['model']
             )
 
+    async def list_models(self) -> List[str]:
+        full_url = f"{self.url}/api/tags"
+        async with self.http.get(full_url) as response:
+            if response.status != 200:
+                return []
+            response_data = json.loads(await response.json())
+            return [model['name'] for model in response_data]
+
     def get_type(self) -> str:
         return "local_ai"
 

maubot_llmplus/platforms.py

@@ -54,6 +54,9 @@ class Platform:
     async def create_chat_completion(self, plugin: Plugin, evt: MessageEvent) -> ChatCompletion:
         raise NotImplementedError()
 
+    async def list_models(self) -> List[str]:
+        raise NotImplementedError()
+
     def get_type(self) -> str:
         raise NotImplementedError()