diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index fbbae8c2..9515514a 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -1,6 +1,10 @@ import os from tqdm.asyncio import tqdm as tqdm_async from dataclasses import dataclass +import pipmaster as pm +if not pm.is_installed("pymongo"): + pm.install("pymongo") + from pymongo import MongoClient from typing import Union from lightrag.utils import logger