gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update tidb_impl.py

Browse source
This commit is contained in:
Saifeddine ALOUI 2025-01-27 09:40:50 +01:00 • committed by GitHub
parent b2c1144219
commit 52037205eb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
lightrag/kg/tidb_impl.py
View file

@ -4,13 +4,18 @@ from dataclasses import dataclass
from typing import Union from typing import Union
import numpy as np 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 sqlalchemy import create_engine, text
from tqdm import tqdm from tqdm import tqdm
from lightrag.base import BaseVectorStorage, BaseKVStorage, BaseGraphStorage from lightrag.base import BaseVectorStorage, BaseKVStorage, BaseGraphStorage
from lightrag.utils import logger from lightrag.utils import logger
class TiDB(object): class TiDB(object):
def __init__(self, config, **kwargs): def __init__(self, config, **kwargs):
self.host = config.get("host", None) self.host = config.get("host", None)