Compare commits

...

22 Commits

Author SHA1 Message Date
c8dd126042 🐛 移除make_chat函数中的tool_choice参数以简化函数调用 2024-12-29 16:33:25 +08:00
9ff8beb4d9 🐛 修复插件加载逻辑,移除多余的空行以提高代码可读性 2024-12-29 16:07:05 +08:00
3003dfad55 🐛 优化插件加载逻辑,修复内置插件加载的问题,动态加载内置插件并增强对sys.path下包的支持 2024-12-29 15:57:11 +08:00
fb428ffc19 更新安装文档,修正链接,添加小棉工具的相关信息 2024-12-29 15:01:10 +08:00
4b2676b9fc 更新marsho函数以处理tool_calls,优化函数调用参数,添加占位符参数以兼容部分模型(如GLM) 2024-12-29 06:33:52 +08:00
7c6319b839 更改主页链接引用 2024-12-25 13:23:43 +08:00
4b7e9d14f7 添加旧readme,版本发布预计时间 2024-12-25 13:18:54 +08:00
0c4835e75b Merge branch 'main' of github.com:LiteyukiStudio/nonebot-plugin-marshoai 2024-12-24 00:54:20 +08:00
3600b62176 添加贡献者展示组件,并更新项目文档以感谢贡献者 2024-12-24 00:54:16 +08:00
6631d84705 无效化需要使用ob11适配器的chat模块 2024-12-24 00:50:51 +08:00
9ba4f0cfa1 优化内存模块导入,确保正确获取数据文件 2024-12-24 00:40:05 +08:00
e4d9fef670 添加小棉工具弃用提示 2024-12-24 00:29:48 +08:00
1c74ddca7d 尝试调教pre-commit 2024-12-23 23:56:08 +08:00
83a5f6ae5d 再次修改description,尝试重新isort 2024-12-23 23:50:56 +08:00
f9dc5e500e 添加强制设置昵称配置项,移动测试插件到单独文件夹,稍微修改memory插件description(应该还是无法达到预期喵) 2024-12-23 23:36:47 +08:00
ba6b02d68e 新增marsho.status命令 2024-12-19 23:24:48 +08:00
c503228a5f 修复package.json语法错误 2024-12-17 23:47:58 +08:00
b9983330ab 重新触发pre-commit检查 2024-12-17 23:37:43 +08:00
19363b22ac 修复函数调用名错误,补充config 2024-12-17 23:33:53 +08:00
8b5a57d223 🐛 修复AI调用名的格式,将点替换为下划线 2024-12-17 23:17:05 +08:00
XuChenXu
9cca629b87 记忆系统实现 (#29)
*  添加记忆系统

* 🎨 black优化格式

* 🐛 删除apscheduler

*  将记忆插件转换为插件形式
2024-12-17 22:56:57 +08:00
Nya_Twisuki
b331a209c3 MegaKits插件移植 (#28)
* 新增了萌百插件(meowiki)

* 更新萌百搜索

* 删除萌百插件, 结束开发

* 新建MegaKits插件

* 修复

* 摩尔斯电码加密/解码

* 猫语转换/翻译
2024-12-17 22:54:04 +08:00
32 changed files with 661 additions and 78 deletions

5
.gitignore vendored
View File

@@ -187,4 +187,7 @@ docs/.vitepress/cache
docs/.vitepress/dist
# viztracer
result.json
result.json
data/*
marshoplugins/*

View File

@@ -27,6 +27,7 @@ _✨ 使用 OpenAI 标准格式 API 的聊天机器人插件 ✨_
插件内置了猫娘小棉(Marsho)的人物设定,可以进行可爱的聊天!
*谁不喜欢回复消息快又可爱的猫娘呢?*
**对 OneBot 以外的适配器与非 GitHub Models API的支持未经过完全验证。**
**MarshoAI v1.0.0 预计在春节前后发布。重写好累啊QwQ**
[Melobot 实现](https://github.com/LiteyukiStudio/marshoai-melo)
## 🐱 设定

0
docs/.vitepress/config/zh.ts Executable file → Normal file
View File

View File

@@ -0,0 +1,23 @@
<script setup lang="ts">
const contributorImgSrc = `https://contrib.rocks/image?repo=LiteyukiStudio/nonebot-plugin-marshoai`
const contributorsUrl = `https://github.com/LiteyukiStudio/nonebot-plugin-marshoai/graphs/contributors`
</script>
<template>
<div class="contributor-bar">
<a :href="contributorsUrl">
<div class="contributor-list">
<img :src=contributorImgSrc alt="Contributors">
</div>
</a>
</div>
</template>
<style scoped>
.contributor-bar {
display: flex;
flex-direction: column;
align-items: center;
}
</style>

View File

@@ -117,6 +117,7 @@ Add options in the `.env` file from the diagram below in nonebot2 project.
| MARSHOAI_DEFAULT_MODEL | `str` | `gpt-4o-mini` | The default model of Marsho |
| MARSHOAI_PROMPT | `str` | Catgirl Marsho's character prompt | Marsho's basic system prompt **※Some models(o1 and so on) don't support it** |
| MARSHOAI_ADDITIONAL_PROMPT | `str` | | Marsho's external system prompt |
| MARSHOAI_ENFORCE_NICKNAME | `bool` | `true` | Enforce user to set nickname or not |
| MARSHOAI_POKE_SUFFIX | `str` | `揉了揉你的猫耳` | When double click Marsho who connected to OneBot adapter, the chat content. When it's empty string, double click function is off. Such as, the default content is `*[昵称]揉了揉你的猫耳。` |
| MARSHOAI_AZURE_ENDPOINT | `str` | `https://models.inference.ai.azure.com` | OpenAI standard API |
| MARSHOAI_TEMPERATURE | `float` | `null` | temperature parameter |

View File

@@ -9,7 +9,9 @@ order: 2
扩展分为两类,一类为插件,一类为工具。
- 插件
- 工具(由于开发的不便利性已经停止维护未来可能会放弃支持如有需求请看README中的内容我们不推荐再使用此功能)
- 工具(由于开发的不便利性已经停止维护未来可能会放弃支持如有需求请看README中的内容我们不推荐再使用此功能)
**`v1.0.0`之前的版本不支持小棉插件。**
## 插件
@@ -57,7 +59,11 @@ async def weather(location: str) -> str:
## 函数调用参数
`on_function_call`装饰器用于标记一个函数为function call`description`参数用于描述这个函数的作用,`params`方法用于定义函数的参数,`String``Integer`等是OpenAI API接受的参数的类型`description`是参数的描述。这些都是给AI看的AI会根据这些信息来调用函数。
`on_function_call`装饰器用于标记一个函数为function call`description`参数用于描述这个函数的作用,`params`方法用于定义函数的参数,`String``Integer`等是OpenAI API接受的参数的类型`description`是参数的描述。这些都是给AI看的AI会根据这些信息来调用函数。
:::warning
参数名不得为`placeholder`。此参数名是Marsho内部保留的用于保证兼容性的占位参数。
:::
```python
@on_function_call(description="可以用于算命").params(

View File

@@ -39,3 +39,11 @@ pre-commit install # 安装 pre-commit 钩子
- [`Google Docstring`](https://sphinxcontrib-napoleon.readthedocs.io/en/latest/example_google.html) 文档规范
可以在编辑器中安装相应的插件进行辅助
## 其他
感谢以下的贡献者们:
<ContributorsBar />
<script setup> import ContributorsBar from '../../components/ContributorsBar.vue' </script>

0
docs/zh/index.md Executable file → Normal file
View File

View File

@@ -0,0 +1,151 @@
---
title: 安装
---
## 💿 安装
<details open>
<summary>使用 nb-cli 安装</summary>
在 nonebot2 项目的根目录下打开命令行, 输入以下指令即可安装
nb plugin install nonebot-plugin-marshoai
</details>
<details>
<summary>使用包管理器安装</summary>
在 nonebot2 项目的插件目录下, 打开命令行, 根据你使用的包管理器, 输入相应的安装命令
<details>
<summary>pip</summary>
pip install nonebot-plugin-marshoai
</details>
<details>
<summary>pdm</summary>
pdm add nonebot-plugin-marshoai
</details>
<details>
<summary>poetry</summary>
poetry add nonebot-plugin-marshoai
</details>
<details>
<summary>conda</summary>
conda install nonebot-plugin-marshoai
</details>
打开 nonebot2 项目根目录下的 `pyproject.toml` 文件, 在 `[tool.nonebot]` 部分追加写入
plugins = ["nonebot_plugin_marshoai"]
</details>
## 🤖 获取 token(GitHub Models)
- 新建一个[personal access token](https://github.com/settings/tokens/new)**不需要给予任何权限**。
- 将新建的 token 复制,添加到`.env`文件中的`marshoai_token`配置项中。
## 🎉 使用
发送`marsho`指令可以获取使用说明(若在配置中自定义了指令前缀请使用自定义的指令前缀)。
#### 👉 戳一戳
当 nonebot 连接到支持的 OneBot v11 实现端时,可以接收头像双击戳一戳消息并进行响应。详见`MARSHOAI_POKE_SUFFIX`配置项。
## 🛠️ ~~小棉工具~~(已弃用)
小棉工具(MarshoTools)是`v0.5.0`版本的新增功能,支持加载外部函数库来为 Marsho 提供 Function Call 功能。
## 🧩 小棉插件
小棉插件是`v1.0.0`的新增功能,替代旧的小棉工具功能。[使用文档](https://marsho.liteyuki.icu/dev/extension)
## 👍 夸赞名单
夸赞名单存储于插件数据目录下的`praises.json`里(该目录路径会在 Bot 启动时输出到日志),当配置项为`true`
时发起一次聊天后自动生成,包含人物名字与人物优点两个基本数据。
存储于其中的人物会被 Marsho “认识”和“喜欢”。
其结构类似于:
```json
{
"like": [
{
"name": "Asankilp",
"advantages": "赋予了Marsho猫娘人格使用vim与vscode为Marsho写了许多代码使Marsho更加可爱"
},
{
"name": "神羽(snowykami)",
"advantages": "人脉很广,经常找小伙伴们开银趴,很会写后端代码"
},
...
]
}
```
## ⚙️ 可配置项
在 nonebot2 项目的`.env`文件中添加下表中的配置
#### 插件行为
| 配置项 | 类型 | 默认值 | 说明 |
| ------------------------ | ------ | ------- | ---------------- |
| MARSHOAI_USE_YAML_CONFIG | `bool` | `false` | 是否使用 YAML 配置文件格式 |
| MARSHOAI_DEVMODE | `bool` | `false` | 是否启用开发者模式 |
#### Marsho 使用方式
| 配置项 | 类型 | 默认值 | 说明 |
| --------------------- | ---------- | ----------- | ----------------- |
| MARSHOAI_DEFAULT_NAME | `str` | `marsho` | 调用 Marsho 默认的命令前缀 |
| MARSHOAI_ALIASES | `set[str]` | `set{"小棉"}` | 调用 Marsho 的命令别名 |
| MARSHOAI_AT | `bool` | `false` | 决定是否使用at触发 |
| MARSHOAI_MAIN_COLOUR | `str` | `FFAAAA` | 主题色,部分工具和功能可用 |
#### AI 调用
| 配置项 | 类型 | 默认值 | 说明 |
| -------------------------------- | ------- | --------------------------------------- | --------------------------------------------------------------------------------------------- |
| MARSHOAI_TOKEN | `str` | | 调用 AI API 所需的 token |
| MARSHOAI_DEFAULT_MODEL | `str` | `gpt-4o-mini` | Marsho 默认调用的模型 |
| MARSHOAI_PROMPT | `str` | 猫娘 Marsho 人设提示词 | Marsho 的基本系统提示词 **※部分模型(o1等)不支持系统提示词。** |
| MARSHOAI_ADDITIONAL_PROMPT | `str` | | Marsho 的扩展系统提示词 |
| MARSHOAI_ENFORCE_NICKNAME | `bool` | `true` | 是否强制用户设置昵称 |
| MARSHOAI_POKE_SUFFIX | `str` | `揉了揉你的猫耳` | 对 Marsho 所连接的 OneBot 用户进行双击戳一戳时,构建的聊天内容。此配置项为空字符串时,戳一戳响应功能会被禁用。例如,默认值构建的聊天内容将为`*[昵称]揉了揉你的猫耳。` |
| MARSHOAI_AZURE_ENDPOINT | `str` | `https://models.inference.ai.azure.com` | OpenAI 标准格式 API 端点 |
| MARSHOAI_TEMPERATURE | `float` | `null` | 推理生成多样性(温度)参数 |
| MARSHOAI_TOP_P | `float` | `null` | 推理核采样参数 |
| MARSHOAI_MAX_TOKENS | `int` | `null` | 最大生成 token 数 |
| MARSHOAI_ADDITIONAL_IMAGE_MODELS | `list` | `[]` | 额外添加的支持图片的模型列表,例如`hunyuan-vision` |
#### 功能开关
| 配置项 | 类型 | 默认值 | 说明 |
| --------------------------------- | ------ | ------ | -------------------------- |
| MARSHOAI_ENABLE_SUPPORT_IMAGE_TIP | `bool` | `true` | 启用后用户发送带图请求时若模型不支持图片,则提示用户 |
| MARSHOAI_ENABLE_NICKNAME_TIP | `bool` | `true` | 启用后用户未设置昵称时提示用户设置 |
| MARSHOAI_ENABLE_PRAISES | `bool` | `true` | 是否启用夸赞名单功能 |
| MARSHOAI_ENABLE_TOOLS | `bool` | `false` | 是否启用小棉工具 |
| MARSHOAI_ENABLE_PLUGINS | `bool` | `true` | 是否启用小棉插件 |
| MARSHOAI_PLUGINS | `list[str]` | `[]` | 要从`sys.path`加载的插件的名称例如从pypi安装的包 |
| MARSHOAI_PLUGIN_DIRS | `list[str]` | `[]` | 插件目录路径列表 |
| MARSHOAI_LOAD_BUILTIN_TOOLS | `bool` | `true` | 是否加载内置工具包 |
| MARSHOAI_TOOLSET_DIR | `list` | `[]` | 外部工具集路径列表 |
| MARSHOAI_DISABLED_TOOLKITS | `list` | `[]` | 禁用的工具包包名列表 |
| MARSHOAI_ENABLE_RICHTEXT_PARSE | `bool` | `true` | 是否启用自动解析消息若包含图片链接则发送图片、若包含LaTeX公式则发送公式图 |
| MARSHOAI_SINGLE_LATEX_PARSE | `bool` | `false` | 单行公式是否渲染(当消息富文本解析启用时可用)(如果单行也渲……只能说不好看) |
#### 开发及调试选项
| 配置项 | 类型 | 默认值 | 说明 |
| ------------------------ | ------ | ------- | ---------------- |
| MARSHOAI_DEVMODE | `bool` | `false` | 是否启用开发者模式 |

View File

@@ -60,13 +60,8 @@ title: 安装
当 nonebot 连接到支持的 OneBot v11 实现端时,可以接收头像双击戳一戳消息并进行响应。详见`MARSHOAI_POKE_SUFFIX`配置项。
## 🛠️ ~~小棉工具~~(已弃用)
小棉工具(MarshoTools)是`v0.5.0`版本的新增功能,支持加载外部函数库来为 Marsho 提供 Function Call 功能。
## 🧩 小棉插件
小棉插件是`v1.0.0`的新增功能,替代旧的小棉工具功能。[使用文档](https://marsho.liteyuki.icu/dev/extension)
## 🛠️ 小棉工具
小棉工具(MarshoTools)是`v0.5.0`版本的新增功能,支持加载外部函数库来为 Marsho 提供 Function Call 功能。[使用文档]
## 👍 夸赞名单
@@ -100,7 +95,6 @@ title: 安装
| 配置项 | 类型 | 默认值 | 说明 |
| ------------------------ | ------ | ------- | ---------------- |
| MARSHOAI_USE_YAML_CONFIG | `bool` | `false` | 是否使用 YAML 配置文件格式 |
| MARSHOAI_DEVMODE | `bool` | `false` | 是否启用开发者模式 |
#### Marsho 使用方式
@@ -133,18 +127,9 @@ title: 安装
| MARSHOAI_ENABLE_SUPPORT_IMAGE_TIP | `bool` | `true` | 启用后用户发送带图请求时若模型不支持图片,则提示用户 |
| MARSHOAI_ENABLE_NICKNAME_TIP | `bool` | `true` | 启用后用户未设置昵称时提示用户设置 |
| MARSHOAI_ENABLE_PRAISES | `bool` | `true` | 是否启用夸赞名单功能 |
| MARSHOAI_ENABLE_TOOLS | `bool` | `false` | 是否启用小棉工具 |
| MARSHOAI_ENABLE_PLUGINS | `bool` | `true` | 是否启用小棉插件 |
| MARSHOAI_PLUGINS | `list[str]` | `[]` | 要从`sys.path`加载的插件的名称例如从pypi安装的包 |
| MARSHOAI_PLUGIN_DIRS | `list[str]` | `[]` | 插件目录路径列表 |
| MARSHOAI_ENABLE_TOOLS | `bool` | `true` | 是否启用小棉工具 |
| MARSHOAI_LOAD_BUILTIN_TOOLS | `bool` | `true` | 是否加载内置工具包 |
| MARSHOAI_TOOLSET_DIR | `list` | `[]` | 外部工具集路径列表 |
| MARSHOAI_DISABLED_TOOLKITS | `list` | `[]` | 禁用的工具包包名列表 |
| MARSHOAI_ENABLE_RICHTEXT_PARSE | `bool` | `true` | 是否启用自动解析消息若包含图片链接则发送图片、若包含LaTeX公式则发送公式图 |
| MARSHOAI_SINGLE_LATEX_PARSE | `bool` | `false` | 单行公式是否渲染(当消息富文本解析启用时可用)(如果单行也渲……只能说不好看) |
#### 开发及调试选项
| 配置项 | 类型 | 默认值 | 说明 |
| ------------------------ | ------ | ------- | ---------------- |
| MARSHOAI_DEVMODE | `bool` | `false` | 是否启用开发者模式 |
| MARSHOAI_SINGLE_LATEX_PARSE | `bool` | `false` | 单行公式是否渲染(当消息富文本解析启用时可用)(如果单行也渲……只能说不好看) |

View File

@@ -72,6 +72,14 @@ marsho_help_cmd = on_alconna(
priority=10,
block=True,
)
marsho_status_cmd = on_alconna(
Alconna(
config.marshoai_default_name + ".status",
),
priority=10,
block=True,
)
marsho_at = on_message(rule=to_me() & at_enable, priority=11)
nickname_cmd = on_alconna(
Alconna(
@@ -109,6 +117,9 @@ async def _preload_tools():
logger.info(
"如果启用小棉工具后使用的模型出现报错,请尝试将 MARSHOAI_ENABLE_TOOLS 设为 false。"
)
logger.opt(colors=True).warning(
"<y>小棉工具已被弃用,可能会在未来版本中移除。</y>"
)
@driver.on_startup
@@ -117,12 +128,13 @@ async def _():
if config.marshoai_enable_plugins:
marshoai_plugin_dirs = config.marshoai_plugin_dirs # 外部插件目录列表
"""加载内置插件"""
marshoai_plugin_dirs.insert(
0, Path(__file__).parent / "plugins"
) # 预置插件目录
for p in os.listdir(Path(__file__).parent / "plugins"):
load_plugin(f"{__package__}.plugins.{p}")
"""加载指定目录插件"""
load_plugins(*marshoai_plugin_dirs)
"""加载sys.path下的包"""
"""加载sys.path下的包, 包括从pip安装的包"""
for package_name in config.marshoai_plugins:
load_plugin(package_name)
logger.info(
@@ -226,6 +238,16 @@ async def marsho_help():
await marsho_help_cmd.finish(metadata.usage)
@marsho_status_cmd.handle()
async def marsho_status():
await marsho_status_cmd.finish(
f"当前使用的模型:{model_name}\n"
# f"当前会话数量:{len(target_list)}\n"
# f"当前上下文数量:{len(context.contexts)}"
f"当前支持图片的模型:{str(SUPPORT_IMAGE_MODELS + config.marshoai_additional_image_models)}"
)
@marsho_at.handle()
@marsho_cmd.handle()
async def marsho(
@@ -246,21 +268,28 @@ async def marsho(
if not text:
# 发送说明
# await UniMessage(metadata.usage + "\n当前使用的模型" + model_name).send()
await marsho_cmd.finish(INTRODUCTION + "\n当前使用的模型:" + model_name)
await marsho_cmd.finish(INTRODUCTION)
try:
user_id = event.get_user_id()
nicknames = await get_nicknames()
user_nickname = nicknames.get(user_id, "")
if user_nickname != "":
nickname_prompt = f"\n*此消息的说话者:{user_nickname}*"
nickname_prompt = (
f"\n*此消息的说话者id为:{user_id},名字为:{user_nickname}*"
)
else:
nickname_prompt = ""
# 用户名无法获取,暂时注释
# user_nickname = event.sender.nickname # 未设置昵称时获取用户名
# nickname_prompt = f"\n*此消息的说话者:{user_nickname}"
if config.marshoai_enforce_nickname:
await UniMessage(
"※你未设置自己的昵称。你**必须**使用「nickname [昵称]」命令设置昵称后才能进行对话。"
).send()
return
if config.marshoai_enable_nickname_tip:
await UniMessage(
"*你未设置自己的昵称。推荐使用'nickname [昵称]'命令设置昵称来获得个性化(可能)回答。"
"你未设置自己的昵称。推荐使用nickname [昵称]命令设置昵称来获得个性化(可能)回答。"
).send()
is_support_image_model = (
@@ -285,7 +314,9 @@ async def marsho(
) # type: ignore
) # type: ignore
elif config.marshoai_enable_support_image_tip:
await UniMessage("*此模型不支持图片处理。").send()
await UniMessage(
"*此模型不支持图片处理或管理员未启用此模型的图片支持。图片将被忽略。"
).send()
backup_context = await get_backup_context(target.id, target.private)
if backup_context:
context.set_context(
@@ -307,6 +338,10 @@ async def marsho(
)
# await UniMessage(str(response)).send()
choice = response.choices[0]
# Sprint(choice)
# 当tool_calls非空时将finish_reason设置为TOOL_CALLS
if choice.message.tool_calls != None:
choice["finish_reason"] = CompletionsFinishReason.TOOL_CALLS
if choice["finish_reason"] == CompletionsFinishReason.STOPPED:
# 当对话成功时将dict的上下文添加到上下文类中
context.append(
@@ -349,6 +384,9 @@ async def marsho(
function_args = json.loads(
tool_call.function.arguments.replace("'", '"')
)
# 删除args的placeholder参数
if "placeholder" in function_args:
del function_args["placeholder"]
logger.info(
f"调用函数 {tool_call.function.name.replace('-', '.')}\n参数:"
+ "\n".join([f"{k}={v}" for k, v in function_args.items()])
@@ -384,15 +422,21 @@ async def marsho(
)
func_return = f"未找到函数 {tool_call.function.name.replace('-', '.')}"
tool_msg.append(
ToolMessage(tool_call_id=tool_call.id, content=func_return) # type: ignore
ToolMessage(tool_call_id=tool_call.id, content=func_return).as_dict() # type: ignore
)
request_msg = context_msg + [UserMessage(content=usermsg).as_dict()] + tool_msg # type: ignore
response = await make_chat(
client=client,
model_name=model_name,
msg=context_msg + [UserMessage(content=usermsg)] + tool_msg, # type: ignore
tools=tools.get_tools_list(),
msg=request_msg, # type: ignore
tools=(
tools_lists if tools_lists else None
), # TODO 临时追加函数,后期优化
)
choice = response.choices[0]
# 当tool_calls非空时将finish_reason设置为TOOL_CALLS
if choice.message.tool_calls != None:
choice["finish_reason"] = CompletionsFinishReason.TOOL_CALLS
if choice["finish_reason"] == CompletionsFinishReason.STOPPED:
# 对话成功 添加上下文

View File

@@ -34,8 +34,9 @@ class ConfigModel(BaseModel):
marshoai_single_latex_parse: bool = False
marshoai_enable_nickname_tip: bool = True
marshoai_enable_support_image_tip: bool = True
marshoai_enforce_nickname: bool = True
marshoai_enable_praises: bool = True
marshoai_enable_time_prompt: bool = True
# marshoai_enable_time_prompt: bool = True
marshoai_enable_tools: bool = False
marshoai_enable_plugins: bool = True
marshoai_load_builtin_tools: bool = True

View File

@@ -32,6 +32,8 @@ marshoai_enable_nickname_tip: true # 是否启用昵称提示。
marshoai_enable_support_image_tip: true # 是否启用支持图片提示。
marshoai_enforce_nickname: true # 是否强制要求设定昵称。
marshoai_enable_praises: true # 是否启用夸赞名单功能。
marshoai_enable_tools: false # 是否启用工具支持。
@@ -46,6 +48,8 @@ marshoai_disabled_toolkits: [] # 已禁用的工具包列表。
marshoai_plugin_dirs: [] # 插件路径。
marshoai_plugins: [] # 导入的插件名可以为pip包或本地导入的使用路径。
marshoai_devmode: false # 是否启用开发者模式。
marshoai_azure_endpoint: "https://models.inference.ai.azure.com" # OpenAI 标准格式 API 的端点。

View File

@@ -30,6 +30,7 @@ INTRODUCTION: str = f"""MarshoAI-NoneBot by LiteyukiStudio
我的主页在这里哦~↓↓↓
https://marsho.liteyuki.icu
※ 使用 「{config.marshoai_default_name}.status」命令获取状态信息。
※ 使用「{config.marshoai_default_name}.help」命令获取使用说明。"""

View File

@@ -32,7 +32,7 @@ driver = get_driver()
function_call = on_alconna(
command=Alconna(
"marsho-function-call",
f"{config.marshoai_default_name}.funccall",
Subcommand(
"call",
Args["function_name", str]["kwargs", MultiVar(str), []],

View File

@@ -163,7 +163,13 @@ class Caller:
"parameters": {
"type": "object",
"properties": {
key: value.data() for key, value in self._parameters.items()
**{
key: value.data() for key, value in self._parameters.items()
},
"placeholder": {
"type": "string",
"description": "占位符,用于显示在对话框中", # 为保证兼容性而设置的无用参数
},
},
},
"required": [
@@ -233,7 +239,7 @@ class Caller:
@property
def aifc_name(self) -> str:
"""AI调用名没有点"""
return self._name.replace(".", "-")
return self.full_name.replace(".", "-")
@property
def full_name(self) -> str:

View File

@@ -107,9 +107,7 @@ def load_plugins(*plugin_dirs: str) -> set[Plugin]:
for plugin_dir in plugin_dirs:
for f in os.listdir(plugin_dir):
path = Path(os.path.join(plugin_dir, f))
module_name = None
if os.path.isfile(path) and f.endswith(".py"):
"""单文件加载"""
module_name = f"{path_to_module_name(Path(plugin_dir))}.{f[:-3]}"

View File

@@ -1,6 +1,6 @@
from nonebot_plugin_marshoai.plugin import PluginMetadata
from .chat import *
# from .chat import *
from .file_io import *
from .liteyuki import *
from .manager import *

View File

@@ -0,0 +1,65 @@
import json
from pathlib import Path
from nonebot import require
from nonebot_plugin_localstore import get_plugin_data_file # type: ignore
from nonebot_plugin_marshoai.plugin import ( # type: ignore
PluginMetadata,
on_function_call,
)
from nonebot_plugin_marshoai.plugin.func_call.params import String # type: ignore
__marsho_meta__ = PluginMetadata(
name="记忆保存",
author="MarshoAI",
description="这个插件可以帮助AI记住一些事情",
)
memory_path = get_plugin_data_file("memory.json")
if not Path(memory_path).exists():
with open(memory_path, "w", encoding="utf-8") as f:
json.dump({}, f, ensure_ascii=False, indent=4)
# print(memory_path)
@on_function_call(
description="当你发现与你对话的用户的一些信息值得你记忆,或者用户让你记忆等时,调用此函数存储记忆内容"
).params(
memory=String(description="你想记住的内容,概括并保留关键内容"),
user_id=String(description="你想记住的人的id"),
)
async def write_memory(memory: str, user_id: str):
with open(memory_path, "r", encoding="utf-8") as f:
memory_data = json.load(f)
memorys = memory_data.get(user_id, [])
memorys.append(memory)
memory_data[user_id] = memorys
with open(memory_path, "w", encoding="utf-8") as f:
json.dump(memory_data, f, ensure_ascii=False, indent=4)
return "记忆已经保存啦~"
@on_function_call(
description="当用户与你发起对话时,你需要回忆有关他的一切,因此调用此函数读取记忆内容"
).params(user_id=String(description="你想读取记忆的人的id"))
async def read_memory(user_id: str):
with open(memory_path, "r", encoding="utf-8") as f:
memory_data = json.load(f)
memorys = memory_data.get(user_id, [])
if not memorys:
return "好像对ta还没有任何记忆呢~"
return "这些是有关ta的记忆" + "\n".join(memorys)
async def organize_memories():
with open(memory_path, "r", encoding="utf-8") as f:
memory_data = json.load(f)
for i in memory_data:
...
# TODO 用大模型对记忆进行整理

View File

@@ -0,0 +1,17 @@
[
{
"type": "function",
"function": {
"name": "marshoai_memory__write_memory",
"description": "如果在上下中你看见并觉得应该记住的人的行为与事件请调用这个函数并将记忆内容写入。请尽量每次都调用总结ta的习惯、爱好和性格,以及你对ta的印象和ta对你的印象;比如用户喜欢干什么吃什么。"
}
},
{
"type": "function",
"function": {
"name": "marshoai_memory__read_memory",
"description": "每当你想要获取更多有关某人的信息的时候,请调用这个函数。"
}
}
]

View File

@@ -0,0 +1,45 @@
from nonebot_plugin_marshoai.plugin import (
Integer,
Parameter,
PluginMetadata,
String,
on_function_call,
)
from . import mk_morse_code, mk_nya_code
__marsho_meta__ = PluginMetadata(
name="MegaKits插件",
description="一个功能混杂的多文件插件",
author="Twisuki",
)
@on_function_call(description="摩尔斯电码加密").params(
msg=String(description="被加密语句")
)
async def morse_encrypt(msg: str) -> str:
"""摩尔斯电码加密"""
return str(await mk_morse_code.morse_encrypt(msg))
@on_function_call(description="摩尔斯电码解密").params(
msg=String(description="被解密语句")
)
async def morse_decrypt(msg: str) -> str:
"""摩尔斯电码解密"""
return str(await mk_morse_code.morse_decrypt(msg))
@on_function_call(description="转换为猫语").params(msg=String(description="被转换语句"))
async def nya_encrypt(msg: str) -> str:
"""转换为猫语"""
return str(await mk_nya_code.nya_encrypt(msg))
@on_function_call(description="将猫语翻译回人类语言").params(
msg=String(description="被翻译语句")
)
async def nya_decrypt(msg: str) -> str:
"""将猫语翻译回人类语言"""
return str(await mk_nya_code.nya_decrypt(msg))

View File

@@ -0,0 +1,82 @@
# MorseCode
MorseEncode = {
"A": ".-",
"B": "-...",
"C": "-.-.",
"D": "-..",
"E": ".",
"F": "..-.",
"G": "--.",
"H": "....",
"I": "..",
"J": ".---",
"K": "-.-",
"L": ".-..",
"M": "--",
"N": "-.",
"O": "---",
"P": ".--.",
"Q": "--.-",
"R": ".-.",
"S": "...",
"T": "-",
"U": "..-",
"V": "...-",
"W": ".--",
"X": "-..-",
"Y": "-.--",
"Z": "--..",
"1": ".----",
"2": "..---",
"3": "...--",
"4": "....-",
"5": ".....",
"6": "-....",
"7": "--...",
"8": "---..",
"9": "----.",
"0": "-----",
".": ".-.-.-",
":": "---...",
",": "--..--",
";": "-.-.-.",
"?": "..--..",
"=": "-...-",
"'": ".----.",
"/": "-..-.",
"!": "-.-.--",
"-": "-....-",
"_": "..--.-",
'"': ".-..-.",
"(": "-.--.",
")": "-.--.-",
"$": "...-..-",
"&": "....",
"@": ".--.-.",
" ": " ",
}
MorseDecode = {value: key for key, value in MorseEncode.items()}
async def morse_encrypt(msg: str):
result = ""
msg = msg.upper()
for char in msg:
if char in MorseEncode:
result += MorseEncode[char]
else:
result += "..--.."
result += " "
return result
async def morse_decrypt(msg: str):
result = ""
msg = msg.replace("_", "-")
msg_arr = msg.split(" ")
for element in msg_arr:
if element in MorseDecode:
result += MorseDecode[element]
else:
result += "?"
return result

View File

@@ -0,0 +1,74 @@
# NyaCode
import base64
import random
NyaCodeCharset = ["", "", "?", "~"]
NyaCodeSpecialCharset = ["", "!", "...", ".."]
NyaCodeEncode = {}
for i in range(64):
triplet = ""
for j in range(3):
index = (i // (4**j)) % 4
triplet += NyaCodeCharset[index]
if i < 26:
char = chr(65 + i) # 大写字母 A-Z
elif i < 52:
char = chr(97 + (i - 26)) # 小写字母 a-z
elif i < 62:
char = chr(48 + (i - 52)) # 数字 0-9
elif i == 62:
char = chr(43) # 特殊字符 +
else:
char = chr(47) # 特殊字符 /
NyaCodeEncode[char] = triplet
NyaCodeDecode = {value: key for key, value in NyaCodeEncode.items()}
async def nya_encrypt(msg: str):
result = ""
b64str = base64.b64encode(msg.encode()).decode().replace("=", "")
nyastr = ""
for b64char in b64str:
nyastr += NyaCodeEncode[b64char]
for char in nyastr:
if char == "" and random.random() < 0.5:
result += "!"
if random.random() < 0.25:
result += random.choice(NyaCodeSpecialCharset) + char
else:
result += char
return result
async def nya_decrypt(msg: str):
msg = msg.replace("", "").replace("!", "").replace(".", "")
nyastr = []
i = 0
if len(msg) % 3 != 0:
return "这句话不是正确的猫语"
while i < len(msg):
nyachar = msg[i : i + 3]
try:
if all(char in NyaCodeCharset for char in nyachar):
nyastr.append(nyachar)
i += 3
except Exception:
return "这句话不是正确的猫语"
b64str = ""
for nyachar in nyastr:
b64str += NyaCodeDecode[nyachar]
b64str += "=" * (4 - len(b64str) % 4)
try:
result = base64.b64decode(b64str.encode()).decode()
except Exception:
return "翻译失败"
return result

View File

@@ -0,0 +1,47 @@
from pathlib import Path
from nonebot import require
require("nonebot_plugin_localstore")
import json
from nonebot_plugin_localstore import get_data_file
memory_path = get_data_file("marshoai", "memory.json")
if not Path(memory_path).exists():
with open(memory_path, "w", encoding="utf-8") as f:
json.dump({}, f, ensure_ascii=False, indent=4)
print(memory_path)
async def write_memory(memory: str, user_id: str):
with open(memory_path, "r", encoding="utf-8") as f:
memory_data = json.load(f)
memorys = memory_data.get(user_id, [])
memorys.append(memory)
memory_data[user_id] = memorys
with open(memory_path, "w", encoding="utf-8") as f:
json.dump(memory_data, f, ensure_ascii=False, indent=4)
return "记忆已经保存啦~"
async def read_memory(user_id: str):
with open(memory_path, "r", encoding="utf-8") as f:
memory_data = json.load(f)
memorys = memory_data.get(user_id, [])
if not memorys:
return "好像对ta还没有任何记忆呢~"
return "这些是有关ta的记忆" + "\n".join(memorys)
async def organize_memories():
with open(memory_path, "r", encoding="utf-8") as f:
memory_data = json.load(f)
for i in memory_data:
...
# TODO 用大模型对记忆进行整理

View File

@@ -0,0 +1,46 @@
[
{
"type": "function",
"function": {
"name": "marshoai_memory__write_memory",
"description": "如果在上下中你看见并觉得应该记住的人的行为与事件请调用这个函数并将记忆内容写入。请尽量每次都调用总结ta的习惯、爱好和性格,以及你对ta的印象和ta对你的印象",
"parameters": {
"type": "object",
"properties": {
"memory": {
"type": "string",
"description": "你想记住的内容,概括并保留关键内容。"
},
"user_id": {
"type": "string",
"description": "你想记住的人的id。"
}
}
},
"required": [
"memory",
"user_id"
]
}
},
{
"type": "function",
"function": {
"name": "marshoai_memory__read_memory",
"description": "每当你想要获取更多有关某人的信息的时候,请调用这个函数。",
"parameters": {
"type": "object",
"properties": {
"user_id": {
"type": "string",
"description": "你想获取的人的id。"
}
}
},
"required": [
"user_id"
]
}
}
]

View File

@@ -1,2 +0,0 @@
async def write_memory(memory: str):
return ""

View File

@@ -1,21 +0,0 @@
[
{
"type": "function",
"function": {
"name": "marshoai_memory__write_memory",
"description": "当你想记住有关与你对话的人的一些信息的时候,调用此函数。",
"parameters": {
"type": "object",
"properties": {
"memory": {
"type": "string",
"description": "你想记住的内容,概括并保留关键内容。"
}
}
},
"required": [
"memory"
]
}
}
]

View File

@@ -1,15 +1,10 @@
{
"type": "module",
"devDependencies": {
"vitepress": "^1.5.0",
"vitepress-sidebar": "^1.30.2"
},
"scripts": {
"docs:dev": "vitepress dev docs --host",
"docs:build": "vitepress build docs",
"docs:preview": "vitepress preview docs"
},
"dependencies": {
"vue": "^3.5.13"
}
}
"type": "module",
"devDependencies": {"vitepress": "^1.5.0", "vitepress-sidebar": "^1.30.2"},
"scripts": {
"docs:dev": "vitepress dev docs --host",
"docs:build": "vitepress build docs",
"docs:preview": "vitepress preview docs"
},
"dependencies": {"vue": "^3.5.13"}
}

View File

@@ -9,7 +9,7 @@ authors = [
{ name="LiteyukiStudio", email = "support@liteyuki.icu"}
]
dependencies = [
"nonebot2>=2.2.0",
"nonebot2>=2.4.0",
"nonebot-plugin-alconna>=0.48.0",
"nonebot-plugin-localstore>=0.7.1",
"zhDatetime>=1.1.1",
@@ -76,3 +76,6 @@ dev = [
test = [
"nonebug>=0.4.3",
]
[tool.ruff.lint]
ignore = ["E402"]