From 5dda38eeb966b8b33ba0330b18f7a8bb48a868c0 Mon Sep 17 00:00:00 2001 From: Tianhong Dai Date: Fri, 31 Jan 2025 17:13:21 +0000 Subject: [PATCH] fix import bugs of postgres example --- examples/lightrag_zhipu_postgres_demo.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/lightrag_zhipu_postgres_demo.py b/examples/lightrag_zhipu_postgres_demo.py index 4ed88602..f2066d09 100644 --- 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()