From 0d4430898781b7df4c955e66625f41d67f169ea8 Mon Sep 17 00:00:00 2001 From: Andrii Lazarchuk Date: Mon, 21 Oct 2024 13:53:28 +0000 Subject: [PATCH] Small fix on demo --- examples/lightrag_ollama_demo.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/examples/lightrag_ollama_demo.py b/examples/lightrag_ollama_demo.py index dfda26e6..93196066 100644 --- a/examples/lightrag_ollama_demo.py +++ b/examples/lightrag_ollama_demo.py @@ -1,7 +1,7 @@ import os 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.llm import ollama_model_complete, ollama_embedding @@ -14,7 +14,6 @@ if not os.path.exists(WORKING_DIR): rag = LightRAG( working_dir=WORKING_DIR, - tiktoken_model_name="mistral:7b", llm_model_func=ollama_model_complete, llm_model_name="mistral:7b", llm_model_max_async=2,