Update postgres_impl_test.py

2025-01-27 09:38:32 +01:00
parent c7c56863b1
commit 57682389e2
1 changed files with 8 additions and 1 deletions
--- a/lightrag/kg/postgres_impl_test.py
+++ b/lightrag/kg/postgres_impl_test.py
@@ -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