From a8a676f86061241864ca0d7313736bef5f3a5db9 Mon Sep 17 00:00:00 2001
From: Pratik Narola <pratik.narola@juspay.in>
Date: Tue, 2 Sep 2025 23:02:59 +0530
Subject: [PATCH] clean up and fixing

---
 config/postgres-init.sql | 2 +-
 docker-compose.yml       | 9 ++-------
 test_integration.py      | 2 +-
 3 files changed, 4 insertions(+), 9 deletions(-)

diff --git a/config/postgres-init.sql b/config/postgres-init.sql
index c54af97..471efc8 100644
--- a/config/postgres-init.sql
+++ b/config/postgres-init.sql
@@ -25,7 +25,7 @@ CREATE TABLE IF NOT EXISTS embeddings (
     id SERIAL PRIMARY KEY,
     user_id VARCHAR(255),
     content TEXT,
-    embedding VECTOR(1536),  -- OpenAI embedding dimension
+    embedding VECTOR(4096),  -- OpenAI embedding dimension
     metadata JSONB,
     created_at TIMESTAMP DEFAULT NOW()
 );
diff --git a/docker-compose.yml b/docker-compose.yml
index 63d55db..630f3d7 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,7 +1,7 @@
 services:
   # PostgreSQL with pgvector extension for vector storage
   postgres:
-    image: pgvector/pgvector:pg15
+    image: pgvector/pgvector:pg17-trixie
     container_name: mem0-postgres
     environment:
       POSTGRES_DB: ${POSTGRES_DB:-mem0_db}
@@ -21,7 +21,7 @@ services:
 
   # Neo4j with APOC for graph relationships
   neo4j:
-    image: neo4j:5.18-community
+    image: neo4j:5.26.4
     container_name: mem0-neo4j
     environment:
       NEO4J_AUTH: ${NEO4J_AUTH:-neo4j/mem0_neo4j_password}
@@ -68,11 +68,6 @@ services:
       LOG_LEVEL: ${LOG_LEVEL:-INFO}
       CORS_ORIGINS: ${CORS_ORIGINS:-http://localhost:3000}
       DEFAULT_MODEL: ${DEFAULT_MODEL:-claude-sonnet-4}
-      EXTRACTION_MODEL: ${EXTRACTION_MODEL:-o4-mini}
-      FAST_MODEL: ${FAST_MODEL:-o4-mini}
-      ANALYTICAL_MODEL: ${ANALYTICAL_MODEL:-gemini-2.5-pro}
-      REASONING_MODEL: ${REASONING_MODEL:-claude-sonnet-4}
-      EXPERT_MODEL: ${EXPERT_MODEL:-o3}
     ports:
       - "${BACKEND_PORT:-8000}:8000"
     depends_on:
diff --git a/test_integration.py b/test_integration.py
index a43a9b3..12c150e 100644
--- a/test_integration.py
+++ b/test_integration.py
@@ -520,7 +520,7 @@ def test_delete_all_user_memories(verbose):
     assert "message" in data, "Delete all response missing success message"
     
     if verbose:
-        print(f"   Deleted all memories for {TEST_USER}")
+        print(f"Deleted all memories for {TEST_USER}")
 
 def test_cleanup_verification(verbose):
     """Verify cleanup was successful"""