fix import bugs of postgres example

2025-01-31 17:13:21 +00:00
parent e59cb7493c
commit 5dda38eeb9
1 changed files with 2 additions and 1 deletions
--- a/examples/lightrag_zhipu_postgres_demo.py
+++ b/examples/lightrag_zhipu_postgres_demo.py
@@ -6,7 +6,8 @@ from dotenv import load_dotenv

 from lightrag import LightRAG, QueryParam
 from lightrag.kg.postgres_impl import PostgreSQLDB
-from lightrag.llm.zhipu import ollama_embedding, zhipu_complete
+from lightrag.llm.zhipu import zhipu_complete
+from lightrag.llm.ollama import ollama_embedding
 from lightrag.utils import EmbeddingFunc

 load_dotenv()