diff --git a/lightrag/kg/mongo_impl.py b/lightrag/kg/mongo_impl.py index 66331520..1c9ce50e 100644 --- a/lightrag/kg/mongo_impl.py +++ b/lightrag/kg/mongo_impl.py @@ -2,11 +2,14 @@ import os from tqdm.asyncio import tqdm as tqdm_async from dataclasses import dataclass import pipmaster as pm -import np +import numpy as np if not pm.is_installed("pymongo"): pm.install("pymongo") +if not pm.is_installed("motor"): + pm.install("motor") + from pymongo import MongoClient from motor.motor_asyncio import AsyncIOMotorClient from typing import Union, List, Tuple