fix linting errors

This commit is contained in:
zrguo
2025-01-27 23:21:34 +08:00
parent 9a7a6f9aac
commit 80451af839
13 changed files with 25 additions and 22 deletions

View File

@@ -48,7 +48,6 @@ Usage:
"""
import asyncio
import os
from dataclasses import dataclass
@@ -133,5 +132,3 @@ class JsonKVStorage(BaseKVStorage):
del self._data[id]
await self.index_done_callback()
logger.info(f"Successfully deleted {len(ids)} items from {self.namespace}")

View File

@@ -7,6 +7,7 @@ from lightrag.utils import logger
from ..base import BaseVectorStorage
import pipmaster as pm
if not pm.is_installed("pymilvus"):
pm.install("pymilvus")
from pymilvus import MilvusClient

View File

@@ -2,6 +2,7 @@ import os
from tqdm.asyncio import tqdm as tqdm_async
from dataclasses import dataclass
import pipmaster as pm
if not pm.is_installed("pymongo"):
pm.install("pymongo")

View File

@@ -47,6 +47,7 @@ Usage:
from lightrag.storage.networkx_storage import NetworkXStorage
"""
import asyncio
import os
from tqdm.asyncio import tqdm as tqdm_async

View File

@@ -4,6 +4,7 @@ import os
from dataclasses import dataclass
from typing import Any, Union, Tuple, List, Dict
import pipmaster as pm
if not pm.is_installed("neo4j"):
pm.install("neo4j")

View File

@@ -47,6 +47,7 @@ Usage:
from lightrag.storage.networkx_storage import NetworkXStorage
"""
import html
import os
from dataclasses import dataclass

View File

@@ -8,6 +8,7 @@ from typing import Union, List, Dict, Set, Any, Tuple
import numpy as np
import pipmaster as pm
if not pm.is_installed("asyncpg"):
pm.install("asyncpg")

View File

@@ -2,6 +2,7 @@ import os
from tqdm.asyncio import tqdm as tqdm_async
from dataclasses import dataclass
import pipmaster as pm
if not pm.is_installed("redis"):
pm.install("redis")

View File

@@ -5,17 +5,19 @@ 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)