Commit
·
cc51c55
1
Parent(s):
0feb46d
Small fix on demo
Browse files
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.
|
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,
|