gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Small fix on demo

Browse source
This commit is contained in:
Andrii Lazarchuk 2024-10-21 13:53:28 +00:00
parent f7d1e43a30
commit 0d44308987
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
examples/lightrag_ollama_demo.py
View file

@ -1,7 +1,7 @@
import os import os
import logging import logging
logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.DEBUG) logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.INFO)
from lightrag import LightRAG, QueryParam from lightrag import LightRAG, QueryParam
from lightrag.llm import ollama_model_complete, ollama_embedding from lightrag.llm import ollama_model_complete, ollama_embedding
@ -14,7 +14,6 @@ if not os.path.exists(WORKING_DIR):
rag = LightRAG( rag = LightRAG(
working_dir=WORKING_DIR, working_dir=WORKING_DIR,
tiktoken_model_name="mistral:7b",
llm_model_func=ollama_model_complete, llm_model_func=ollama_model_complete,
llm_model_name="mistral:7b", llm_model_name="mistral:7b",
llm_model_max_async=2, llm_model_max_async=2,