diff --git a/lightrag/kg/milvus_impl.py b/lightrag/kg/milvus_impl.py index bf20ffd7..905a08b5 100644 --- a/lightrag/kg/milvus_impl.py +++ b/lightrag/kg/milvus_impl.py @@ -6,6 +6,9 @@ import numpy as np from lightrag.utils import logger from ..base import BaseVectorStorage +import pipmaster as pm +if not pm.is_installed("pymilvus"): + pm.install("pymilvus") from pymilvus import MilvusClient