CODITECT MEMORY-CONTEXT ChromaDB Setup
Configures ChromaDB for vector storage and semantic search. Creates collections for sessions and patterns with embeddings.
Usage: python3 scripts/core/chromadb_setup.py [--reset] [--verbose]
Options: --reset Delete existing collections before creating --verbose Show detailed progress
Dependencies: pip install chromadb sentence-transformers
Author: AZ1.AI CODITECT Team Sprint: Sprint +1 - MEMORY-CONTEXT Implementation Day 3 Date: 2025-11-16
File: chromadb_setup.py
Classes
ChromaDBSetup
Setup ChromaDB for CODITECT MEMORY-CONTEXT.
Functions
main()
Main entry point.
get_client()
Create ChromaDB client.
create_sessions_collection(client, reset)
Create collection for session embeddings.
create_patterns_collection(client, reset)
Create collection for pattern embeddings.
add_sample_embeddings(sessions_collection, patterns_collection)
Add sample embeddings for testing.
test_similarity_search(sessions_collection, patterns_collection)
Test similarity search functionality.
get_collection_stats(client)
Get statistics for all collections.
setup(reset, add_samples)
Setup ChromaDB collections.
Usage
python chromadb_setup.py