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

nonebot_plugin_githubmodels-0.1.9/LICENSE

@@ -0,0 +1,19 @@
+The MIT License (MIT)
+Copyright (c) 2024 lyqgzbl
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+the Software, and to permit persons to whom the Software is furnished to do so,
+subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

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

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

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

@@ -1,18 +1,18 @@
 import nonebot
-from openai import AsyncOpenAI
-from nonebot import on_command, get_plugin_config, require
+from nonebot import require, get_plugin_config
 require("nonebot_plugin_alconna")
 require("nonebot_plugin_htmlrender")
-from nonebot.adapters import Message
-from nonebot.params import CommandArg
+from openai import AsyncOpenAI
+from arclet.alconna import Args, Option, Alconna
 from nonebot.plugin import PluginMetadata
-from nonebot_plugin_alconna import UniMessage, Image 
 from nonebot_plugin_htmlrender import md_to_pic
+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
 
 endpoint = "https://models.inference.ai.azure.com"
@@ -22,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,
@@ -55,14 +65,16 @@ 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})
 
+    if REPLY_IMAGE:
+        pic = await md_to_pic(md=reply)
+        await UniMessage.image(raw=pic).send(reply_to=True)
+    else:
+        await UniMessage.text(reply).send(reply_to=True)
 
-    pic = await md_to_pic(md=reply)
-    await UniMessage.image(raw=pic).send(reply_to=True)
-		
 __plugin_meta__ = PluginMetadata(
     name="githubmodels",
     description="API 调用 GitHub Models 的大语言模型",

{nonebot_plugin_githubmodels-0.1.7 → nonebot_plugin_githubmodels-0.1.9}/nonebot_plugin_githubmodels/config.py

@@ -4,4 +4,5 @@ from pydantic import BaseModel, Field
 class Config(BaseModel):
     github_token: Optional[str] = None
     ai_model_name: str = "gpt-4o-mini"
-    max_context_length: int = Field(20, gt=0)
+    max_context_length: int = Field(20, gt=0)
+    ai_reply_image: bool = False

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

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