Separate LaTeX parsing into independent tool for LLM calling

This commit is contained in:
2025-09-11 19:15:00 +08:00
parent ffa7f78c25
commit 099625c2f2
2 changed files with 50 additions and 25 deletions

View File

@@ -0,0 +1,46 @@
package top.jie65535.mirai.tools
import com.aallam.openai.api.chat.Tool
import com.aallam.openai.api.core.Parameters
import kotlinx.serialization.json.*
import net.mamoe.mirai.event.events.MessageEvent
import top.jie65535.mirai.LaTeXConverter
import net.mamoe.mirai.utils.ExternalResource.Companion.toExternalResource
class SendLaTeXExpression : BaseAgent(
tool = Tool.function(
name = "sendLaTeXExpression",
description = "发送LaTeX数学表达式将其渲染为图片并发送",
parameters = Parameters.buildJsonObject {
put("type", "object")
putJsonObject("properties") {
putJsonObject("expression") {
put("type", "string")
put("description", "LaTeX数学表达式")
}
}
putJsonArray("required") {
add("expression")
}
}
)
) {
override suspend fun execute(args: JsonObject?, event: MessageEvent): String {
requireNotNull(args)
val expression = args.getValue("expression").jsonPrimitive.content
try {
// 将LaTeX表达式转换为图片
val imageByteArray = LaTeXConverter.convertToImage(expression, "png")
val resource = imageByteArray.toExternalResource("png")
val image = event.subject.uploadImage(resource)
// 发送图片消息
event.subject.sendMessage(image)
return "成功发送LaTeX表达式"
} catch (ex: Throwable) {
return "处理LaTeX表达式时发生异常: ${ex.message}"
}
}
}