alazarchuk commited on
Commit
cc51c55
·
1 Parent(s): 0feb46d

Small fix on demo

Browse files
Files changed (1) hide show
  1. examples/lightrag_ollama_demo.py +1 -2
examples/lightrag_ollama_demo.py CHANGED
@@ -1,7 +1,7 @@
1
  import os
2
  import logging
3
 
4
- logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.DEBUG)
5
 
6
  from lightrag import LightRAG, QueryParam
7
  from lightrag.llm import ollama_model_complete, ollama_embedding
@@ -14,7 +14,6 @@ if not os.path.exists(WORKING_DIR):
14
 
15
  rag = LightRAG(
16
  working_dir=WORKING_DIR,
17
- tiktoken_model_name="mistral:7b",
18
  llm_model_func=ollama_model_complete,
19
  llm_model_name="mistral:7b",
20
  llm_model_max_async=2,
 
1
  import os
2
  import logging
3
 
4
+ logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.INFO)
5
 
6
  from lightrag import LightRAG, QueryParam
7
  from lightrag.llm import ollama_model_complete, ollama_embedding
 
14
 
15
  rag = LightRAG(
16
  working_dir=WORKING_DIR,
 
17
  llm_model_func=ollama_model_complete,
18
  llm_model_name="mistral:7b",
19
  llm_model_max_async=2,