gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 1 deletions
Show stats 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 asyncpg
import sys
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
from psycopg_pool import AsyncConnectionPool
from lightrag.kg.postgres_impl import PostgreSQLDB, PGGraphStorage