diff --git a/mem0/vector_stores/pgvector.py b/mem0/vector_stores/pgvector.py index 062b4ecd8..9906089bc 100644 --- a/mem0/vector_stores/pgvector.py +++ b/mem0/vector_stores/pgvector.py @@ -2,20 +2,20 @@ import json import logging from typing import List, Optional -from psycopg.types.json import Json from pydantic import BaseModel # Try to import psycopg (psycopg3) first, then fall back to psycopg2 try: import psycopg from psycopg import execute_values + from psycopg.types.json import Json PSYCOPG_VERSION = 3 logger = logging.getLogger(__name__) logger.info("Using psycopg (psycopg3) for PostgreSQL connections") except ImportError: try: import psycopg2 - from psycopg2.extras import execute_values + from psycopg2.extras import execute_values, Json PSYCOPG_VERSION = 2 logger = logging.getLogger(__name__) logger.info("Using psycopg2 for PostgreSQL connections") diff --git a/server/requirements.txt b/server/requirements.txt index b427c6a9f..a12efbe19 100644 --- a/server/requirements.txt +++ b/server/requirements.txt @@ -3,4 +3,4 @@ uvicorn==0.34.0 pydantic==2.10.4 mem0ai>=0.1.48 python-dotenv==1.0.1 -psycopg2-binary==2.9.10 +psycopg>=3.2.8