yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-12-08 21:37:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix init

Browse Source
This commit is contained in:
zrguo 2025-07-08 15:13:09 +08:00
parent 71cb3adb4f
commit cf26e52d89
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docs/rerank_integration.md
View File

@ -144,6 +144,7 @@ Reranking is automatically applied at these key retrieval stages:
import asyncio import asyncio
from lightrag import LightRAG, QueryParam from lightrag import LightRAG, QueryParam
from lightrag.llm.openai import gpt_4o_mini_complete, openai_embedding from lightrag.llm.openai import gpt_4o_mini_complete, openai_embedding
from lightrag.kg.shared_storage import initialize_pipeline_status
from lightrag.rerank import jina_rerank from lightrag.rerank import jina_rerank
async def my_rerank_func(query: str, documents: list, top_k: int = None, **kwargs): async def my_rerank_func(query: str, documents: list, top_k: int = None, **kwargs):
@ -167,6 +168,9 @@ async def main():
rerank_model_func=my_rerank_func, rerank_model_func=my_rerank_func,
) )
await rag.initialize_storages()
await initialize_pipeline_status()
# Insert documents # Insert documents
await rag.ainsert([ await rag.ainsert([
"Document 1 content...", "Document 1 content...",