gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
1 changed files with 6 additions and 1 deletions
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)