From a445e556d8ea11e3edd305f78d98cdf709f71a21 Mon Sep 17 00:00:00 2001
From: Haotian Zhang <htplex@gmail.com>
Date: Sat, 9 Nov 2024 14:59:41 -0500
Subject: [PATCH] Update README.md

add missing imports to examples in README.md
---
 README.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/README.md b/README.md
index b726f605..7d798b17 100644
--- 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(