remove rag group

Author: jonahdc

patchwork/common/utils/utils.py

@@ -181,10 +181,12 @@ def huggingface_embedding_model(
 def get_embedding_function(inputs: dict) -> "chromadb.api.types.EmbeddingFunction"["chromadb.api.types.Documents"]:
     embedding_function = next(
         (func(inputs) for input_key, func in _EMBEDDING_TO_API_KEY_NAME.items() if input_key in inputs.keys()),
-        chromadb().utils.embedding_functions.SentenceTransformerEmbeddingFunction(),
+        None,
     )
     if embedding_function is None:
-        raise ValueError(f"Missing required input data: one of {_EMBEDDING_TO_API_KEY_NAME.keys()}")
+        raise ValueError(
+            f"Must specify an embedding model. Available options: {list(_EMBEDDING_TO_API_KEY_NAME.keys())}"
+        )
 
     return embedding_function
 

pyproject.toml

@@ -73,24 +73,20 @@ owasp-depscan = { version = "5.2.15", optional = true }
 
 # rag
 chromadb = { version = "~0.4.24", optional = true }
-sentence-transformers = { version = "~2.7.0", optional = true }
 
 # rag - pinning transitive dependencies
 opentelemetry-exporter-otlp-proto-grpc = { version = "1.25.0", optional = true }
 onnxruntime = { version = "1.19.2", optional = true }
-torch = { version = "~2.5.1", optional = true }
 orjson = { version = "~3.9.15", optional = true }
 
 # notifications
 slack-sdk = { version = "~3.28.0", optional = true }
 
 [tool.poetry.extras]
 security = ["semgrep", "owasp-depscan"]
-rag = ["chromadb", "sentence-transformers", "torch", "orjson"]
 notifications = ["slack-sdk"]
 all = [
     "semgrep", "owasp-depscan",
-    "chromadb", "sentence-transformers", "torch", "orjson",
     "slack-sdk",
 ]