From 3211816da8157578de0d2c827d8f5db17375cf5f Mon Sep 17 00:00:00 2001 From: zrguo Date: Wed, 5 Feb 2025 02:48:12 +0800 Subject: [PATCH] Reapply "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