From c1ff66db98aa89a937f615ecda86ed3fb87d6758 Mon Sep 17 00:00:00 2001 From: ArnoChen Date: Wed, 5 Feb 2025 01:52:33 +0800 Subject: [PATCH] fix mongo impl --- lightrag/kg/mongo_impl.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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