From 52037205ebd8dc24485480d32e794ae556f5c7e0 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Mon, 27 Jan 2025 09:40:50 +0100 Subject: [PATCH] Update tidb_impl.py --- lightrag/kg/tidb_impl.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/lightrag/kg/tidb_impl.py b/lightrag/kg/tidb_impl.py index 2cf698e1..8ba1de65 100644 --- a/lightrag/kg/tidb_impl.py +++ b/lightrag/kg/tidb_impl.py @@ -4,13 +4,18 @@ from dataclasses import dataclass from typing import Union import numpy as np +import pipmaster as pm +if not pm.is_installed("pymysql"): + pm.install("pymysql") +if not pm.is_installed("sqlalchemy"): + pm.install("sqlalchemy") + from sqlalchemy import create_engine, text from tqdm import tqdm from lightrag.base import BaseVectorStorage, BaseKVStorage, BaseGraphStorage from lightrag.utils import logger - class TiDB(object): def __init__(self, config, **kwargs): self.host = config.get("host", None)