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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nonebot-plugin-githubmodels
-Version: 0.1.1
+Version: 0.1.3
 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.3}/nonebot_plugin_githubmodels/__init__.py

@@ -4,17 +4,15 @@ from nonebot import on_command
 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
 
-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 +39,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,6 +52,9 @@ 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(
@@ -62,4 +63,5 @@ __plugin_meta__ = PluginMetadata(
     usage="AI",
     type="application",
     homepage="https://github.com/lyqgzbl/nonebot-plugin-githubmodels",
+    supported_adapters=None,
 )

nonebot_plugin_githubmodels-0.1.3/nonebot_plugin_githubmodels/config.py

@@ -0,0 +1,25 @@
+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

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nonebot-plugin-githubmodels"
-version = "0.1.1"
+version = "0.1.3"
 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]