pratik/knowledge-base
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

clean up and fixing

Browse source
This commit is contained in:
Pratik Narola 2025-09-02 23:02:59 +05:30
parent aa7742c5ad
commit a8a676f860
3 changed files with 4 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
config/postgres-init.sql
View file

@ -25,7 +25,7 @@ CREATE TABLE IF NOT EXISTS embeddings (
id SERIAL PRIMARY KEY, id SERIAL PRIMARY KEY,
user_id VARCHAR(255), user_id VARCHAR(255),
content TEXT, content TEXT,
embedding VECTOR(1536), -- OpenAI embedding dimension embedding VECTOR(4096), -- OpenAI embedding dimension
metadata JSONB, metadata JSONB,
created_at TIMESTAMP DEFAULT NOW() created_at TIMESTAMP DEFAULT NOW()
); );

9
docker-compose.yml
View file

@ -1,7 +1,7 @@
services: services:
# PostgreSQL with pgvector extension for vector storage # PostgreSQL with pgvector extension for vector storage
postgres: postgres:
image: pgvector/pgvector:pg15 image: pgvector/pgvector:pg17-trixie
container_name: mem0-postgres container_name: mem0-postgres
environment: environment:
POSTGRES_DB: ${POSTGRES_DB:-mem0_db} POSTGRES_DB: ${POSTGRES_DB:-mem0_db}
@ -21,7 +21,7 @@ services:
# Neo4j with APOC for graph relationships # Neo4j with APOC for graph relationships
neo4j: neo4j:
image: neo4j:5.18-community image: neo4j:5.26.4
container_name: mem0-neo4j container_name: mem0-neo4j
environment: environment:
NEO4J_AUTH: ${NEO4J_AUTH:-neo4j/mem0_neo4j_password} NEO4J_AUTH: ${NEO4J_AUTH:-neo4j/mem0_neo4j_password}
@ -68,11 +68,6 @@ services:
LOG_LEVEL: ${LOG_LEVEL:-INFO} LOG_LEVEL: ${LOG_LEVEL:-INFO}
CORS_ORIGINS: ${CORS_ORIGINS:-http://localhost:3000} CORS_ORIGINS: ${CORS_ORIGINS:-http://localhost:3000}
DEFAULT_MODEL: ${DEFAULT_MODEL:-claude-sonnet-4} 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: ports:
- "${BACKEND_PORT:-8000}:8000" - "${BACKEND_PORT:-8000}:8000"
depends_on: depends_on:

2
test_integration.py
View file

@ -520,7 +520,7 @@ def test_delete_all_user_memories(verbose):
assert "message" in data, "Delete all response missing success message" assert "message" in data, "Delete all response missing success message"
if verbose: if verbose:
print(f" Deleted all memories for {TEST_USER}") print(f"Deleted all memories for {TEST_USER}")
def test_cleanup_verification(verbose): def test_cleanup_verification(verbose):
"""Verify cleanup was successful""" """Verify cleanup was successful"""