fix: prompts

cssxsh · Jul 14, 2023 · 45a81b4 · 45a81b4
1 parent eb8149c
commit 45a81b4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/main/kotlin/xyz/cssxsh/mirai/openai/MiraiOpenAiListener.kt b/src/main/kotlin/xyz/cssxsh/mirai/openai/MiraiOpenAiListener.kt
@@ -503,12 +503,12 @@ internal object MiraiOpenAiListener : SimpleListenerHost() {
         val content = message.contentToString()
         if (content.startsWith(MiraiOpenAiConfig.prompts).not()) return
         val index = """\d+""".toRegex().find(content)?.value?.toInt() ?: 1
-        val page = MiraiOpenAiPrompts.keys().chunked(MiraiOpenAiConfig.page)
-            .getOrNull(index - 1).orEmpty()
+        val files = MiraiOpenAiPrompts.files()
+        val page = files.chunked(MiraiOpenAiConfig.page).getOrNull(index - 1).orEmpty()
 
         launch {
             subject.sendMessage(buildMessageChain {
-                appendLine("第 $page 页")
+                appendLine("共 ${files.size} 个, 第 $index 页 ")
                 appendLine()
                 page.forEach { name ->
                     appendLine(name)

diff --git a/src/main/kotlin/xyz/cssxsh/mirai/openai/data/MiraiOpenAiPrompts.kt b/src/main/kotlin/xyz/cssxsh/mirai/openai/data/MiraiOpenAiPrompts.kt
@@ -57,7 +57,7 @@ public object MiraiOpenAiPrompts : AutoSavePluginData("prompts") {
         return bind.remove(id)
     }
 
-    public fun keys(): List<String> {
+    public fun files(): List<String> {
         return folder.list { _, name -> name.endsWith(".txt") }
             ?.asList().orEmpty()
     }