gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update age_impl.py

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-02-19 19:36:14 +01:00
committed by GitHub
parent 52abf9cc50
commit 5d19a888b0
1 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lightrag/kg/age_impl.py
View File

@@ -23,7 +23,6 @@ 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())
@@ -34,15 +33,9 @@ if not pm.is_installed("psycopg-pool"):
if not pm.is_installed("asyncpg"): if not pm.is_installed("asyncpg"):
pm.install("asyncpg") pm.install("asyncpg")
try: import psycopg
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
except ImportError:
raise ImportError(
"`psycopg-pool, psycopg[binary,pool], asyncpg` library is not installed. Please install it via pip: `pip install psycopg-pool psycopg[binary,pool] asyncpg`."
)
class AGEQueryException(Exception): class AGEQueryException(Exception):
"""Exception for the AGE queries.""" """Exception for the AGE queries."""