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