From af245eb73eb88dab3ed6c52af575fe285e21709c Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Mon, 27 Jan 2025 09:35:08 +0100 Subject: [PATCH] Update age_impl.py --- lightrag/kg/age_impl.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/lightrag/kg/age_impl.py b/lightrag/kg/age_impl.py index 275f5775..df32b7cb 100644 --- a/lightrag/kg/age_impl.py +++ b/lightrag/kg/age_impl.py @@ -6,6 +6,14 @@ import sys from contextlib import asynccontextmanager from dataclasses import dataclass from typing import Any, Dict, List, NamedTuple, Optional, Tuple, Union +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 psycopg from psycopg.rows import namedtuple_row