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

Update postgres_impl_test.py

Browse Source
This commit is contained in:
Saifeddine ALOUI
2025-01-27 09:38:32 +01:00
committed by GitHub
parent c7c56863b1
commit 57682389e2
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
lightrag/kg/postgres_impl_test.py
View File

@@ -1,8 +1,15 @@
import asyncio import asyncio
import asyncpg
import sys import sys
import os import os
import pipmaster as pm
if not pm.is_installed("psycopg-pool"):
pm.install("psycopg-pool")
pm.install("psycopg[binary,pool]")
if not pm.is_installed("asyncpg"):
pm.install("asyncpg")
import asyncpg
import psycopg import psycopg
from psycopg_pool import AsyncConnectionPool from psycopg_pool import AsyncConnectionPool
from lightrag.kg.postgres_impl import PostgreSQLDB, PGGraphStorage from lightrag.kg.postgres_impl import PostgreSQLDB, PGGraphStorage