Update README.md

add missing imports to examples in README.md
HKUDS · Nov 9, 2024 · 20a8136 · 20a8136
1 parent d86aed7
commit 20a8136
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -142,6 +142,7 @@ rag = LightRAG(
 ```python
 from lightrag.llm import hf_model_complete, hf_embedding
 from transformers import AutoModel, AutoTokenizer
+from lightrag.utils import EmbeddingFunc
 
 # Initialize LightRAG with Hugging Face model
 rag = LightRAG(
@@ -172,6 +173,7 @@ Then you only need to set LightRAG as follows:
 
 ```python
 from lightrag.llm import ollama_model_complete, ollama_embedding
+from lightrag.utils import EmbeddingFunc
 
 # Initialize LightRAG with Ollama model
 rag = LightRAG(