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