From 57682389e2cded5a837c6ce700ed28c2c4792298 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Mon, 27 Jan 2025 09:38:32 +0100 Subject: [PATCH] Update postgres_impl_test.py --- lightrag/kg/postgres_impl_test.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/lightrag/kg/postgres_impl_test.py b/lightrag/kg/postgres_impl_test.py index 274f03de..eb6e6e73 100644 --- 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