nonebot-plugin-aigf-master 0.1.2__tar.gz → 0.1.4__tar.gz

{nonebot_plugin_aigf_master-0.1.2 → nonebot_plugin_aigf_master-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nonebot-plugin-aigf-master
-Version: 0.1.2
+Version: 0.1.4
 Summary: 群聊特化LLM聊天机器人（增强版），具有记忆、表情包、跨插件调用和跨bot通信能力
 License-Expression: MIT
 Requires-Python: <4.0,>=3.10
@@ -329,12 +329,14 @@ LLM 可以：
 
 ### 搜索 API
 
-| API | 说明 | 额外依赖 |
-|-----|------|---------|
-| `tavily`（默认） | 专为 AI 设计，返回格式友好 | `tavily-python` |
-| `bocha` | 国产 AI 搜索 API，中文搜索效果好 | 无 |
-| `bing` | 微软必应搜索 | 无 |
-| `openwebsearch` | 本地 Node.js 搜索服务，聚合多引擎（bing/baidu/duckduckgo 等） | 需先启动 daemon：`cd open-webSearch && npm install && npm run serve`，配置 `AIGFM_OPENWEBSEARCH_URL` |
+| API | 说明 | 额外依赖 | 获取 Key / 部署 |
+|-----|------|---------|----------------|
+| `tavily`（默认） | 专为 AI 设计，返回格式友好 | `tavily-python` | https://app.tavily.com |
+| `bocha` | 国产 AI 搜索 API，中文搜索效果好 | 无 | https://open.bochaai.com |
+| `bing` | 微软必应搜索（Azure Bing Search API v7） | 无 | https://portal.azure.com |
+| `openwebsearch` | 本地 Node.js 搜索服务，聚合多引擎（bing/baidu/duckduckgo 等） | 需先启动 daemon：`cd open-webSearch && npm install && npm run serve`，配置 `AIGFM_OPENWEBSEARCH_URL` | 无需 Key，项目：https://github.com/Aas-ee/open-webSearch |
+
+> **Tavily** 需额外安装依赖：`pip install tavily-python`（或安装本插件时带上 extra：`pip install "nonebot-plugin-aigf-master[tavily]"`）
 
 ## 🧠 记忆系统
 

{nonebot_plugin_aigf_master-0.1.2 → nonebot_plugin_aigf_master-0.1.4}/README.md

@@ -309,12 +309,14 @@ LLM 可以：
 
 ### 搜索 API
 
-| API | 说明 | 额外依赖 |
-|-----|------|---------|
-| `tavily`（默认） | 专为 AI 设计，返回格式友好 | `tavily-python` |
-| `bocha` | 国产 AI 搜索 API，中文搜索效果好 | 无 |
-| `bing` | 微软必应搜索 | 无 |
-| `openwebsearch` | 本地 Node.js 搜索服务，聚合多引擎（bing/baidu/duckduckgo 等） | 需先启动 daemon：`cd open-webSearch && npm install && npm run serve`，配置 `AIGFM_OPENWEBSEARCH_URL` |
+| API | 说明 | 额外依赖 | 获取 Key / 部署 |
+|-----|------|---------|----------------|
+| `tavily`（默认） | 专为 AI 设计，返回格式友好 | `tavily-python` | https://app.tavily.com |
+| `bocha` | 国产 AI 搜索 API，中文搜索效果好 | 无 | https://open.bochaai.com |
+| `bing` | 微软必应搜索（Azure Bing Search API v7） | 无 | https://portal.azure.com |
+| `openwebsearch` | 本地 Node.js 搜索服务，聚合多引擎（bing/baidu/duckduckgo 等） | 需先启动 daemon：`cd open-webSearch && npm install && npm run serve`，配置 `AIGFM_OPENWEBSEARCH_URL` | 无需 Key，项目：https://github.com/Aas-ee/open-webSearch |
+
+> **Tavily** 需额外安装依赖：`pip install tavily-python`（或安装本插件时带上 extra：`pip install "nonebot-plugin-aigf-master[tavily]"`）
 
 ## 🧠 记忆系统
 

{nonebot_plugin_aigf_master-0.1.2 → nonebot_plugin_aigf_master-0.1.4}/nonebot_plugin_aigf_master/llm_client.py

@@ -52,6 +52,7 @@ class LLMClient:
     async def chat_with_tools(
         self, prompt: str, model: str,
         tools: list[dict], tool_handler,
+        json_mode: bool = False,
     ) -> tuple[str | None, bool]:
         """支持 function calling 的响应生成"""
         messages = [{"role": "user", "content": prompt}]
@@ -81,6 +82,8 @@ class LLMClient:
                 messages.append({"role": "tool", "tool_call_id": tool_call.id, "content": result})
 
             kwargs: dict = {"messages": messages, "model": model, "temperature": 0.5, "timeout": 300}
+            if json_mode:
+                kwargs["response_format"] = {"type": "json_object"}
             response = await self._client.chat.completions.create(**kwargs)
             content = response.choices[0].message.content
             return (_strip_think_tags(content) if content else None), used_tool

{nonebot_plugin_aigf_master-0.1.2 → nonebot_plugin_aigf_master-0.1.4}/nonebot_plugin_aigf_master/processor.py

@@ -373,7 +373,10 @@ class MessageProcessor:
                 return result
             return f"未知工具: {name}"
 
-        return await self.llm.chat_with_tools(prompt, self.config.aigfm_llm_model, tools, handler)
+        return await self.llm.chat_with_tools(
+            prompt, self.config.aigfm_llm_model, tools, handler,
+            json_mode=self.config.aigfm_llm_json_mode,
+        )
 
     async def _save_memes(self, save_meme: list, cached_stickers: list[dict]):
         current_ids = {s["id"] for s in cached_stickers}

{nonebot_plugin_aigf_master-0.1.2 → nonebot_plugin_aigf_master-0.1.4}/nonebot_plugin_aigf_master/prompt_builder.py

@@ -108,13 +108,29 @@ def build_prompt(
             sticker_section = """
 ## 当前消息中出现的图片（可收藏）
 以下图片已通过视觉输入提供。请结合上下文判断是否值得收藏。
+- 标记为"可能是表情包"的图片已附带情感和内容描述，是较可能的收藏候选
+- 仅标记为"图片"的通常为普通图片，一般不需要收藏
+请结合上下文进一步判断：
+- 如果是表情包（能表达一定的情感，适合在群聊中反复使用），且你的表情包库内没有类似的图片，且你觉得值得保存的，可以收藏。
+- 如果只是群友分享的照片、截图、梗图、普通图片，或者表情包库内已有类似图片，则不要收藏。
+- 若群友连续发了多张图片，通常是在分享普通图片
+- 若群友只发了一张图片，而前后均没有与这张图片相关的内容，或是在图片前后仅有对该图片的介绍或评论，通常是在分享普通图片
+- 若群友发的图片内容与之前的内容有关联，或这张图片是在其它群友发言之后发出的，且**包含较为明显的情感**（开心、愤怒、疑惑等），则可能是表情包
 如果值得收藏，在 memory.save_meme 中填入 id、简短描述和关键词。
 可用 id：""" + ", ".join(sticker_ids)
         else:
             sticker_section = """
 ## 当前消息中出现的图片（可收藏）
 消息中已标注初步判断结果。请结合上下文进一步判断是否值得收藏。
-如果值得收藏，在 memory.save_meme 中填入 id 和简短描述。
+- 标记为"可能是表情包"的图片已附带情感和内容描述，是较可能的收藏候选
+- 仅标记为"图片"的通常为普通图片，一般不需要收藏
+请结合上下文进一步判断：
+- 如果是表情包（能表达一定的情感，适合在群聊中反复使用），且你的表情包库内没有类似的图片，且你觉得值得保存的，可以收藏。
+- 如果只是群友分享的照片、截图、梗图、普通图片，或者表情包库内已有类似图片，则不要收藏。
+- 若群友连续发了多张图片，通常是在分享普通图片
+- 若群友只发了一张图片，而前后均没有与这张图片相关的内容，或是在图片前后仅有对该图片的介绍或评论，通常是在分享普通图片
+- 若群友发的图片内容与之前的内容有关联，或这张图片是在其它群友发言之后发出的，且**包含较为明显的情感**（开心、愤怒、疑惑等），则可能是表情包
+如果值得收藏，在 memory.save_meme 中填入 id 和你写的简短描述。
 可用 id：""" + ", ".join(sticker_ids)
 
     # 群友列表
@@ -129,7 +145,7 @@ def build_prompt(
 当用户要求使用以下**本机**功能，且聊天记录中**还没有**该命令的调用记录时，调用 invoke_plugin 工具来执行命令，不要自己编造回复：
 {cmd_list}
 
-调用示例：用户说"今日小猪"，且你此前未调用过，应调用 invoke_plugin(command="今日小猪")
+调用示例：用户说"帮我查一下今日小猪"，且你此前未调用过，应调用 invoke_plugin(command="今日小猪")
 
 ## 调用身份
 调用 invoke_plugin 或 invoke_peer_plugin 时，可通过 user_id 参数指定以哪位群友的身份调用（从上方"相关群友信息/已知群友昵称"中选择 QQ 号）。
@@ -140,7 +156,7 @@ def build_prompt(
 调用 invoke_plugin / invoke_peer_plugin 后，插件的响应（文本或图片描述）会以 `[插件名] 内容` 或 `[bot名] 内容` 的形式作为**新消息**出现在后续聊天记录中。
 - 调用后，本次回复必须为空（`"reply": []`），**不要说话**，等插件响应出现
 - 当聊天记录中出现 `[插件名] 内容` 或 `[bot名] 内容` 后，再根据插件响应决定是否回复、回复什么，**不要编造**插件的内容
-- **不要重复调用**：如果聊天记录中已经出现 `[小助手] 已调用命令「xxx」`，说明该命令已调用过，**绝不要再调用**，直接根据已有内容判断是否回复
+- **不要重复调用**：如果聊天记录中已经出现 `[你的名字] 已调用命令「xxx」`，说明该命令已调用过，**绝不要再调用**，直接根据已有内容判断是否回复
 - 如果插件响应一直未出现，再视情况告知用户插件未响应
 
 ## 重要：已知命令的处理
@@ -271,10 +287,15 @@ fields: info(一般信息), aliases(称呼), nickname(QQ昵称), past_nicknames(
     "friends": {{"123456": {{"add": [], "modify": [], "delete": [], "add_alias": [], "remove_alias": []}}}},
     "save_meme": [{{ "id": "cache_id", "description": "描述", "keywords": ["关键词"] }}],
     "culture": {{"add": [{{"term": "", "meaning": "", "context": ""}}], "modify": [], "delete": []}}
-  }}
+  }},
+  "command_learning": [{{"name": "命令名", "parameters": "可选参数说明(无则留空)", "usage": "命令 [参数]", "hook_type": "current或peer", "source": "current填插件名/peer填bot名", "examples": ["用户发送的原文"]}}],
+  "command_edit": [{{"name": "命令名", "parameters": "更新后参数说明", "usage": "更新后用法", "examples": ["更新后示例"]}}],
+  "command_delete": ["命令名"]
 }}
 ```
 
+> `command_learning` / `command_edit` / `command_delete` 是**可选**顶层字段：只在识别到新命令、需修正或删除已有命令时才输出，无相关操作时**整个字段省略**（`reply`、`memory` 各子项为空数组时也要保留外层结构）。
+
 ## 回复决策
 - 有人 @ 你 → 通常回复
 - 有人回复了你之前的消息 → 回复

{nonebot_plugin_aigf_master-0.1.2 → nonebot_plugin_aigf_master-0.1.4}/nonebot_plugin_aigf_master/response_parser.py

@@ -6,12 +6,61 @@ import re
 from .models import MemoryOps, ReplySegment
 
 
+def _extract_json_object(text: str) -> str | None:
+    """扫描每个 '{' 起点做花括号平衡（正确处理字符串内的 {} 与 \\" 转义），
+    返回第一个能成功解析为 JSON 的对象子串；找不到返回 None。"""
+    n = len(text)
+    i = 0
+    while i < n:
+        if text[i] != "{":
+            i += 1
+            continue
+        depth = 0
+        in_str = False
+        esc = False
+        j = i
+        while j < n:
+            ch = text[j]
+            if in_str:
+                if esc:
+                    esc = False
+                elif ch == "\\":
+                    esc = True
+                elif ch == '"':
+                    in_str = False
+            else:
+                if ch == '"':
+                    in_str = True
+                elif ch == "{":
+                    depth += 1
+                elif ch == "}":
+                    depth -= 1
+                    if depth == 0:
+                        break
+            j += 1
+        if j < n:
+            candidate = text[i:j + 1]
+            try:
+                json.loads(candidate)
+                return candidate
+            except json.JSONDecodeError:
+                pass
+        i += 1
+    return None
+
+
 def parse_llm_response(raw: str) -> dict | None:
-    """解析 LLM 返回的 JSON，去除 think 标签和代码块标记"""
+    """解析 LLM 返回的 JSON，去除 think 标签和代码块标记；容忍前导/尾随文字"""
     cleaned = re.sub(r"^```json\s*|\s*```$", "", raw.strip())
     try:
         return json.loads(cleaned)
     except json.JSONDecodeError:
+        candidate = _extract_json_object(cleaned)
+        if candidate:
+            try:
+                return json.loads(candidate)
+            except json.JSONDecodeError:
+                return None
         return None
 
 

{nonebot_plugin_aigf_master-0.1.2 → nonebot_plugin_aigf_master-0.1.4}/nonebot_plugin_aigf_master.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nonebot-plugin-aigf-master
-Version: 0.1.2
+Version: 0.1.4
 Summary: 群聊特化LLM聊天机器人（增强版），具有记忆、表情包、跨插件调用和跨bot通信能力
 License-Expression: MIT
 Requires-Python: <4.0,>=3.10
@@ -329,12 +329,14 @@ LLM 可以：
 
 ### 搜索 API
 
-| API | 说明 | 额外依赖 |
-|-----|------|---------|
-| `tavily`（默认） | 专为 AI 设计，返回格式友好 | `tavily-python` |
-| `bocha` | 国产 AI 搜索 API，中文搜索效果好 | 无 |
-| `bing` | 微软必应搜索 | 无 |
-| `openwebsearch` | 本地 Node.js 搜索服务，聚合多引擎（bing/baidu/duckduckgo 等） | 需先启动 daemon：`cd open-webSearch && npm install && npm run serve`，配置 `AIGFM_OPENWEBSEARCH_URL` |
+| API | 说明 | 额外依赖 | 获取 Key / 部署 |
+|-----|------|---------|----------------|
+| `tavily`（默认） | 专为 AI 设计，返回格式友好 | `tavily-python` | https://app.tavily.com |
+| `bocha` | 国产 AI 搜索 API，中文搜索效果好 | 无 | https://open.bochaai.com |
+| `bing` | 微软必应搜索（Azure Bing Search API v7） | 无 | https://portal.azure.com |
+| `openwebsearch` | 本地 Node.js 搜索服务，聚合多引擎（bing/baidu/duckduckgo 等） | 需先启动 daemon：`cd open-webSearch && npm install && npm run serve`，配置 `AIGFM_OPENWEBSEARCH_URL` | 无需 Key，项目：https://github.com/Aas-ee/open-webSearch |
+
+> **Tavily** 需额外安装依赖：`pip install tavily-python`（或安装本插件时带上 extra：`pip install "nonebot-plugin-aigf-master[tavily]"`）
 
 ## 🧠 记忆系统
 

{nonebot_plugin_aigf_master-0.1.2 → nonebot_plugin_aigf_master-0.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "nonebot-plugin-aigf-master"
-version = "0.1.2"
+version = "0.1.4"
 description = "群聊特化LLM聊天机器人（增强版），具有记忆、表情包、跨插件调用和跨bot通信能力"
 readme = "README.md"
 license = "MIT"