This commit is contained in:
2025-08-26 13:45:10 +08:00
parent 1a07d383ed
commit 05bd4334ea
2 changed files with 27 additions and 0 deletions

View File

@@ -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,
)
# 处理文档

View File

@@ -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"
}
}