From dcae408dcd4b7fb247946856aa308e48932cfdd3 Mon Sep 17 00:00:00 2001
From: chopratejas <chopratejas@gmail.com>
Date: Thu, 22 Jan 2026 23:58:54 -0800
Subject: [PATCH] Add hnswlib to dev dependencies for CI testing

- Add hnswlib>=0.8.0 to dev dependencies so CI can run HNSW tests
- Add new 'memory' extras group for users who want vector search
- Include 'memory' in 'all' extras
---
 pyproject.toml | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index f2b6752a2..03e218326 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -90,6 +90,10 @@ evals = [
     "anthropic>=0.18.0",          # Anthropic API for evals
     "openai>=1.0.0",              # OpenAI API for evals
 ]
+# Memory system extras (hierarchical memory with vector search)
+memory = [
+    "hnswlib>=0.8.0",  # HNSW vector index for semantic search
+]
 # Development dependencies
 dev = [
     "pytest>=7.0.0",
@@ -102,10 +106,11 @@ dev = [
     "anthropic>=0.18.0",
     "ollama>=0.4.0",  # For Ollama integration tests (local LLM, no API key needed)
     "langchain-ollama>=0.2.0",  # For LangChain+Ollama integration tests
+    "hnswlib>=0.8.0",  # For memory system tests
 ]
 # All optional dependencies
 all = [
-    "headroom-ai[relevance,proxy,reports,llmlingua,code,evals]",
+    "headroom-ai[relevance,proxy,reports,llmlingua,code,evals,memory]",
 ]
 
 [project.scripts]