mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git
synced 2026-01-04 17:31:49 +00:00
Compare commits
7 Commits
snowykami-
...
v1.1.3
| Author | SHA1 | Date | |
|---|---|---|---|
| c9d2ef7885 | |||
|
|
ff6369c1a5 | ||
|
|
c00cb19e9e | ||
| e4490334fa | |||
| fce3152e17 | |||
| 9878114376 | |||
| 21b695f2d4 |
@@ -1,6 +1,6 @@
|
||||
<!--suppress LongLine -->
|
||||
<div align="center">
|
||||
<a href="https://marsho.liteyuki.icu"><img src="https://marsho.liteyuki.icu/marsho-full.svg" width="800" height="430" alt="MarshoLogo"></a>
|
||||
<a href="https://marshoai-docs.pages.liteyuki.icu"><img src="https://marshoai-docs.pages.liteyuki.icu/marsho-full.svg" width="800" height="430" alt="MarshoLogo"></a>
|
||||
<br>
|
||||
</div>
|
||||
|
||||
@@ -48,7 +48,7 @@ _谁不喜欢回复消息快又可爱的猫娘呢?_
|
||||
|
||||
## 😼 使用
|
||||
|
||||
请查看[使用文档](https://marsho.liteyuki.icu/start/use)
|
||||
请查看[使用文档](https://marshoai-docs.pages.liteyuki.icu/start/use.html)
|
||||
|
||||
## ❤ 鸣谢&版权说明
|
||||
|
||||
|
||||
@@ -1,6 +1,6 @@
|
||||
<!--suppress LongLine -->
|
||||
<div align="center">
|
||||
<a href="https://marsho.liteyuki.icu"><img src="https://marsho.liteyuki.icu/marsho-full.svg" width="800" height="430" alt="MarshoLogo"></a>
|
||||
<a href="https://marshoai-docs.pages.liteyuki.icu"><img src="https://marshoai-docs.pages.liteyuki.icu/marsho-full.svg" width="800" height="430" alt="MarshoLogo"></a>
|
||||
<br>
|
||||
</div>
|
||||
|
||||
@@ -48,7 +48,7 @@ Plugin internally installed the catgirl character of Marsho, is able to have a c
|
||||
- 🐾 Play! I like play with friends!
|
||||
|
||||
## 😼 Usage
|
||||
Please read [Documentation](https://marsho.liteyuki.icu/start/install)
|
||||
Please read [Documentation](https://marshoai-docs.pages.liteyuki.icu/start/use.html)
|
||||
|
||||
## ❤ Thanks&Copyright
|
||||
This project uses the following code from other projects:
|
||||
|
||||
@@ -65,7 +65,7 @@ When nonebot linked to OneBot v11 adapter, can recieve double click and response
|
||||
MarshoTools is a feature added in `v0.5.0`, support loading external function library to provide Function Call for Marsho.
|
||||
|
||||
## 🧩 Marsho Plugin
|
||||
Marsho Plugin is a feature added in `v1.0.0`, replacing the old MarshoTools feature. [Documentation](https://marsho.liteyuki.icu/dev/extension)
|
||||
Marsho Plugin is a feature added in `v1.0.0`, replacing the old MarshoTools feature. [Documentation](https://marshoai-docs.pages.liteyuki.icu/dev/extension)
|
||||
|
||||
## 👍 Praise list
|
||||
|
||||
|
||||
@@ -68,7 +68,7 @@ GitHub Models API 的限制较多,不建议使用,建议通过修改`MARSHOA
|
||||
|
||||
## 🧩 小棉插件
|
||||
|
||||
小棉插件是`v1.0.0`的新增功能,替代旧的小棉工具功能。[使用文档](https://marsho.liteyuki.icu/dev/extension)
|
||||
小棉插件是`v1.0.0`的新增功能,替代旧的小棉工具功能。[使用文档](https://marshoai-docs.pages.liteyuki.icu/dev/extension)
|
||||
|
||||
## 👍 夸赞名单
|
||||
|
||||
|
||||
@@ -37,7 +37,7 @@ OPENAI_NEW_MODELS: list = [
|
||||
INTRODUCTION: str = f"""MarshoAI-NoneBot by LiteyukiStudio
|
||||
你好喵~我是一只可爱的猫娘AI,名叫小棉~🐾!
|
||||
我的主页在这里哦~↓↓↓
|
||||
https://marsho.liteyuki.icu
|
||||
https://marshoai-docs.pages.liteyuki.icu
|
||||
|
||||
※ 使用 「{config.marshoai_default_name}.status」命令获取状态信息。
|
||||
※ 使用「{config.marshoai_default_name}.help」命令获取使用说明。"""
|
||||
|
||||
@@ -103,7 +103,7 @@ class MarshoHandler:
|
||||
处理单条聊天
|
||||
"""
|
||||
|
||||
context_msg = get_prompt(model_name) + (
|
||||
context_msg = await get_prompt(model_name) + (
|
||||
self.context.build(self.target.id, self.target.private)
|
||||
)
|
||||
response = await make_chat_openai(
|
||||
|
||||
@@ -288,7 +288,7 @@ with contextlib.suppress(ImportError): # 优化先不做()
|
||||
response = await make_chat_openai(
|
||||
client=client,
|
||||
model_name=model_name,
|
||||
msg=get_prompt(model_name)
|
||||
msg=await get_prompt(model_name)
|
||||
+ [
|
||||
UserMessage(
|
||||
content=f"*{user_nickname}{config.marshoai_poke_suffix}"
|
||||
|
||||
@@ -2,6 +2,7 @@ import base64
|
||||
import json
|
||||
import mimetypes
|
||||
import re
|
||||
import ssl
|
||||
import uuid
|
||||
from typing import Any, Dict, List, Optional, Union
|
||||
|
||||
@@ -58,6 +59,8 @@ _praises_init_data = {
|
||||
"""
|
||||
初始夸赞名单之数据
|
||||
"""
|
||||
_ssl_context = ssl.create_default_context()
|
||||
_ssl_context.set_ciphers("DEFAULT")
|
||||
|
||||
|
||||
async def get_image_raw_and_type(
|
||||
@@ -74,7 +77,7 @@ async def get_image_raw_and_type(
|
||||
tuple[bytes, str]: 图片二进制数据, 图片MIME格式
|
||||
"""
|
||||
|
||||
async with httpx.AsyncClient() as client:
|
||||
async with httpx.AsyncClient(verify=_ssl_context) as client:
|
||||
response = await client.get(url, headers=_browser_headers, timeout=timeout)
|
||||
if response.status_code == 200:
|
||||
# 获取图片数据
|
||||
@@ -98,9 +101,7 @@ async def get_image_b64(url: str, timeout: int = 10) -> Optional[str]:
|
||||
return: 图片base64编码
|
||||
"""
|
||||
|
||||
if data_type := await get_image_raw_and_type(
|
||||
url.replace("https://", "http://"), timeout
|
||||
):
|
||||
if data_type := await get_image_raw_and_type(url, timeout):
|
||||
# image_format = content_type.split("/")[1] if content_type else "jpeg"
|
||||
base64_image = base64.b64encode(data_type[0]).decode("utf-8")
|
||||
data_url = "data:{};base64,{}".format(data_type[1], base64_image)
|
||||
@@ -136,15 +137,15 @@ async def make_chat_openai(
|
||||
|
||||
|
||||
@from_cache("praises")
|
||||
def get_praises():
|
||||
async def get_praises():
|
||||
praises_file = store.get_plugin_data_file(
|
||||
"praises.json"
|
||||
) # 夸赞名单文件使用localstore存储
|
||||
if not praises_file.exists():
|
||||
with open(praises_file, "w", encoding="utf-8") as f:
|
||||
json.dump(_praises_init_data, f, ensure_ascii=False, indent=4)
|
||||
with open(praises_file, "r", encoding="utf-8") as f:
|
||||
data = json.load(f)
|
||||
async with aiofiles.open(praises_file, "w", encoding="utf-8") as f:
|
||||
await f.write(json.dumps(_praises_init_data, ensure_ascii=False, indent=4))
|
||||
async with aiofiles.open(praises_file, "r", encoding="utf-8") as f:
|
||||
data = json.loads(await f.read())
|
||||
praises_json = data
|
||||
return praises_json
|
||||
|
||||
@@ -160,8 +161,8 @@ async def refresh_praises_json():
|
||||
return data
|
||||
|
||||
|
||||
def build_praises() -> str:
|
||||
praises = get_praises()
|
||||
async def build_praises() -> str:
|
||||
praises = await get_praises()
|
||||
result = ["你喜欢以下几个人物,他们有各自的优点:"]
|
||||
for item in praises["like"]:
|
||||
result.append(f"名字:{item['name']},优点:{item['advantages']}")
|
||||
@@ -213,8 +214,8 @@ async def set_nickname(user_id: str, name: str):
|
||||
data[user_id] = name
|
||||
if name == "" and user_id in data:
|
||||
del data[user_id]
|
||||
with open(filename, "w", encoding="utf-8") as f:
|
||||
json.dump(data, f, ensure_ascii=False, indent=4)
|
||||
async with aiofiles.open(filename, "w", encoding="utf-8") as f:
|
||||
await f.write(json.dumps(data, ensure_ascii=False, indent=4))
|
||||
return data
|
||||
|
||||
|
||||
@@ -237,11 +238,11 @@ async def refresh_nickname_json():
|
||||
logger.error("刷新 nickname_json 表错误:无法载入 nickname.json 文件")
|
||||
|
||||
|
||||
def get_prompt(model: str) -> List[Dict[str, Any]]:
|
||||
async def get_prompt(model: str) -> List[Dict[str, Any]]:
|
||||
"""获取系统提示词"""
|
||||
prompts = config.marshoai_additional_prompt
|
||||
if config.marshoai_enable_praises:
|
||||
praises_prompt = build_praises()
|
||||
praises_prompt = await build_praises()
|
||||
prompts += praises_prompt
|
||||
|
||||
if config.marshoai_enable_time_prompt:
|
||||
|
||||
@@ -34,7 +34,7 @@ dependencies = [
|
||||
license = { text = "MIT, Mulan PSL v2" }
|
||||
|
||||
[project.urls]
|
||||
Homepage = "https://marsho.liteyuki.icu/"
|
||||
Homepage = "https://marshoai-docs.pages.liteyuki.icu/"
|
||||
|
||||
|
||||
[tool.nonebot]
|
||||
|
||||
Reference in New Issue
Block a user