Update tidb_impl.py

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)