JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixing Json import for the psycopg and psycopg2 (#3310)

Browse Source
This commit is contained in:
Parshva Daftari
2025-08-13 23:52:58 +05:30
committed by GitHub
parent 1bc31b6ae3
commit b4bd7b48df
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/vector_stores/pgvector.py
+2 -2
View File
@@ -2,20 +2,20 @@ import json
import logging import logging
from typing import List, Optional from typing import List, Optional
from psycopg.types.json import Json
from pydantic import BaseModel from pydantic import BaseModel
# Try to import psycopg (psycopg3) first, then fall back to psycopg2 # Try to import psycopg (psycopg3) first, then fall back to psycopg2
try: try:
import psycopg import psycopg
from psycopg import execute_values from psycopg import execute_values
from psycopg.types.json import Json
PSYCOPG_VERSION = 3 PSYCOPG_VERSION = 3
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
logger.info("Using psycopg (psycopg3) for PostgreSQL connections") logger.info("Using psycopg (psycopg3) for PostgreSQL connections")
except ImportError: except ImportError:
try: try:
import psycopg2 import psycopg2
from psycopg2.extras import execute_values from psycopg2.extras import execute_values, Json
PSYCOPG_VERSION = 2 PSYCOPG_VERSION = 2
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
logger.info("Using psycopg2 for PostgreSQL connections") logger.info("Using psycopg2 for PostgreSQL connections")
server/requirements.txt
+1 -1
View File
@@ -3,4 +3,4 @@ uvicorn==0.34.0
pydantic==2.10.4 pydantic==2.10.4
mem0ai>=0.1.48 mem0ai>=0.1.48
python-dotenv==1.0.1 python-dotenv==1.0.1
psycopg2-binary==2.9.10 psycopg>=3.2.8