fix demo
This commit is contained in:
@@ -18,6 +18,7 @@
|
||||
"from lightrag import LightRAG, QueryParam\n",
|
||||
"from lightrag.llm.openai import openai_complete_if_cache, openai_embed\n",
|
||||
"from lightrag.utils import EmbeddingFunc\n",
|
||||
"from lightrag.kg.shared_storage import initialize_pipeline_status\n",
|
||||
"import nest_asyncio"
|
||||
]
|
||||
},
|
||||
@@ -25,7 +26,9 @@
|
||||
"cell_type": "markdown",
|
||||
"id": "dd17956ec322b361",
|
||||
"metadata": {},
|
||||
"source": "#### split by character"
|
||||
"source": [
|
||||
"#### split by character"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
@@ -109,14 +112,26 @@
|
||||
}
|
||||
],
|
||||
"source": [
|
||||
"rag = LightRAG(\n",
|
||||
" working_dir=WORKING_DIR,\n",
|
||||
" llm_model_func=llm_model_func,\n",
|
||||
" embedding_func=EmbeddingFunc(\n",
|
||||
" embedding_dim=4096, max_token_size=8192, func=embedding_func\n",
|
||||
" ),\n",
|
||||
" chunk_token_size=512,\n",
|
||||
")"
|
||||
"import asyncio\n",
|
||||
"import nest_asyncio\n",
|
||||
"\n",
|
||||
"nest_asyncio.apply()\n",
|
||||
"\n",
|
||||
"async def initialize_rag():\n",
|
||||
" rag = LightRAG(\n",
|
||||
" working_dir=WORKING_DIR,\n",
|
||||
" llm_model_func=llm_model_func,\n",
|
||||
" embedding_func=EmbeddingFunc(\n",
|
||||
" embedding_dim=4096, max_token_size=8192, func=embedding_func\n",
|
||||
" ),\n",
|
||||
" chunk_token_size=512,\n",
|
||||
" )\n",
|
||||
" await rag.initialize_storages()\n",
|
||||
" await initialize_pipeline_status()\n",
|
||||
"\n",
|
||||
" return rag\n",
|
||||
"\n",
|
||||
"rag = asyncio.run(initialize_rag())"
|
||||
]
|
||||
},
|
||||
{
|
||||
@@ -908,7 +923,9 @@
|
||||
"cell_type": "markdown",
|
||||
"id": "4e5bfad24cb721a8",
|
||||
"metadata": {},
|
||||
"source": "#### split by character only"
|
||||
"source": [
|
||||
"#### split by character only"
|
||||
]
|
||||
},
|
||||
{
|
||||
"cell_type": "code",
|
||||
|
Reference in New Issue
Block a user