From 08f34806d614ffc5ea62ce9d12fa7ebbbc93873f Mon Sep 17 00:00:00 2001
From: KoalaTrapLord <265133236+KoalaTrapLord@users.noreply.github.com>
Date: Fri, 27 Mar 2026 12:20:35 -0400
Subject: [PATCH] fix(deps): add rag and coach-embeddings optional-dependencies
 to pyproject.toml (closes #129)

---
 pyproject.toml | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/pyproject.toml b/pyproject.toml
index 4daf5aa..9b0166d 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -25,6 +25,28 @@ dev = [
     "httpx>=0.24.0",
     "ruff>=0.1.0",
 ]
+# RAG Phase 2: ChromaDB vector store + Ollama embedding support
+# Install with:  pip install ".[rag]"
+rag = [
+    "chromadb>=0.5.0",
+]
+# Local coach embeddings (sentence-transformers, no Ollama required)
+# Install with:  pip install ".[coach-embeddings]"
+coach-embeddings = [
+    "sentence-transformers>=3.0.0",
+    "torch>=2.0.0",
+    "pandas>=1.5.0",
+    "pyarrow>=12.0.0",
+]
+# Full local AI stack (RAG + coach embeddings)
+# Install with:  pip install ".[all]"
+all = [
+    "chromadb>=0.5.0",
+    "sentence-transformers>=3.0.0",
+    "torch>=2.0.0",
+    "pandas>=1.5.0",
+    "pyarrow>=12.0.0",
+]
 
 [tool.setuptools.packages.find]
 where = ["src"]