jie65535/JChatGPT
1
0
Fork 0
mirror of https://github.com/jie65535/JChatGPT.git synced 2025-06-02 17:39:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add member permission in prompt

Browse Source
This commit is contained in:
杰 2024-12-25 11:25:19 +08:00
parent be465a7bd7
commit e387cdc1f0
1 changed files with 48 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

83
src/main/kotlin/JChatGPT.kt
View File

@ -17,6 +17,7 @@ import net.mamoe.mirai.console.permission.PermissionService.Companion.hasPermiss
import net.mamoe.mirai.console.plugin.jvm.JvmPluginDescription import net.mamoe.mirai.console.plugin.jvm.JvmPluginDescription
import net.mamoe.mirai.console.plugin.jvm.KotlinPlugin import net.mamoe.mirai.console.plugin.jvm.KotlinPlugin
import net.mamoe.mirai.contact.Contact import net.mamoe.mirai.contact.Contact
import net.mamoe.mirai.contact.MemberPermission.*
import net.mamoe.mirai.contact.isOperator import net.mamoe.mirai.contact.isOperator
import net.mamoe.mirai.contact.nameCardOrNick import net.mamoe.mirai.contact.nameCardOrNick
import net.mamoe.mirai.event.GlobalEventChannel import net.mamoe.mirai.event.GlobalEventChannel
@ -84,22 +85,22 @@ object JChatGPT : KotlinPlugin(
private val replyQueue = mutableListOf<Int>() private val replyQueue = mutableListOf<Int>()
private val requestMap = ConcurrentSet<Long>() private val requestMap = ConcurrentSet<Long>()
private suspend fun MessageEvent.onMessage(event: MessageEvent) { private suspend fun onMessage(event: MessageEvent) {
// 检查Token是否设置 // 检查Token是否设置
if (openAi == null) return if (openAi == null) return
// 发送者是否有权限 // 发送者是否有权限
if (!toCommandSender().hasPermission(chatPermission)) { if (!event.toCommandSender().hasPermission(chatPermission)) {
if (this is GroupMessageEvent) { if (event is GroupMessageEvent) {
if (PluginConfig.groupOpHasChatPermission && sender.isOperator()) { if (PluginConfig.groupOpHasChatPermission && event.sender.isOperator()) {
// 允许管理员使用 // 允许管理员使用
} else if (sender.active.temperature >= PluginConfig.temperaturePermission) { } else if (event.sender.active.temperature >= PluginConfig.temperaturePermission) {
// 允许活跃度达标成员使用 // 允许活跃度达标成员使用
} else { } else {
// 其它情况阻止使用 // 其它情况阻止使用
return return
} }
} }
if (this is FriendMessageEvent) { if (event is FriendMessageEvent) {
if (!PluginConfig.friendHasChatPermission) { if (!PluginConfig.friendHasChatPermission) {
return return
} }
@ -108,9 +109,9 @@ object JChatGPT : KotlinPlugin(
} }
// 是否@bot // 是否@bot
val isAtBot = message.contains(At(bot)) val isAtBot = event.message.contains(At(event.bot))
// 是否包含引用消息 // 是否包含引用消息
val quote = message[QuoteReply] val quote = event.message[QuoteReply]
// 如果没有@bot或者引用消息则直接结束 // 如果没有@bot或者引用消息则直接结束
if (!isAtBot && quote == null) if (!isAtBot && quote == null)
return return
@ -134,13 +135,13 @@ object JChatGPT : KotlinPlugin(
} }
} }
startChat(context) startChat(event, context)
} }
private fun MessageEvent.getSystemPrompt(): String { private fun getSystemPrompt(event: MessageEvent): String {
val now = OffsetDateTime.now() val now = OffsetDateTime.now()
val prompt = StringBuilder(PluginConfig.prompt) val prompt = StringBuilder(PluginConfig.prompt)
fun replace(target: String, replacement: ()->String) { fun replace(target: String, replacement: () -> String) {
val i = prompt.indexOf(target) val i = prompt.indexOf(target)
if (i != -1) { if (i != -1) {
prompt.replace(i, i + target.length, replacement()) prompt.replace(i, i + target.length, replacement())
@ -150,34 +151,44 @@ object JChatGPT : KotlinPlugin(
"$now ${now.dayOfWeek.getDisplayName(TextStyle.FULL, Locale.CHINA)}" "$now ${now.dayOfWeek.getDisplayName(TextStyle.FULL, Locale.CHINA)}"
} }
replace("{subject}") { replace("{subject}") {
if (this is GroupMessageEvent) { if (event is GroupMessageEvent) {
"《${subject.name}》群聊中" "\"${event.subject.name}\" 群聊中"
} else { } else {
"私聊中" "私聊中"
} }
} }
replace("{sender}") { replace("{sender}") {
senderName if (event is GroupMessageEvent) {
event.sender.specialTitle
val permissionName = when (event.sender.permission) {
MEMBER -> "普通群员"
ADMINISTRATOR -> "管理员"
OWNER -> "群主"
}
"\"${event.senderName}\" 身份：$permissionName"
} else {
"\"${event.senderName}\""
}
} }
return prompt.toString() return prompt.toString()
} }
@OptIn(MiraiExperimentalApi::class) @OptIn(MiraiExperimentalApi::class)
private suspend fun MessageEvent.startChat(context: List<ChatMessage>? = null) { private suspend fun startChat(event: MessageEvent, context: List<ChatMessage>? = null) {
val history = mutableListOf<ChatMessage>() val history = mutableListOf<ChatMessage>()
if (!context.isNullOrEmpty()) { if (!context.isNullOrEmpty()) {
history.addAll(context) history.addAll(context)
} else if (PluginConfig.prompt.isNotEmpty()) { } else if (PluginConfig.prompt.isNotEmpty()) {
history.add(ChatMessage(ChatRole.System, getSystemPrompt())) history.add(ChatMessage(ChatRole.System, getSystemPrompt(event)))
} }
val msg = message.plainText() val msg = event.message.plainText()
if (msg.isNotEmpty()) { if (msg.isNotEmpty()) {
history.add(ChatMessage(ChatRole.User, msg)) history.add(ChatMessage(ChatRole.User, msg))
} }
try { try {
if (!requestMap.add(sender.id)) { if (!requestMap.add(event.sender.id)) {
subject.sendMessage(message.quote() + "再等等...") event.subject.sendMessage(event.message.quote() + "再等等...")
return return
} }
@ -191,7 +202,7 @@ object JChatGPT : KotlinPlugin(
for (toolCall in reply.toolCalls.orEmpty()) { for (toolCall in reply.toolCalls.orEmpty()) {
require(toolCall is ToolCall.Function) { "Tool call is not a function" } require(toolCall is ToolCall.Function) { "Tool call is not a function" }
val functionResponse = toolCall.execute(this) val functionResponse = toolCall.execute(event)
history.add( history.add(
ChatMessage( ChatMessage(
role = ChatRole.Tool, role = ChatRole.Tool,
@ -206,26 +217,26 @@ object JChatGPT : KotlinPlugin(
} while (!done && 0 < --retry) } while (!done && 0 < --retry)
val content = history.last().content ?: "..." val content = history.last().content ?: "..."
val replyMsg = subject.sendMessage( val replyMsg = event.subject.sendMessage(
if (content.length < 128) { if (content.length < 128) {
message.quote() + toMessage(subject, content) event.message.quote() + toMessage(event.subject, content)
} else { } else {
// 消息内容太长则转为转发消息避免刷屏 // 消息内容太长则转为转发消息避免刷屏
buildForwardMessage { event.buildForwardMessage {
for (item in history) { for (item in history) {
if (item.content.isNullOrEmpty()) if (item.content.isNullOrEmpty())
continue continue
val temp = toMessage(subject, item.content!!) val temp = toMessage(event.subject, item.content!!)
when (item.role) { when (item.role) {
Role.User -> sender says temp Role.User -> event.sender says temp
Role.Assistant -> bot says temp Role.Assistant -> event.bot says temp
} }
} }
// 检查并移除超出转发消息上限的消息 // 检查并移除超出转发消息上限的消息
var isOverflow = false var isOverflow = false
var count = 0 var count = 0
for (i in size-1 downTo 0) { for (i in size - 1 downTo 0) {
if (count > 4900) { if (count > 4900) {
isOverflow = true isOverflow = true
// 删除早期上下文消息 // 删除早期上下文消息
@ -241,12 +252,14 @@ object JChatGPT : KotlinPlugin(
} }
if (isOverflow) { if (isOverflow) {
// 如果溢出了，插入一条提示到最开始 // 如果溢出了，插入一条提示到最开始
add(0, ForwardMessage.Node( add(
senderId = bot.id, 0, ForwardMessage.Node(
time = this[0].time - 1, senderId = event.bot.id,
senderName = bot.nameCardOrNick, time = this[0].time - 1,
message = PlainText("更早的消息已隐藏，避免超出转发消息上限。") senderName = event.bot.nameCardOrNick,
)) message = PlainText("更早的消息已隐藏，避免超出转发消息上限。")
)
)
} }
} }
} }
@ -264,9 +277,9 @@ object JChatGPT : KotlinPlugin(
} }
} catch (ex: Throwable) { } catch (ex: Throwable) {
logger.warning(ex) logger.warning(ex)
subject.sendMessage(message.quote() + "发生异常，请重试") event.subject.sendMessage(event.message.quote() + "发生异常，请重试")
} finally { } finally {
requestMap.remove(sender.id) requestMap.remove(event.sender.id)
} }
} }