使得其成为Amrita插件

JohnRichard4096 · JohnRichard4096 · commit 623e45f3cbff · 2025-12-06T22:32:41.000+08:00
diff --git a/README.md b/README.md
@@ -1,9 +1,9 @@
 <div align="center">
-  <a href="https://github.com/JohnRichard4096/nonebot_plugin_omikuji/">
+  <a href="https://github.com/JohnRichard4096/amrita_plugin_omikuji/">
     <img src="https://github.com/user-attachments/assets/b5162036-5b17-4cf4-b0cb-8ec842a71bc6" width="200" alt="omikuji Logo">
   </a>
   <h1>Omikuji</h1>
-  <h3>适用于SuggarChat的御神签插件！</h3>
+  <h3>适用于AmritaBot的御神签插件！</h3>
 
   <p>
     <a href="https://pypi.org/project/nonebot-plugin-omikuji/">
@@ -16,47 +16,47 @@
       <img src="https://img.shields.io/badge/nonebot2-2.4.3+-blue?style=flat-square" alt="NoneBot Version">
     </a>
     <a href="LICENSE">
-      <img src="https://img.shields.io/github/license/LiteSuggarDEV/nonebot_plugin_omikuji?style=flat-square" alt="License">
+      <img src="https://img.shields.io/github/license/AmritaBot/amrita_plugin_omikuji?style=flat-square" alt="License">
     </a>
-    <a href="https://qm.qq.com/q/PFcfb4296m">
-      <img src="https://img.shields.io/badge/QQ%E7%BE%A4-1002495699-blue?style=flat-square" alt="QQ Group">
+    <a href="https://qm.qq.com/q/5URbtujxx6">
+      <img src="https://img.shields.io/badge/QQ%E7%BE%A4-1006893368-blue?style=flat-square" alt="QQ Group">
     </a>
   </p>
 </div>
 
 ## 🌸 简介
 
-**Omikuji（御神签）** 是一款基于大型语言模型（LLM）的 [NoneBot2](https://nonebot.dev/) 插件，专为 [SuggarChat](https://github.com/LiteSuggarDEV/SuggarChat) 框架设计。该插件为用户提供传统日本神社抽签体验的现代化数字版本，通过AI生成个性化、富有文化氛围的签文。
+**Omikuji（御神签）** 是一款基于大型语言模型（LLM）的 [NoneBot2](https://nonebot.dev/) 插件，专为 [AmritaBot](https://github.com/AmritaBot/Amrita) 框架设计。该插件为用户提供传统日本神社抽签体验的现代化数字版本，通过 AI 生成个性化、富有文化氛围的签文。
 
-御神签（おみくじ）是日本神道教中一种传统的占卜方式，参拜者在神社或寺庙中摇动签筒，随机抽取一支签，上面写着对未来的预言或建议。本插件将这一传统文化与现代AI技术相结合，每次抽取都会根据主题和运势等级生成独特的签文内容。
+御神签（おみくじ）是日本神道教中一种传统的占卜方式，参拜者在神社或寺庙中摇动签筒，随机抽取一支签，上面写着对未来的预言或建议。本插件将这一传统文化与现代 AI 技术相结合，每次抽取都会根据主题和运势等级生成独特的签文内容。
 
 ### 🌟 特性
 
-- **AI驱动签文生成**：利用大型语言模型生成富有创意和文化内涵的签文
+- **AI 驱动签文生成**：利用大型语言模型生成富有创意和文化内涵的签文
 - **多样化主题**：支持多个主题，包括综合运势、恋爱姻缘、学业考试、事业财运等
-- **智能缓存系统**：内置缓存机制，提高响应速度并减少API调用
-- **丰富的运势等级**：从大吉到大凶共7个等级，增加占卜体验的真实感
+- **智能缓存系统**：内置缓存机制，提高响应速度并减少 API 调用
+- **丰富的运势等级**：从大吉到大凶共 7 个等级，增加占卜体验的真实感
 - **灵活配置**：支持多种配置选项，可根据需求调整插件行为
-- **多平台支持**：基于NoneBot2开发，支持多种聊天平台
+- **多平台支持**：基于 NoneBot2 开发，支持多种聊天平台
 
 ## 🚀 安装
 
 ### 环境要求
 
 - Python 3.10+
 - NoneBot2 2.4.3+
-- SuggarChat 框架
-- 支持的LLM服务（如OpenAI、Anthropic等）
+- AmritaBot 框架
+- 支持的 LLM 服务（如 OpenAI、Anthropic 等）
 
-### 使用pip安装（推荐）
+### 使用 Amrita CLI 安装
 
-```bash
-pip install nonebot-plugin-omikuji
+```shell
+amrita plugin install amrita_plugin_omikuji
 ```
 
-### 使用uv安装
+### 使用 uv 安装
 
-```bash
+```shell
 uv add nonebot-plugin-omikuji
 ```
 
@@ -71,7 +71,7 @@ ENABLE_OMIKUJI=true
 # 是否交给模型进行二次响应（默认：False）
 OMIKUJI_SEND_BY_CHAT=false
 
-# 是否加入SuggarChat的系统提示（默认：True）
+# 是否加入AmritaBot的系统提示（默认：True）
 OMIKUJI_ADD_SYSTEM_PROMPT=true
 
 # 是否使用语料库的缓存（默认：True）
@@ -101,15 +101,19 @@ OMIKUJI_LONG_CACHE_UPDATE_MAX_COUNT=100
 ### 命令触发
 
 1. **随机主题抽签**：
+
    ```
    /omikuji
    ```
 
 2. **指定主题抽签**：
+
    ```
    /omikuji <主题>
    ```
+
    支持的主题包括：
+
    - 综合运势
    - 恋爱姻缘
    - 学业考试
@@ -129,24 +133,27 @@ OMIKUJI_LONG_CACHE_UPDATE_MAX_COUNT=100
 ### 别名触发
 
 也可以使用以下别名触发抽签：
+
 - `/御神签`
 - `/抽签`
 
 ### 聊天触发
 
-在启用了SuggarChat的环境中，也可以通过自然语言触发，例如：
+在启用了 AmritaBot 的环境中，也可以通过自然语言触发，例如：
+
 - "我想抽个签"
 - "给我来个御神签"
 
 ## 🧠 工作原理
 
 1. 用户触发抽签命令或通过聊天触发
 2. 插件根据主题和随机运势等级生成请求
-3. 调用配置的LLM服务生成符合要求的签文内容
+3. 调用配置的 LLM 服务生成符合要求的签文内容
 4. 将生成的签文按照传统御神签格式进行排版
 5. 返回给用户完整的签文体验
 
 签文通常包括：
+
 - 签文编号
 - 天启名称
 - 运势等级和主题
@@ -156,7 +163,7 @@ OMIKUJI_LONG_CACHE_UPDATE_MAX_COUNT=100
 
 ## 📁 缓存机制
 
-为了提高响应速度和减少API调用，插件实现了多层缓存机制：
+为了提高响应速度和减少 API 调用，插件实现了多层缓存机制：
 
 1. **短期缓存**：临时存储用户最近一次抽签结果
 2. **语料库缓存**：存储已生成的签文内容，按主题和运势等级分类
@@ -167,7 +174,7 @@ OMIKUJI_LONG_CACHE_UPDATE_MAX_COUNT=100
 ## 🤝 依赖
 
 - [nonebot2](https://github.com/nonebot/nonebot2)
-- [nonebot-plugin-suggarchat](https://github.com/LiteSuggarDEV/SuggarChat)
+- [Amrita](https://github.com/AmritaBot/Amrita)
 - [nonebot-adapter-onebot](https://github.com/nonebot/adapter-onebot)
 - [nonebot-plugin-localstore](https://github.com/nonebot/nonebot-plugin-localstore)
 - [nonebot-plugin-orm](https://github.com/nonebot/nonebot-plugin-orm)
@@ -179,7 +186,8 @@ OMIKUJI_LONG_CACHE_UPDATE_MAX_COUNT=100
 
 ## 🙏 鸣谢
 
-特别感谢以下项目和贡献者：
+特别感谢以下项目的贡献者：
 
 - [NoneBot2](https://github.com/nonebot/nonebot2)
-- [SuggarChat](https://github.com/LiteSuggarDEV/SuggarChat)
+- [Amrita](https://github.com/AmritaBot/Amrita)
+- [nonebot-plugin-suggarchat](https://github.com/LiteSuggarDEV/nonebot-plugin-suggarchat)
diff --git a/amrita_plugin_omikuji/__init__.py b/amrita_plugin_omikuji/__init__.py
@@ -1,16 +1,15 @@
 import contextlib
 import os
 from datetime import datetime
+from importlib import metadata
 
 from nonebot import get_driver, logger
 from nonebot.plugin import PluginMetadata, require
 
-require("nonebot_plugin_suggarchat")
-require("nonebot_plugin_localstore")
-require("nonebot_plugin_orm")
-from importlib import metadata
+require("amrita.plugins.chat")
+require("amrita.plugins.menu")
 
-from nonebot_plugin_suggarchat.API import Menu, ToolsManager
+from amrita.plugins.chat.API import ToolsManager
 
 from . import commands, llm_tool, sql_models
 from .cache import OmikujiCache
@@ -19,28 +18,25 @@
 
 __plugin_meta__ = PluginMetadata(
     name="御神签",
-    description="依赖SuggarChat的聊天御神签抽签插件模块",
+    description="依赖AmritaBot的聊天御神签抽签插件模块",
     usage="/omikuji [板块]\n/omikuji 解签\n或者使用聊天直接抽签。",
     type="application",
-    homepage="https://github.com/LiteSuggarDEV/nonebot_plugin_omikuji",
+    homepage="https://github.com/AmritaBot/amrita_plugin_omikuji",
     supported_adapters={"~onebot.v11"},
 )
 
-__all__ = ["commands", "llm_tool", "sql_models"]
-
 
 @get_driver().on_startup
 async def init():
     version = "Unknown"
     with contextlib.suppress(Exception):
-        version = metadata.version("nonebot_plugin_omikuji")
+        version = metadata.version("amrita_plugin_omikuji")
         if "dev" in version:
             logger.warning("当前版本为开发版本，可能存在不稳定情况！")
     logger.info(f"Loading OMIKUJI V{version}......")
     conf = get_config()
     if conf.enable_omikuji:
         ToolsManager().register_tool(TOOL_DATA)
-        Menu().reg_menu("omikuji", "抽御神签", "[可选]主题")
     logger.info("正在初始化缓存数据......")
     os.makedirs(get_cache_dir(), exist_ok=True)
     for cache in get_cache_dir().glob("*.json"):
@@ -50,3 +46,6 @@ async def init():
             if not data.timestamp.date() == datetime.now().date():
                 os.remove(str(cache))
     logger.info("缓存数据初始化完成！")
+
+
+__all__ = ["commands", "llm_tool", "sql_models"]
diff --git a/amrita_plugin_omikuji/cache.py b/amrita_plugin_omikuji/cache.py
diff --git a/amrita_plugin_omikuji/commands.py b/amrita_plugin_omikuji/commands.py
@@ -1,6 +1,7 @@
 import random
 import typing
 
+from amrita.plugins.menu import MatcherData
 from nonebot import on_command
 from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, Message, MessageEvent
 from nonebot.params import CommandArg
@@ -16,6 +17,9 @@
     priority=10,
     block=True,
     rule=lambda: get_config().enable_omikuji,
+    state=MatcherData(
+        name="omikuji", description="抽御神签", usage="/omikuji [主题]"
+    ).model_dump(),
 )
 
 
diff --git a/amrita_plugin_omikuji/config.py b/amrita_plugin_omikuji/config.py
@@ -9,13 +9,13 @@
 
 class Config(BaseModel):
     """
-    Configuration for nonebot_plugin_omikuji
+    Configuration for amrita_plugin_omikuji
     """
 
     omikuji_send_by_chat: bool = False  # 是否交给模型进行二次响应
     enable_omikuji: bool = True  # 是否启用
     omikuji_add_system_prompt: bool = (
-        True  # 是否加入SuggarChat的系统提示(生成更符合角色设定的答案)
+        True  # 是否加入AmritaBot的系统提示(生成更符合角色设定的答案)
     )
     omikuji_use_cache: bool = True  # 是否使用语料库的缓存（LLM生成后的缓存）
     omikuji_cache_expire_days: int = (
diff --git a/amrita_plugin_omikuji/llm_tool.py b/amrita_plugin_omikuji/llm_tool.py
@@ -1,20 +1,16 @@
 import typing
 
-from nonebot import get_bot, logger
-from nonebot.adapters.onebot.v11 import MessageEvent
-from nonebot_plugin_suggarchat.API import (
-    FunctionDefinitionSchema,
-    FunctionParametersSchema,
-    FunctionPropertySchema,
+from amrita.plugins.chat.API import (
     ToolContext,
     ToolData,
-    ToolFunctionSchema,
 )
+from nonebot import get_bot, logger
+from nonebot.adapters.onebot.v11 import MessageEvent
 
-from nonebot_plugin_omikuji.cache import cache_omikuji, get_cached_omikuji
+from amrita_plugin_omikuji.cache import cache_omikuji, get_cached_omikuji
 
 from .config import get_config
-from .models import OMIKUJI_THEMES, OmikujiData
+from .models import FUNC_META, OmikujiData
 from .utils import generate_omikuji
 
 LEVEL = ["大吉", "吉", "中吉", "小吉", "末吉", "凶", "大凶"]
@@ -58,23 +54,4 @@ async def omikuji(ctx: ToolContext):
     ctx.matcher.cancel_nonebot_process()
 
 
-FUNC_META = ToolFunctionSchema(
-    strict=True,
-    function=FunctionDefinitionSchema(
-        name="御神签(omikuji)",
-        description="抽取一个御神签",
-        parameters=FunctionParametersSchema(
-            type="object",
-            properties={
-                "theme": FunctionPropertySchema(
-                    type="string",
-                    description="御神签主题（如果包含不良内容则随机选择）",
-                    enum=OMIKUJI_THEMES,
-                )
-            },
-            required=["theme"],
-        ),
-    ),
-)
-
 TOOL_DATA = ToolData(data=FUNC_META, func=omikuji, custom_run=True)
diff --git a/amrita_plugin_omikuji/migrations/c303c8b3d73b_init.py b/amrita_plugin_omikuji/migrations/c303c8b3d73b_init.py
@@ -38,7 +38,7 @@ def upgrade(name: str = "") -> None:
         sa.Column("updated_date", sa.String(length=30), nullable=False),
         sa.PrimaryKeyConstraint("id", name=op.f("pk_omikuji_cache")),
         sa.UniqueConstraint("level", "theme", name="uq_omikuji_cache_level_theme"),
-        info={"bind_key": "nonebot_plugin_omikuji"},
+        info={"bind_key": "amrita_plugin_omikuji"},
     )
     with op.batch_alter_table("omikuji_cache", schema=None) as batch_op:
         batch_op.create_index(
diff --git a/amrita_plugin_omikuji/models.py b/amrita_plugin_omikuji/models.py
@@ -3,7 +3,7 @@
 import random
 from typing import Literal
 
-from nonebot_plugin_suggarchat.API import (
+from amrita.plugins.chat.API import (
     FunctionDefinitionSchema,
     FunctionParametersSchema,
     FunctionPropertySchema,
@@ -125,9 +125,9 @@ class OmikujiData(BaseModel):
 )
 
 FUNC_META = ToolFunctionSchema(
-    strict=False,
+    strict=True,
     function=FunctionDefinitionSchema(
-        name="御神签(omikuji)",
+        name="omikuji",
         description="抽取一个御神签",
         parameters=FunctionParametersSchema(
             type="object",
diff --git a/amrita_plugin_omikuji/sql_models.py b/amrita_plugin_omikuji/sql_models.py
diff --git a/amrita_plugin_omikuji/utils.py b/amrita_plugin_omikuji/utils.py
@@ -3,11 +3,11 @@
 from copy import deepcopy
 from datetime import datetime, timedelta
 
-from nonebot import logger
-from nonebot_plugin_suggarchat.API import (
+from amrita.plugins.chat.API import (
     config_manager,
     tools_caller,
 )
+from nonebot import logger
 
 from .cache import OmikujiCacheData
 from .config import get_config
diff --git a/pyproject.toml b/pyproject.toml
diff --git a/uv.lock b/uv.lock

Original file line number	Diff line number	Diff line change
`@@ -38,7 +38,7 @@ def upgrade(name: str = "") -> None:`
`38`	`38`	`sa.Column("updated_date", sa.String(length=30), nullable=False),`
`39`	`39`	`sa.PrimaryKeyConstraint("id", name=op.f("pk_omikuji_cache")),`
`40`	`40`	`sa.UniqueConstraint("level", "theme", name="uq_omikuji_cache_level_theme"),`
`41`		`- info={"bind_key": "nonebot_plugin_omikuji"},`
	`41`	`+ info={"bind_key": "amrita_plugin_omikuji"},`
`42`	`42`	`)`
`43`	`43`	`with op.batch_alter_table("omikuji_cache", schema=None) as batch_op:`
`44`	`44`	`batch_op.create_index(`