From 05bd4334eaa41679db9b45e41e887dee37d6518b Mon Sep 17 00:00:00 2001
From: HuangHai <10402852@qq.com>
Date: Tue, 26 Aug 2025 13:45:10 +0800
Subject: [PATCH] 'commit'

---
 dsRagAnything/Test.py                            | 11 +++++++++++
 .../rag_storage/kv_store_parse_cache.json        | 16 ++++++++++++++++
 2 files changed, 27 insertions(+)
 create mode 100644 dsRagAnything/rag_storage/kv_store_parse_cache.json

diff --git a/dsRagAnything/Test.py b/dsRagAnything/Test.py
index 37fbeb58..6528d247 100644
--- a/dsRagAnything/Test.py
+++ b/dsRagAnything/Test.py
@@ -1,4 +1,5 @@
 import asyncio
+
 from raganything import RAGAnything, RAGAnythingConfig
 from lightrag.llm.openai import openai_complete_if_cache, openai_embed
 from lightrag.utils import EmbeddingFunc
@@ -94,6 +95,15 @@ async def main():
             base_url=Config.Config.EMBED_BASE_URL,
         ),
     )
+    from functools import partial
+    from lightrag.rerank import cohere_rerank
+
+    rerank_model_func = partial(
+        cohere_rerank,
+        model=Config.Config.RERANK_MODEL,
+        api_key=Config.Config.RERANK_BINDING_API_KEY,
+        base_url=Config.Config.RERANK_BASE_URL,
+    )
 
     # 初始化 RAGAnything
     rag = RAGAnything(
@@ -101,6 +111,7 @@ async def main():
         llm_model_func=llm_model_func,
         vision_model_func=vision_model_func,
         embedding_func=embedding_func,
+        #rerank_model_func=rerank_model_func,
     )
 
     # 处理文档
diff --git a/dsRagAnything/rag_storage/kv_store_parse_cache.json b/dsRagAnything/rag_storage/kv_store_parse_cache.json
new file mode 100644
index 00000000..6c80149d
--- /dev/null
+++ b/dsRagAnything/rag_storage/kv_store_parse_cache.json
@@ -0,0 +1,16 @@
+{
+  "cf4c4c9f961df2a5953e69e074752aaf": {
+    "content_list": [],
+    "doc_id": "doc-d41d8cd98f00b204e9800998ecf8427e",
+    "mtime": 1756180574.9404216,
+    "parse_config": {
+      "parser": "mineru",
+      "parse_method": "auto"
+    },
+    "cached_at": 1756187088.1039026,
+    "cache_version": "1.0",
+    "create_time": 1756187088,
+    "update_time": 1756187088,
+    "_id": "cf4c4c9f961df2a5953e69e074752aaf"
+  }
+}
\ No newline at end of file