fixed linting
This commit is contained in:
@@ -23,6 +23,7 @@ from ..base import BaseGraphStorage
|
||||
|
||||
if sys.platform.startswith("win"):
|
||||
import asyncio.windows_events
|
||||
|
||||
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
|
||||
|
||||
|
||||
@@ -37,6 +38,7 @@ import psycopg
|
||||
from psycopg.rows import namedtuple_row
|
||||
from psycopg_pool import AsyncConnectionPool, PoolTimeout
|
||||
|
||||
|
||||
class AGEQueryException(Exception):
|
||||
"""Exception for the AGE queries."""
|
||||
|
||||
|
@@ -28,6 +28,7 @@ from gremlin_python.driver import client, serializer
|
||||
from gremlin_python.driver.aiohttp.transport import AiohttpTransport
|
||||
from gremlin_python.driver.protocol import GremlinServerError
|
||||
|
||||
|
||||
@final
|
||||
@dataclass
|
||||
class GremlinStorage(BaseGraphStorage):
|
||||
|
@@ -20,6 +20,7 @@ if not pm.is_installed("nano-vectordb"):
|
||||
|
||||
from nano_vectordb import NanoVectorDB
|
||||
|
||||
|
||||
@final
|
||||
@dataclass
|
||||
class NanoVectorDBStorage(BaseVectorStorage):
|
||||
|
@@ -41,6 +41,7 @@ if not pm.is_installed("asyncpg"):
|
||||
import asyncpg
|
||||
from asyncpg import Pool
|
||||
|
||||
|
||||
class PostgreSQLDB:
|
||||
def __init__(self, config: dict[str, Any], **kwargs: Any):
|
||||
self.host = config.get("host", "localhost")
|
||||
|
@@ -23,7 +23,6 @@ if not pm.is_installed("sqlalchemy"):
|
||||
from sqlalchemy import create_engine, text
|
||||
|
||||
|
||||
|
||||
class TiDB:
|
||||
def __init__(self, config, **kwargs):
|
||||
self.host = config.get("host", None)
|
||||
|
Reference in New Issue
Block a user