From b4bd7b48df72a010cf80837d81d01f4266050c42 Mon Sep 17 00:00:00 2001
From: Parshva Daftari <89991302+parshvadaftari@users.noreply.github.com>
Date: Wed, 13 Aug 2025 23:52:58 +0530
Subject: [PATCH] Fixing Json import for the psycopg and psycopg2 (#3310)

---
 mem0/vector_stores/pgvector.py | 4 ++--
 server/requirements.txt        | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

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