nonebot-plugin-githubmodels 0.1.1__tar.gz → 0.1.4__tar.gz

{nonebot_plugin_githubmodels-0.1.1 → nonebot_plugin_githubmodels-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nonebot-plugin-githubmodels
-Version: 0.1.1
+Version: 0.1.4
 Summary: API 调用 GitHub Models 的 GPT-4o 模型
 Home-page: https://github.com/lyqgzbl/nonebot-plugin-githubmodels
 License: MIT
@@ -14,6 +14,6 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Requires-Dist: nonebot2 (>=2.0.0,<3.0.0)
+Requires-Dist: nonebot2 (>=2.2.1,<3.0.0)
 Requires-Dist: openai (>=1.44.1,<2.0.0)
 Project-URL: Repository, https://github.com/lyqgzbl/nonebot-plugin-githubmodels

{nonebot_plugin_githubmodels-0.1.1 → nonebot_plugin_githubmodels-0.1.4}/nonebot_plugin_githubmodels/__init__.py

@@ -1,20 +1,25 @@
 import nonebot
 from openai import AsyncOpenAI
 from nonebot import on_command
+from nonebot import get_plugin_config
 from nonebot.adapters import Message
 from nonebot.params import CommandArg
 from nonebot.plugin import PluginMetadata
+from .config import Config
+
+plugin_config = get_plugin_config(Config)
+
+TOKEN = plugin_config.github_token 
+MODEL_NAME = plugin_config.ai_model_name
+MAX_CONTEXT_LENGTH = plugin_config.MAX_CONTEXT_LENGTH
 
-config = nonebot.get_driver().config
-token = config.github_token
 endpoint = "https://models.inference.ai.azure.com"
-model_name = "gpt-4o-mini"
 client = AsyncOpenAI(
     base_url=endpoint,
-    api_key=token,
+    api_key=TOKEN,
 )
+
 shared_context = []
-MAX_CONTEXT_LENGTH = 20  # 上下文最大保留条数
 AI = on_command("AI", priority=10, block=True)
 
 @AI.handle()
@@ -41,7 +46,7 @@ async def handle_function(args: Message = CommandArg()):
     
     response = await client.chat.completions.create(
         messages=messages,
-        model=model_name,
+        model=MODEL_NAME,
         temperature=1,
         max_tokens=500,
         top_p=1,
@@ -54,12 +59,14 @@ async def handle_function(args: Message = CommandArg()):
         shared_context = shared_context[-MAX_CONTEXT_LENGTH:]
     
     await AI.send(reply, reply_message=True)
-
-
+    reply = response.choices[0].message.content
+    shared_context.append({"role": "assistant", "content": reply})
+    
 __plugin_meta__ = PluginMetadata(
     name="githubmodels",
     description="API 调用 GitHub Models 的 GPT-4o 模型",
     usage="AI",
     type="application",
     homepage="https://github.com/lyqgzbl/nonebot-plugin-githubmodels",
+    supported_adapters=None,
 )

nonebot_plugin_githubmodels-0.1.4/nonebot_plugin_githubmodels/config.py

@@ -0,0 +1,14 @@
+from typing import Optional
+from pydantic import BaseModel, field_validator
+
+class Config(BaseModel):
+    github_token: Optional[str] = None
+    ai_model_name: str = "gpt-4o-mini"
+    MAX_CONTEXT_LENGTH: int = 20
+
+    @field_validator("MAX_CONTEXT_LENGTH")
+    @classmethod
+    def check_max_context_length(cls, v: int) -> int:
+        if v > 0:
+            return v
+        raise ValueError("MAX_CONTEXT_LENGTH must be greater than 0")

{nonebot_plugin_githubmodels-0.1.1 → nonebot_plugin_githubmodels-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nonebot-plugin-githubmodels"
-version = "0.1.1"
+version = "0.1.4"
 description = "API 调用 GitHub Models 的 GPT-4o 模型"
 authors = ["lyqgzbl <admin@lyqgzbl.com>"]
 license = "MIT"
@@ -9,7 +9,7 @@ repository = "https://github.com/lyqgzbl/nonebot-plugin-githubmodels"
 
 [tool.poetry.dependencies]
 python = "^3.8"
-nonebot2 = "^2.0.0"
+nonebot2 = "^2.2.1"
 openai = "^1.44.1"
 
 [build-system]