Fix bugs

2025-02-11 11:42:46 +08:00
parent 87e6bc5c42
commit 0c3b754108
2 changed files with 3 additions and 2 deletions
--- a/examples/lightrag_openai_demo.py
+++ b/examples/lightrag_openai_demo.py
@@ -1,7 +1,7 @@
 import os

 from lightrag import LightRAG, QueryParam
-from lightrag.llm.openai import gpt_4o_mini_complete
+from lightrag.llm.openai import gpt_4o_mini_complete, openai_embed

 WORKING_DIR = "./dickens"

@@ -10,6 +10,7 @@ if not os.path.exists(WORKING_DIR):

 rag = LightRAG(
    working_dir=WORKING_DIR,
+    embedding_func=openai_embed,
    llm_model_func=gpt_4o_mini_complete,
    # llm_model_func=gpt_4o_complete
 )