From 45ee4dd08cba90523e32863b19ca446c02875615 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Wed, 19 Feb 2025 20:50:39 +0100 Subject: [PATCH] fixed linting --- lightrag/kg/age_impl.py | 2 ++ lightrag/kg/gremlin_impl.py | 1 + lightrag/kg/nano_vector_db_impl.py | 1 + lightrag/kg/postgres_impl.py | 1 + lightrag/kg/tidb_impl.py | 1 - 5 files changed, 5 insertions(+), 1 deletion(-) diff --git a/lightrag/kg/age_impl.py b/lightrag/kg/age_impl.py index cb28ab6f..97b3825d 100644 --- a/lightrag/kg/age_impl.py +++ b/lightrag/kg/age_impl.py @@ -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.""" diff --git a/lightrag/kg/gremlin_impl.py b/lightrag/kg/gremlin_impl.py index 5d1ea47b..3a26401d 100644 --- a/lightrag/kg/gremlin_impl.py +++ b/lightrag/kg/gremlin_impl.py @@ -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): diff --git a/lightrag/kg/nano_vector_db_impl.py b/lightrag/kg/nano_vector_db_impl.py index 2c29632b..315b5a8f 100644 --- a/lightrag/kg/nano_vector_db_impl.py +++ b/lightrag/kg/nano_vector_db_impl.py @@ -20,6 +20,7 @@ if not pm.is_installed("nano-vectordb"): from nano_vectordb import NanoVectorDB + @final @dataclass class NanoVectorDBStorage(BaseVectorStorage): diff --git a/lightrag/kg/postgres_impl.py b/lightrag/kg/postgres_impl.py index 874c6f22..4ffa2fb2 100644 --- a/lightrag/kg/postgres_impl.py +++ b/lightrag/kg/postgres_impl.py @@ -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") diff --git a/lightrag/kg/tidb_impl.py b/lightrag/kg/tidb_impl.py index 7afff265..8b8fa2b6 100644 --- a/lightrag/kg/tidb_impl.py +++ b/lightrag/kg/tidb_impl.py @@ -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)