nonebot-plugin-githubmodels 0.1.8__tar.gz → 0.1.9__tar.gz

{nonebot_plugin_githubmodels-0.1.8 → nonebot_plugin_githubmodels-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nonebot-plugin-githubmodels
-Version: 0.1.8
+Version: 0.1.9
 Summary: API 调用 GitHub Models 的大语言模型
 Home-page: https://github.com/lyqgzbl/nonebot-plugin-githubmodels
 License: MIT

{nonebot_plugin_githubmodels-0.1.8 → nonebot_plugin_githubmodels-0.1.9}/nonebot_plugin_githubmodels/__init__.py

@@ -1,19 +1,16 @@
 import nonebot
-from .config import Config
-from nonebot import require
-from openai import AsyncOpenAI
-from nonebot import on_command
+from nonebot import require, get_plugin_config
 require("nonebot_plugin_alconna")
 require("nonebot_plugin_htmlrender")
-from nonebot.adapters import Message
-from nonebot import get_plugin_config
-from nonebot.params import CommandArg
+from openai import AsyncOpenAI
+from arclet.alconna import Args, Option, Alconna
 from nonebot.plugin import PluginMetadata
 from nonebot_plugin_htmlrender import md_to_pic
-from nonebot_plugin_alconna import UniMessage, Image 
+from nonebot_plugin_alconna import UniMessage, on_alconna, Match
+from .config import Config
 
 plugin_config = get_plugin_config(Config)
-TOKEN = plugin_config.github_token 
+TOKEN = plugin_config.github_token
 MODEL_NAME = plugin_config.ai_model_name
 REPLY_IMAGE = plugin_config.ai_reply_image
 MAX_CONTEXT_LENGTH = plugin_config.max_context_length
@@ -25,32 +22,42 @@ client = AsyncOpenAI(
 )
 shared_context = []
 
-AI = on_command("AI", priority=10, block=True)
+ai = on_alconna(
+    Alconna(
+        "AI",
+        Args["user_input?", str],
+        Option("-重置|--reset"),
+    ),
+    use_cmd_start=True,
+    block=True,
+    aliases={"ai"},
+)
 
-@AI.handle()
-async def handle_function(args: Message = CommandArg()):
+@ai.assign("resett")
+async def ai_reset():
     global shared_context
-    user_input = args.extract_plain_text().strip()
+    shared_context = []
+    await ai.finish("上下文已重置")
+
+@ai.handle()
+async def handle_function(user_input: Match[str]):
+    if user_input.available:
+        ai.set_path_arg("user_input", user_input.result)
 
-    if user_input.lower() == "重置":
-        shared_context = []
-        await AI.finish("上下文已重置")
-    
-    if not user_input:
-        await AI.finish("请输入有效的问题")
-    
+@ai.got_path("user_input", prompt="请输入有效问题")
+async def got_location(user_input: str):
+    global shared_context
     shared_context.append({"role": "user", "content": user_input})
-    
     if len(shared_context) > MAX_CONTEXT_LENGTH:
         shared_context = shared_context[-MAX_CONTEXT_LENGTH:]
-    
+
     messages = [
         {
             "role": "system",
-            "content": "回答尽量简练，请始终用中文回答。",
+            "content": "回答尽量简练,请始终用中文回答",
         }
     ] + shared_context
-    
+
     response = await client.chat.completions.create(
         messages=messages,
         model=MODEL_NAME,
@@ -58,7 +65,7 @@ async def handle_function(args: Message = CommandArg()):
         max_tokens=500,
         top_p=1,
     )
-    
+
     reply = response.choices[0].message.content
     shared_context.append({"role": "assistant", "content": reply})
 
@@ -67,7 +74,7 @@ async def handle_function(args: Message = CommandArg()):
         await UniMessage.image(raw=pic).send(reply_to=True)
     else:
         await UniMessage.text(reply).send(reply_to=True)
-			
+
 __plugin_meta__ = PluginMetadata(
     name="githubmodels",
     description="API 调用 GitHub Models 的大语言模型",
@@ -75,4 +82,4 @@ __plugin_meta__ = PluginMetadata(
     type="application",
     homepage="https://github.com/lyqgzbl/nonebot-plugin-githubmodels",
     supported_adapters=None,
-)
+)

{nonebot_plugin_githubmodels-0.1.8 → nonebot_plugin_githubmodels-0.1.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nonebot-plugin-githubmodels"
-version = "0.1.8"
+version = "0.1.9"
 description = "API 调用 GitHub Models 的大语言模型"
 authors = ["lyqgzbl <admin@lyqgzbl.com>"]
 license = "MIT"