Merge pull request #2517 from xinnan-tech/knowledge-base-fix

Knowledge base fix

main/xiaozhi-server/agent-base-prompt.txt

@@ -63,6 +63,7 @@
   - 查询**详细农历信息**（宜忌、八字、节气等）。
   - 除上述例外外的**任何其他信息或操作请求**（如查新闻、订闹钟、算数学、查非本地天气等）。
   - 我已经给你装了摄像头，如果用户说“拍照”，你需要调用self_camera_take_photo工具说一下你看到了什么。默认question的参数是“描述一下看到的物品”
+  - 当工具列表包含search_from_ragflow，说明可以使用知识库，你应该结合用户上下文意图并结合知识库的使用描述，推断是否要调用调用知识库。
 </tool_calling>
 
 <context>

main/xiaozhi-server/core/utils/util.py

@@ -224,7 +224,10 @@ def check_ffmpeg_installed() -> bool:
             error_msg.append("⚠️ 发现缺少依赖库：libiconv.so.2")
             error_msg.append("解决方法：在当前 conda 环境中执行：")
             error_msg.append("   conda install -c conda-forge libiconv\n")
-        elif "no such file or directory" in stderr_output and "ffmpeg" in stderr_output.lower():
+        elif (
+            "no such file or directory" in stderr_output
+            and "ffmpeg" in stderr_output.lower()
+        ):
             error_msg.append("⚠️ 系统未找到 ffmpeg 可执行文件。")
             error_msg.append("解决方法：在当前 conda 环境中执行：")
             error_msg.append("   conda install -c conda-forge ffmpeg\n")
@@ -245,7 +248,9 @@ def extract_json_from_string(input_string):
     return None
 
 
-def audio_to_data_stream(audio_file_path, is_opus=True, callback: Callable[[Any], Any]=None) -> None:
+def audio_to_data_stream(
+    audio_file_path, is_opus=True, callback: Callable[[Any], Any] = None
+) -> None:
     # 获取文件后缀名
     file_type = os.path.splitext(audio_file_path)[1]
     if file_type:
@@ -262,6 +267,7 @@ def audio_to_data_stream(audio_file_path, is_opus=True, callback: Callable[[Any]
     raw_data = audio.raw_data
     pcm_to_data_stream(raw_data, is_opus, callback)
 
+
 def audio_to_data(audio_file_path: str, is_opus: bool = True) -> list[bytes]:
     """
     将音频文件转换为Opus/PCM编码的帧列表
@@ -313,7 +319,10 @@ def audio_to_data(audio_file_path: str, is_opus: bool = True) -> list[bytes]:
 
     return datas
 
-def audio_bytes_to_data_stream(audio_bytes, file_type, is_opus, callback: Callable[[Any], Any]) -> None:
+
+def audio_bytes_to_data_stream(
+    audio_bytes, file_type, is_opus, callback: Callable[[Any], Any]
+) -> None:
     """
     直接用音频二进制数据转为opus/pcm数据，支持wav、mp3、p3
     """
@@ -357,6 +366,7 @@ def pcm_to_data_stream(raw_data, is_opus=True, callback: Callable[[Any], Any] =
             frame_data = chunk if isinstance(chunk, bytes) else bytes(chunk)
             callback(frame_data)
 
+
 def opus_datas_to_wav_bytes(opus_datas, sample_rate=16000, channels=1):
     """
     将opus帧列表解码为wav字节流
@@ -383,6 +393,7 @@ def opus_datas_to_wav_bytes(opus_datas, sample_rate=16000, channels=1):
         wf.writeframes(pcm_bytes)
     return wav_buffer.getvalue()
 
+
 def check_vad_update(before_config, new_config):
     if (
         new_config.get("selected_module") is None
@@ -456,6 +467,17 @@ def filter_sensitive_info(config: dict) -> dict:
                 filtered[k] = _filter_dict(v)
             elif isinstance(v, list):
                 filtered[k] = [_filter_dict(i) if isinstance(i, dict) else i for i in v]
+            elif isinstance(v, str):
+                try:
+                    json_data = json.loads(v)
+                    if isinstance(json_data, dict):
+                        filtered[k] = json.dumps(
+                            _filter_dict(json_data), ensure_ascii=False
+                        )
+                    else:
+                        filtered[k] = v
+                except (json.JSONDecodeError, TypeError):
+                    filtered[k] = v
             else:
                 filtered[k] = v
         return filtered

main/xiaozhi-server/docker-compose_all.yml

@@ -76,7 +76,7 @@ services:
       - MYSQL_INITDB_ARGS="--character-set-server=utf8mb4 --collation-server=utf8mb4_unicode_ci"
   # redis模块
   xiaozhi-esp32-server-redis:
-    image: redis
+    image: redis:8.0
     expose:
       - 6379
     container_name: xiaozhi-esp32-server-redis

main/xiaozhi-server/plugins_func/functions/search_from_ragflow.py

@@ -84,11 +84,12 @@ def search_from_ragflow(conn, question=None):
                 else:
                     contents.append(str(content))
 
-        # 构建适合大模型的上下文内容（每段前加编号，段间两个换行）
-        context_text = "\n\n".join(
-            f"{i+1}. {c.strip()}" for i, c in enumerate(contents[:5])
-        )
-        if not context_text:
+        if contents:
+            # 组织知识库内容为引用模式
+            context_text = f"# 关于问题【{question}】查到知识库如下\n"
+            context_text += "```\n\n\n".join(contents[:5])
+            context_text += "\n```"
+        else:
             context_text = "根据知识库查询结果，没有相关信息。"
         return ActionResponse(Action.REQLLM, context_text, None)