nonebot-plugin-githubmodels 0.1.3__tar.gz → 0.1.5__tar.gz

{nonebot_plugin_githubmodels-0.1.3 → nonebot_plugin_githubmodels-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nonebot-plugin-githubmodels
-Version: 0.1.3
+Version: 0.1.5
 Summary: API 调用 GitHub Models 的 GPT-4o 模型
 Home-page: https://github.com/lyqgzbl/nonebot-plugin-githubmodels
 License: MIT

{nonebot_plugin_githubmodels-0.1.3 → nonebot_plugin_githubmodels-0.1.5}/nonebot_plugin_githubmodels/__init__.py

@@ -1,10 +1,17 @@
 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 TOKEN,MODEL_NAME,MAX_CONTEXT_LENGTH,Config
+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
 
 endpoint = "https://models.inference.ai.azure.com"
 client = AsyncOpenAI(
@@ -55,8 +62,6 @@ async def handle_function(args: Message = CommandArg()):
     reply = response.choices[0].message.content
     shared_context.append({"role": "assistant", "content": reply})
     
-
-
 __plugin_meta__ = PluginMetadata(
     name="githubmodels",
     description="API 调用 GitHub Models 的 GPT-4o 模型",

nonebot_plugin_githubmodels-0.1.5/nonebot_plugin_githubmodels/config.py

@@ -0,0 +1,7 @@
+from typing import Optional
+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)

{nonebot_plugin_githubmodels-0.1.3 → nonebot_plugin_githubmodels-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nonebot-plugin-githubmodels"
-version = "0.1.3"
+version = "0.1.5"
 description = "API 调用 GitHub Models 的 GPT-4o 模型"
 authors = ["lyqgzbl <admin@lyqgzbl.com>"]
 license = "MIT"

nonebot_plugin_githubmodels-0.1.3/nonebot_plugin_githubmodels/config.py

@@ -1,25 +0,0 @@
-from typing import Optional
-from nonebot import get_driver
-from nonebot import get_plugin_config
-
-class Config:
-    def __init__(self):
-        try:
-            self.github_token = get_driver().config.github_token  
-        except AttributeError:
-            self.github_token = None
-            
-        try:
-            self.model_name = get_driver().config.model_name
-        except AttributeError:
-            self.model_name = "gpt-4o-mini"
-            
-        try:
-            self.MAX_CONTEXT_LENGTH = get_driver().config.MAX_CONTEXT_LENGTH
-        except AttributeError:
-            self.MAX_CONTEXT_LENGTH = 20
-
-config = Config()  
-TOKEN = config.github_token 
-MODEL_NAME = config.model_name
-MAX_CONTEXT_LENGTH = config.MAX_CONTEXT_LENGTH