mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git
synced 2024-12-02 18:24:59 +08:00
Akarin~
GitHub
commit
8327ee5dd1
4
.gitignore
vendored
4
.gitignore
vendored
@ -1,3 +1,7 @@
|
||||
|
||||
# Other Things
|
||||
test.md
|
||||
|
||||
# Byte-compiled / optimized / DLL files
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
|
||||
3
.vscode/settings.json
vendored
Normal file
3
.vscode/settings.json
vendored
Normal file
@ -0,0 +1,3 @@
|
||||
{
|
||||
"python.analysis.typeCheckingMode": "standard"
|
||||
}
|
||||
@ -1,15 +1,19 @@
|
||||
from nonebot.plugin import require
|
||||
|
||||
require("nonebot_plugin_alconna")
|
||||
require("nonebot_plugin_localstore")
|
||||
from .azure import *
|
||||
#from .hunyuan import *
|
||||
|
||||
from nonebot import get_driver, logger
|
||||
import nonebot_plugin_localstore as store
|
||||
|
||||
# from .hunyuan import *
|
||||
from .azure import *
|
||||
from .config import config
|
||||
from .metadata import metadata
|
||||
import nonebot_plugin_localstore as store
|
||||
|
||||
__author__ = "Asankilp"
|
||||
__plugin_meta__ = metadata
|
||||
|
||||
driver = get_driver()
|
||||
|
||||
|
||||
|
||||
@ -1,5 +1,6 @@
|
||||
import contextlib
|
||||
import uuid
|
||||
import traceback
|
||||
import contextlib
|
||||
from typing import Optional
|
||||
from pathlib import Path
|
||||
|
||||
@ -15,15 +16,21 @@ from azure.ai.inference.models import (
|
||||
ChatCompletionsToolCall,
|
||||
)
|
||||
from azure.core.credentials import AzureKeyCredential
|
||||
from nonebot import on_command, on_message, logger
|
||||
from nonebot import on_command, on_message, logger, get_driver
|
||||
from nonebot.adapters import Message, Event
|
||||
from nonebot.params import CommandArg
|
||||
from nonebot.permission import SUPERUSER
|
||||
from nonebot.rule import Rule, to_me
|
||||
from nonebot_plugin_alconna import on_alconna, MsgTarget
|
||||
from nonebot_plugin_alconna.uniseg import UniMessage, UniMsg
|
||||
from nonebot_plugin_alconna import (
|
||||
on_alconna,
|
||||
MsgTarget,
|
||||
UniMessage,
|
||||
UniMsg,
|
||||
Text as TextMsg,
|
||||
Image as ImageMsg,
|
||||
)
|
||||
import nonebot_plugin_localstore as store
|
||||
from nonebot import get_driver
|
||||
|
||||
|
||||
from .constants import *
|
||||
from .metadata import metadata
|
||||
@ -37,15 +44,23 @@ async def at_enable():
|
||||
|
||||
driver = get_driver()
|
||||
|
||||
changemodel_cmd = on_command("changemodel", permission=SUPERUSER, priority=10, block=True)
|
||||
changemodel_cmd = on_command(
|
||||
"changemodel", permission=SUPERUSER, priority=10, block=True
|
||||
)
|
||||
resetmem_cmd = on_command("reset", priority=10, block=True)
|
||||
# setprompt_cmd = on_command("prompt",permission=SUPERUSER)
|
||||
praises_cmd = on_command("praises", permission=SUPERUSER, priority=10, block=True)
|
||||
add_usermsg_cmd = on_command("usermsg", permission=SUPERUSER, priority=10, block=True)
|
||||
add_assistantmsg_cmd = on_command("assistantmsg", permission=SUPERUSER, priority=10, block=True)
|
||||
add_assistantmsg_cmd = on_command(
|
||||
"assistantmsg", permission=SUPERUSER, priority=10, block=True
|
||||
)
|
||||
contexts_cmd = on_command("contexts", permission=SUPERUSER, priority=10, block=True)
|
||||
save_context_cmd = on_command("savecontext", permission=SUPERUSER, priority=10, block=True)
|
||||
load_context_cmd = on_command("loadcontext", permission=SUPERUSER, priority=10, block=True)
|
||||
save_context_cmd = on_command(
|
||||
"savecontext", permission=SUPERUSER, priority=10, block=True
|
||||
)
|
||||
load_context_cmd = on_command(
|
||||
"loadcontext", permission=SUPERUSER, priority=10, block=True
|
||||
)
|
||||
marsho_cmd = on_alconna(
|
||||
Alconna(
|
||||
config.marshoai_default_name,
|
||||
@ -53,18 +68,20 @@ marsho_cmd = on_alconna(
|
||||
),
|
||||
aliases=config.marshoai_aliases,
|
||||
priority=10,
|
||||
block=True
|
||||
block=True,
|
||||
)
|
||||
marsho_at = on_message(rule=to_me()&at_enable, priority=11)
|
||||
marsho_at = on_message(rule=to_me() & at_enable, priority=11)
|
||||
nickname_cmd = on_alconna(
|
||||
Alconna(
|
||||
"nickname",
|
||||
Args["name?", str],
|
||||
),
|
||||
priority = 10,
|
||||
block = True
|
||||
priority=10,
|
||||
block=True,
|
||||
)
|
||||
refresh_data_cmd = on_command(
|
||||
"refresh_data", permission=SUPERUSER, priority=10, block=True
|
||||
)
|
||||
refresh_data_cmd = on_command("refresh_data", permission=SUPERUSER, priority=10, block=True)
|
||||
|
||||
command_start = driver.config.command_start
|
||||
model_name = config.marshoai_default_model
|
||||
@ -86,7 +103,9 @@ async def _preload_tools():
|
||||
tools.load_tools(store.get_plugin_data_dir() / "tools")
|
||||
for tool_dir in config.marshoai_toolset_dir:
|
||||
tools.load_tools(tool_dir)
|
||||
logger.info("如果启用小棉工具后使用的模型出现报错，请尝试将 MARSHOAI_ENABLE_TOOLS 设为 false。")
|
||||
logger.info(
|
||||
"如果启用小棉工具后使用的模型出现报错，请尝试将 MARSHOAI_ENABLE_TOOLS 设为 false。"
|
||||
)
|
||||
|
||||
|
||||
@add_usermsg_cmd.handle()
|
||||
@ -132,7 +151,9 @@ async def save_context(target: MsgTarget, arg: Message = CommandArg()):
|
||||
@load_context_cmd.handle()
|
||||
async def load_context(target: MsgTarget, arg: Message = CommandArg()):
|
||||
if msg := arg.extract_plain_text():
|
||||
await get_backup_context(target.id, target.private) # 为了将当前会话添加到"已恢复过备份"的列表而添加，防止上下文被覆盖（好奇怪QwQ
|
||||
await get_backup_context(
|
||||
target.id, target.private
|
||||
) # 为了将当前会话添加到"已恢复过备份"的列表而添加，防止上下文被覆盖（好奇怪QwQ
|
||||
context.set_context(
|
||||
await load_context_from_json(msg, "contexts"), target.id, target.private
|
||||
)
|
||||
@ -178,11 +199,113 @@ async def refresh_data():
|
||||
await refresh_data_cmd.finish("已刷新数据")
|
||||
|
||||
|
||||
"""
|
||||
以下函数依照 Apache 2.0 协议授权
|
||||
|
||||
函数： get_back_uuidcodeblock、send_markdown
|
||||
|
||||
版权所有 © 2024 金羿ELS
|
||||
Copyright (R) 2024 Eilles(EillesWan@outlook.com)
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
"""
|
||||
|
||||
|
||||
async def get_back_uuidcodeblock(msg: str, code_blank_uuid_map: list[tuple[str, str]]):
|
||||
|
||||
for torep, rep in code_blank_uuid_map:
|
||||
msg = msg.replace(torep, rep)
|
||||
|
||||
return msg
|
||||
|
||||
|
||||
async def send_markdown(msg: str):
|
||||
"""
|
||||
人工智能给出的回答一般不会包含 HTML 嵌入其中，但是包含图片或者 LaTeX 公式、代码块，都很正常。
|
||||
这个函数会把这些都以图片形式嵌入消息体。
|
||||
"""
|
||||
|
||||
if not IMG_TAG_PATTERN.search(msg): # 没有图片标签
|
||||
await UniMessage(msg).send(reply_to=True)
|
||||
return
|
||||
|
||||
result_msg = UniMessage()
|
||||
code_blank_uuid_map = [
|
||||
(uuid.uuid4().hex, cbp.group()) for cbp in CODE_BLOCK_PATTERN.finditer(msg)
|
||||
]
|
||||
|
||||
last_tag_index = 0
|
||||
|
||||
# 代码块渲染麻烦，先不处理
|
||||
for rep, torep in code_blank_uuid_map:
|
||||
msg = msg.replace(torep, rep)
|
||||
|
||||
# for to_rep in CODE_SINGLE_PATTERN.finditer(msg):
|
||||
# code_blank_uuid_map.append((rep := uuid.uuid4().hex, to_rep.group()))
|
||||
# msg = msg.replace(to_rep.group(), rep)
|
||||
|
||||
print("#####################\n", msg, "\n\n")
|
||||
|
||||
# 插入图片
|
||||
for each_img_tag in IMG_TAG_PATTERN.finditer(msg):
|
||||
img_tag = await get_back_uuidcodeblock(
|
||||
each_img_tag.group(), code_blank_uuid_map
|
||||
)
|
||||
image_description = img_tag[2 : img_tag.find("]")]
|
||||
image_url = img_tag[img_tag.find("(") + 1 : -1]
|
||||
|
||||
result_msg.append(
|
||||
TextMsg(
|
||||
await get_back_uuidcodeblock(
|
||||
msg[last_tag_index : msg.find(img_tag)], code_blank_uuid_map
|
||||
)
|
||||
)
|
||||
)
|
||||
|
||||
last_tag_index = msg.find(img_tag) + len(img_tag)
|
||||
|
||||
if image_ := await get_image_raw_and_type(image_url):
|
||||
|
||||
result_msg.append(
|
||||
ImageMsg(
|
||||
raw=image_[0], mimetype=image_[1], name=image_description + ".png"
|
||||
)
|
||||
)
|
||||
result_msg.append(TextMsg("（{}）".format(image_description)))
|
||||
|
||||
else:
|
||||
result_msg.append(TextMsg(img_tag))
|
||||
|
||||
result_msg.append(
|
||||
TextMsg(await get_back_uuidcodeblock(msg[last_tag_index:], code_blank_uuid_map))
|
||||
)
|
||||
|
||||
await result_msg.send(reply_to=True)
|
||||
|
||||
|
||||
"""
|
||||
Apache 2.0 协议授权部分结束
|
||||
"""
|
||||
|
||||
|
||||
@marsho_at.handle()
|
||||
@marsho_cmd.handle()
|
||||
async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None):
|
||||
global target_list
|
||||
if event.get_message().extract_plain_text() and (not text and event.get_message().extract_plain_text() != config.marshoai_default_name):
|
||||
if event.get_message().extract_plain_text() and (
|
||||
not text
|
||||
and event.get_message().extract_plain_text() != config.marshoai_default_name
|
||||
):
|
||||
text = event.get_message()
|
||||
if not text:
|
||||
# 发送说明
|
||||
@ -204,7 +327,10 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None)
|
||||
"*你未设置自己的昵称。推荐使用'nickname [昵称]'命令设置昵称来获得个性化(可能）回答。"
|
||||
).send()
|
||||
|
||||
is_support_image_model = model_name.lower() in SUPPORT_IMAGE_MODELS + config.marshoai_additional_image_models
|
||||
is_support_image_model = (
|
||||
model_name.lower()
|
||||
in SUPPORT_IMAGE_MODELS + config.marshoai_additional_image_models
|
||||
)
|
||||
is_reasoning_model = model_name.lower() in REASONING_MODELS
|
||||
usermsg = [] if is_support_image_model else ""
|
||||
for i in text:
|
||||
@ -217,14 +343,18 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None)
|
||||
if is_support_image_model:
|
||||
usermsg.append(
|
||||
ImageContentItem(
|
||||
image_url=ImageUrl(url=str(await get_image_b64(i.data["url"])))
|
||||
image_url=ImageUrl(
|
||||
url=str(await get_image_b64(i.data["url"]))
|
||||
)
|
||||
)
|
||||
)
|
||||
elif config.marshoai_enable_support_image_tip:
|
||||
await UniMessage("*此模型不支持图片处理。").send()
|
||||
backup_context = await get_backup_context(target.id, target.private)
|
||||
if backup_context:
|
||||
context.set_context(backup_context, target.id, target.private) # 加载历史记录
|
||||
context.set_context(
|
||||
backup_context, target.id, target.private
|
||||
) # 加载历史记录
|
||||
logger.info(f"已恢复会话 {target.id} 的上下文备份~")
|
||||
context_msg = context.build(target.id, target.private)
|
||||
if not is_reasoning_model:
|
||||
@ -234,46 +364,74 @@ async def marsho(target: MsgTarget, event: Event, text: Optional[UniMsg] = None)
|
||||
client=client,
|
||||
model_name=model_name,
|
||||
msg=context_msg + [UserMessage(content=usermsg)],
|
||||
tools=tools.get_tools_list()
|
||||
tools=tools.get_tools_list(),
|
||||
)
|
||||
# await UniMessage(str(response)).send()
|
||||
choice = response.choices[0]
|
||||
if (choice["finish_reason"] == CompletionsFinishReason.STOPPED): # 当对话成功时，将dict的上下文添加到上下文类中
|
||||
if choice["finish_reason"] == CompletionsFinishReason.STOPPED:
|
||||
# 当对话成功时，将dict的上下文添加到上下文类中
|
||||
context.append(
|
||||
UserMessage(content=usermsg).as_dict(), target.id, target.private
|
||||
)
|
||||
context.append(choice.message.as_dict(), target.id, target.private)
|
||||
if [target.id, target.private] not in target_list:
|
||||
target_list.append([target.id, target.private])
|
||||
await UniMessage(str(choice.message.content)).send(reply_to=True)
|
||||
|
||||
# 对话成功发送消息
|
||||
await send_markdown(str(choice.message.content))
|
||||
elif choice["finish_reason"] == CompletionsFinishReason.CONTENT_FILTERED:
|
||||
await UniMessage("*已被内容过滤器过滤。请调整聊天内容后重试。").send(reply_to=True)
|
||||
|
||||
# 对话失败，消息过滤
|
||||
|
||||
await UniMessage("*已被内容过滤器过滤。请调整聊天内容后重试。").send(
|
||||
reply_to=True
|
||||
)
|
||||
return
|
||||
elif choice["finish_reason"] == CompletionsFinishReason.TOOL_CALLS:
|
||||
|
||||
# 需要获取额外信息，调用函数工具
|
||||
tool_msg = []
|
||||
while choice.message.tool_calls != None:
|
||||
tool_msg.append(AssistantMessage(tool_calls=response.choices[0].message.tool_calls))
|
||||
tool_msg.append(
|
||||
AssistantMessage(tool_calls=response.choices[0].message.tool_calls)
|
||||
)
|
||||
for tool_call in choice.message.tool_calls:
|
||||
if isinstance(tool_call, ChatCompletionsToolCall): # 循环调用工具直到不需要调用
|
||||
function_args = json.loads(tool_call.function.arguments.replace("'", '"'))
|
||||
logger.info(f"调用函数 {tool_call.function.name} ,参数为 {function_args}")
|
||||
await UniMessage(f"调用函数 {tool_call.function.name} ,参数为 {function_args}").send()
|
||||
func_return = await tools.call(tool_call.function.name, function_args) # 获取返回值
|
||||
tool_msg.append(ToolMessage(tool_call_id=tool_call.id, content=func_return))
|
||||
if isinstance(
|
||||
tool_call, ChatCompletionsToolCall
|
||||
): # 循环调用工具直到不需要调用
|
||||
function_args = json.loads(
|
||||
tool_call.function.arguments.replace("'", '"')
|
||||
)
|
||||
logger.info(
|
||||
f"调用函数 {tool_call.function.name} ,参数为 {function_args}"
|
||||
)
|
||||
await UniMessage(
|
||||
f"调用函数 {tool_call.function.name} ,参数为 {function_args}"
|
||||
).send()
|
||||
func_return = await tools.call(
|
||||
tool_call.function.name, function_args
|
||||
) # 获取返回值
|
||||
tool_msg.append(
|
||||
ToolMessage(tool_call_id=tool_call.id, content=func_return)
|
||||
)
|
||||
response = await make_chat(
|
||||
client=client,
|
||||
model_name=model_name,
|
||||
msg=context_msg + [UserMessage(content=usermsg)] + tool_msg,
|
||||
tools=tools.get_tools_list()
|
||||
tools=tools.get_tools_list(),
|
||||
)
|
||||
choice = response.choices[0]
|
||||
if choice["finish_reason"] == CompletionsFinishReason.STOPPED:
|
||||
|
||||
# 对话成功 添加上下文
|
||||
context.append(
|
||||
UserMessage(content=usermsg).as_dict(), target.id, target.private
|
||||
)
|
||||
# context.append(tool_msg, target.id, target.private)
|
||||
# context.append(tool_msg, target.id, target.private)
|
||||
context.append(choice.message.as_dict(), target.id, target.private)
|
||||
await UniMessage(str(choice.message.content)).send(reply_to=True)
|
||||
|
||||
# 发送消息
|
||||
await send_markdown(str(choice.message.content))
|
||||
else:
|
||||
await marsho_cmd.finish(f"意外的完成原因:{choice['finish_reason']}")
|
||||
else:
|
||||
@ -288,7 +446,6 @@ with contextlib.suppress(ImportError): # 优化先不做（）
|
||||
import nonebot.adapters.onebot.v11 # type: ignore
|
||||
from .azure_onebot import poke_notify
|
||||
|
||||
|
||||
@poke_notify.handle()
|
||||
async def poke(event: Event):
|
||||
|
||||
@ -327,5 +484,7 @@ async def auto_backup_context():
|
||||
target_uid = "private_" + target_id
|
||||
else:
|
||||
target_uid = "group_" + target_id
|
||||
await save_context_to_json(f"back_up_context_{target_uid}", contexts_data, "contexts/backup")
|
||||
await save_context_to_json(
|
||||
f"back_up_context_{target_uid}", contexts_data, "contexts/backup"
|
||||
)
|
||||
logger.info(f"已保存会话 {target_id} 的上下文备份，将在下次对话时恢复~")
|
||||
|
||||
@ -55,19 +55,19 @@ destination_file = destination_folder / "config.yaml"
|
||||
|
||||
|
||||
def copy_config(source_template, destination_file):
|
||||
'''
|
||||
"""
|
||||
复制模板配置文件到config
|
||||
'''
|
||||
"""
|
||||
shutil.copy(source_template, destination_file)
|
||||
|
||||
|
||||
def check_yaml_is_changed(source_template):
|
||||
'''
|
||||
"""
|
||||
检查配置文件是否需要更新
|
||||
'''
|
||||
with open(config_file_path, 'r', encoding="utf-8") as f:
|
||||
"""
|
||||
with open(config_file_path, "r", encoding="utf-8") as f:
|
||||
old = yaml.load(f)
|
||||
with open(source_template, 'r', encoding="utf-8") as f:
|
||||
with open(source_template, "r", encoding="utf-8") as f:
|
||||
example_ = yaml.load(f)
|
||||
keys1 = set(example_.keys())
|
||||
keys2 = set(old.keys())
|
||||
@ -78,9 +78,9 @@ def check_yaml_is_changed(source_template):
|
||||
|
||||
|
||||
def merge_configs(old_config, new_config):
|
||||
'''
|
||||
"""
|
||||
合并配置文件
|
||||
'''
|
||||
"""
|
||||
for key, value in new_config.items():
|
||||
if key in old_config:
|
||||
continue
|
||||
@ -89,6 +89,7 @@ def merge_configs(old_config, new_config):
|
||||
old_config[key] = value
|
||||
return old_config
|
||||
|
||||
|
||||
config: ConfigModel = get_plugin_config(ConfigModel)
|
||||
if config.marshoai_use_yaml_config:
|
||||
if not config_file_path.exists():
|
||||
@ -102,15 +103,15 @@ if config.marshoai_use_yaml_config:
|
||||
yaml_2 = YAML()
|
||||
logger.info("插件新的配置已更新, 正在更新")
|
||||
|
||||
with open(config_file_path, 'r', encoding="utf-8") as f:
|
||||
with open(config_file_path, "r", encoding="utf-8") as f:
|
||||
old_config = yaml_2.load(f)
|
||||
|
||||
with open(source_template, 'r', encoding="utf-8") as f:
|
||||
with open(source_template, "r", encoding="utf-8") as f:
|
||||
new_config = yaml_2.load(f)
|
||||
|
||||
merged_config = merge_configs(old_config, new_config)
|
||||
|
||||
with open(destination_file, 'w', encoding="utf-8") as f:
|
||||
with open(destination_file, "w", encoding="utf-8") as f:
|
||||
yaml_2.dump(merged_config, f)
|
||||
|
||||
with open(config_file_path, "r", encoding="utf-8") as f:
|
||||
@ -118,4 +119,6 @@ if config.marshoai_use_yaml_config:
|
||||
|
||||
config = ConfigModel(**yaml_config)
|
||||
else:
|
||||
logger.info("MarshoAI 支持新的 YAML 配置系统，若要使用，请将 MARSHOAI_USE_YAML_CONFIG 配置项设置为 true。")
|
||||
logger.info(
|
||||
"MarshoAI 支持新的 YAML 配置系统，若要使用，请将 MARSHOAI_USE_YAML_CONFIG 配置项设置为 true。"
|
||||
)
|
||||
|
||||
@ -1,4 +1,6 @@
|
||||
import re
|
||||
from .config import config
|
||||
|
||||
USAGE: str = f"""MarshoAI-NoneBot Beta by Asankilp
|
||||
用法：
|
||||
{config.marshoai_default_name} <聊天内容> : 与 Marsho 进行对话。当模型为 GPT-4o(-mini) 等时，可以带上图片进行对话。
|
||||
@ -15,9 +17,15 @@ USAGE: str = f"""MarshoAI-NoneBot Beta by Asankilp
|
||||
refresh_data : 从文件刷新已加载的昵称与夸赞名单。
|
||||
※本AI的回答"按原样"提供，不提供任何担保。AI也会犯错，请仔细甄别回答的准确性。"""
|
||||
|
||||
SUPPORT_IMAGE_MODELS: list = ["gpt-4o","gpt-4o-mini","phi-3.5-vision-instruct","llama-3.2-90b-vision-instruct","llama-3.2-11b-vision-instruct"]
|
||||
REASONING_MODELS: list = ["o1-preview","o1-mini"]
|
||||
INTRODUCTION: str = """你好喵~我是一只可爱的猫娘AI，名叫小棉~🐾！
|
||||
SUPPORT_IMAGE_MODELS: list = [
|
||||
"gpt-4o",
|
||||
"gpt-4o-mini",
|
||||
"phi-3.5-vision-instruct",
|
||||
"llama-3.2-90b-vision-instruct",
|
||||
"llama-3.2-11b-vision-instruct",
|
||||
]
|
||||
REASONING_MODELS: list = ["o1-preview", "o1-mini"]
|
||||
INTRODUCTION: str = """你好喵~我是一只可爱的猫娘AI，名叫小棉~🐾！
|
||||
我的代码在这里哦~↓↓↓
|
||||
https://github.com/LiteyukiStudio/nonebot-plugin-marshoai
|
||||
|
||||
@ -25,3 +33,19 @@ https://github.com/LiteyukiStudio/nonebot-plugin-marshoai
|
||||
https://github.com/Meloland/melobot
|
||||
我与 Melobot 酱贴贴的代码在这里喵~↓↓↓
|
||||
https://github.com/LiteyukiStudio/marshoai-melo"""
|
||||
|
||||
|
||||
# 正则匹配代码块
|
||||
CODE_BLOCK_PATTERN = re.compile(
|
||||
r"```(.*?)```|`(.*?)`", re.DOTALL
|
||||
)
|
||||
# 正则匹配完整图片标签字段
|
||||
IMG_TAG_PATTERN = re.compile(r"!\[[^\]]*\]\([^()]*\)")
|
||||
# # 正则匹配图片标签中的图片url字段
|
||||
# INTAG_URL_PATTERN = re.compile(r'\(([^)]*)')
|
||||
# # 正则匹配图片标签中的文本描述字段
|
||||
# INTAG_TEXT_PATTERN = re.compile(r'!\[([^\]]*)\]')
|
||||
# 正则匹配 LaTeX 公式内容
|
||||
LATEX_PATTERN = re.compile(
|
||||
r"\\begin\{equation\}(.*?)\\end\{equation\}|(?<!\$)(\$(.*?)\$|\$\$(.*?)\$\$|\\\[(.*?)\\\]|\\\[.*?\\\]|\\\((.*?)\\\))",
|
||||
)
|
||||
|
||||
@ -4,20 +4,19 @@ import os
|
||||
import re
|
||||
import json
|
||||
import importlib
|
||||
#import importlib.util
|
||||
|
||||
# import importlib.util
|
||||
import traceback
|
||||
from nonebot import logger
|
||||
|
||||
|
||||
class MarshoContext:
|
||||
"""
|
||||
Marsho 的上下文类
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.contents = {
|
||||
"private": {},
|
||||
"non-private": {}
|
||||
}
|
||||
self.contents = {"private": {}, "non-private": {}}
|
||||
|
||||
def _get_target_dict(self, is_private):
|
||||
return self.contents["private"] if is_private else self.contents["non-private"]
|
||||
@ -60,6 +59,7 @@ class MarshoTools:
|
||||
"""
|
||||
Marsho 的工具类
|
||||
"""
|
||||
|
||||
def __init__(self):
|
||||
self.tools_list = []
|
||||
self.imported_packages = {}
|
||||
@ -74,16 +74,20 @@ class MarshoTools:
|
||||
|
||||
for package_name in os.listdir(tools_dir):
|
||||
package_path = os.path.join(tools_dir, package_name)
|
||||
if os.path.isdir(package_path) and os.path.exists(os.path.join(package_path, '__init__.py')):
|
||||
json_path = os.path.join(package_path, 'tools.json')
|
||||
if os.path.isdir(package_path) and os.path.exists(
|
||||
os.path.join(package_path, "__init__.py")
|
||||
):
|
||||
json_path = os.path.join(package_path, "tools.json")
|
||||
if os.path.exists(json_path):
|
||||
try:
|
||||
with open(json_path, 'r', encoding="utf-8") as json_file:
|
||||
with open(json_path, "r", encoding="utf-8") as json_file:
|
||||
data = json.load(json_file)
|
||||
for i in data:
|
||||
self.tools_list.append(i)
|
||||
# 导入包
|
||||
spec = importlib.util.spec_from_file_location(package_name, os.path.join(package_path, "__init__.py"))
|
||||
spec = importlib.util.spec_from_file_location(
|
||||
package_name, os.path.join(package_path, "__init__.py")
|
||||
)
|
||||
package = importlib.util.module_from_spec(spec)
|
||||
spec.loader.exec_module(package)
|
||||
self.imported_packages[package_name] = package
|
||||
@ -94,7 +98,9 @@ class MarshoTools:
|
||||
logger.error(f"加载工具包时发生错误: {e}")
|
||||
traceback.print_exc()
|
||||
else:
|
||||
logger.warning(f"在工具包 {package_path} 下找不到tools.json，跳过加载。")
|
||||
logger.warning(
|
||||
f"在工具包 {package_path} 下找不到tools.json，跳过加载。"
|
||||
)
|
||||
else:
|
||||
logger.warning(f"{package_path} 不是有效的工具包路径，跳过加载。")
|
||||
|
||||
@ -125,5 +131,3 @@ class MarshoTools:
|
||||
if not self.tools_list or not config.marshoai_enable_tools:
|
||||
return None
|
||||
return self.tools_list
|
||||
|
||||
|
||||
|
||||
@ -1,20 +1,23 @@
|
||||
import os
|
||||
from datetime import datetime
|
||||
from zhDateTime import DateTime
|
||||
|
||||
|
||||
async def get_weather(location: str):
|
||||
return f"{location}的温度是114514℃。"
|
||||
|
||||
|
||||
async def get_current_env():
|
||||
ver = os.popen("uname -a").read()
|
||||
return str(ver)
|
||||
|
||||
|
||||
async def get_current_time():
|
||||
current_time = datetime.now().strftime("%Y.%m.%d %H:%M:%S")
|
||||
current_weekday = datetime.now().weekday()
|
||||
current_time = DateTime.now().strftime("%Y.%m.%d %H:%M:%S")
|
||||
current_weekday = DateTime.now().weekday()
|
||||
|
||||
weekdays = ["星期一", "星期二", "星期三", "星期四", "星期五", "星期六", "星期日"]
|
||||
current_weekday_name = weekdays[current_weekday]
|
||||
|
||||
current_lunar_date = (DateTime.now().to_lunar().date_hanzify()[5:])
|
||||
current_lunar_date = DateTime.now().to_lunar().date_hanzify()[5:]
|
||||
time_prompt = f"现在的时间是{current_time}，{current_weekday_name}，农历{current_lunar_date}。"
|
||||
return time_prompt
|
||||
@ -2,13 +2,12 @@ import base64
|
||||
import mimetypes
|
||||
import os
|
||||
import json
|
||||
from typing import Any
|
||||
from typing import Any, Optional
|
||||
import httpx
|
||||
import nonebot_plugin_localstore as store
|
||||
from datetime import datetime
|
||||
|
||||
from nonebot.log import logger
|
||||
from zhDateTime import DateTime # type: ignore
|
||||
from zhDateTime import DateTime
|
||||
from azure.ai.inference.aio import ChatCompletionsClient
|
||||
from azure.ai.inference.models import SystemMessage
|
||||
from .config import config
|
||||
@ -17,30 +16,66 @@ nickname_json = None # 记录昵称
|
||||
praises_json = None # 记录夸赞名单
|
||||
loaded_target_list = [] # 记录已恢复备份的上下文的列表
|
||||
|
||||
# noinspection LongLine
|
||||
chromium_headers = {
|
||||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
|
||||
}
|
||||
|
||||
async def get_image_b64(url):
|
||||
# noinspection LongLine
|
||||
headers = {
|
||||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
|
||||
}
|
||||
|
||||
async def get_image_raw_and_type(
|
||||
url: str, timeout: int = 10
|
||||
) -> Optional[tuple[bytes, str]]:
|
||||
"""
|
||||
获取图片的二进制数据
|
||||
|
||||
参数:
|
||||
url: str 图片链接
|
||||
timeout: int 超时时间 秒
|
||||
|
||||
return:
|
||||
tuple[bytes, str]: 图片二进制数据, 图片MIME格式
|
||||
|
||||
"""
|
||||
|
||||
async with httpx.AsyncClient() as client:
|
||||
response = await client.get(url, headers=headers)
|
||||
response = await client.get(url, headers=chromium_headers, timeout=timeout)
|
||||
if response.status_code == 200:
|
||||
# 获取图片数据
|
||||
image_data = response.content
|
||||
content_type = response.headers.get("Content-Type")
|
||||
if not content_type:
|
||||
content_type = mimetypes.guess_type(url)[0]
|
||||
# image_format = content_type.split("/")[1] if content_type else "jpeg"
|
||||
base64_image = base64.b64encode(image_data).decode("utf-8")
|
||||
data_url = f"data:{content_type};base64,{base64_image}"
|
||||
return data_url
|
||||
return response.content, str(content_type)
|
||||
else:
|
||||
return None
|
||||
|
||||
|
||||
async def make_chat(client: ChatCompletionsClient, msg: list, model_name: str, tools: list = None):
|
||||
async def get_image_b64(url: str, timeout: int = 10) -> Optional[str]:
|
||||
"""
|
||||
获取图片的base64编码
|
||||
|
||||
参数:
|
||||
url: 图片链接
|
||||
timeout: 超时时间 秒
|
||||
|
||||
return: 图片base64编码
|
||||
"""
|
||||
|
||||
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)
|
||||
return data_url
|
||||
else:
|
||||
return None
|
||||
|
||||
|
||||
async def make_chat(
|
||||
client: ChatCompletionsClient,
|
||||
msg: list,
|
||||
model_name: str,
|
||||
tools: Optional[list] = None,
|
||||
):
|
||||
"""调用ai获取回复
|
||||
|
||||
参数:
|
||||
@ -60,7 +95,9 @@ async def make_chat(client: ChatCompletionsClient, msg: list, model_name: str, t
|
||||
def get_praises():
|
||||
global praises_json
|
||||
if praises_json is None:
|
||||
praises_file = store.get_plugin_data_file("praises.json") # 夸赞名单文件使用localstore存储
|
||||
praises_file = store.get_plugin_data_file(
|
||||
"praises.json"
|
||||
) # 夸赞名单文件使用localstore存储
|
||||
if not os.path.exists(praises_file):
|
||||
init_data = {
|
||||
"like": [
|
||||
@ -207,5 +244,7 @@ async def get_backup_context(target_id: str, target_private: bool) -> list:
|
||||
target_uid = f"group_{target_id}"
|
||||
if target_uid not in loaded_target_list:
|
||||
loaded_target_list.append(target_uid)
|
||||
return await load_context_from_json(f"back_up_context_{target_uid}", "contexts/backup")
|
||||
return await load_context_from_json(
|
||||
f"back_up_context_{target_uid}", "contexts/backup"
|
||||
)
|
||||
return []
|
||||
|
||||
@ -3,11 +3,17 @@ import types
|
||||
from tencentcloud.common import credential
|
||||
from tencentcloud.common.profile.client_profile import ClientProfile
|
||||
from tencentcloud.common.profile.http_profile import HttpProfile
|
||||
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
|
||||
from tencentcloud.common.exception.tencent_cloud_sdk_exception import (
|
||||
TencentCloudSDKException,
|
||||
)
|
||||
from tencentcloud.hunyuan.v20230901 import hunyuan_client, models
|
||||
from .config import config
|
||||
|
||||
|
||||
def generate_image(prompt: str):
|
||||
cred = credential.Credential(config.marshoai_tencent_secretid, config.marshoai_tencent_secretkey)
|
||||
cred = credential.Credential(
|
||||
config.marshoai_tencent_secretid, config.marshoai_tencent_secretkey
|
||||
)
|
||||
# 实例化一个http选项，可选的，没有特殊需求可以跳过
|
||||
httpProfile = HttpProfile()
|
||||
httpProfile.endpoint = "hunyuan.tencentcloudapi.com"
|
||||
@ -18,11 +24,7 @@ def generate_image(prompt: str):
|
||||
client = hunyuan_client.HunyuanClient(cred, "ap-guangzhou", clientProfile)
|
||||
|
||||
req = models.TextToImageLiteRequest()
|
||||
params = {
|
||||
"Prompt": prompt,
|
||||
"RspImgType": "url",
|
||||
"Resolution": "1080:1920"
|
||||
}
|
||||
params = {"Prompt": prompt, "RspImgType": "url", "Resolution": "1080:1920"}
|
||||
req.from_json_string(json.dumps(params))
|
||||
|
||||
# 返回的resp是一个TextToImageLiteResponse的实例，与请求对象对应
|
||||
|
||||
Loading…
Reference in New Issue
Block a user