feat: Making llama-embeder optional dependency (#34)

Closes #32
amikos-tech · Sep 6, 2024 · 9f9b6d5 · 9f9b6d5
1 parent 0bf7d96
commit 9f9b6d5
Show file tree

Hide file tree

Showing 5 changed files with 127 additions and 149 deletions.
diff --git a/.github/workflows/test.yaml b/.github/workflows/test.yaml
@@ -43,5 +43,5 @@ jobs:
       - name: Run tests
         run: |
           set -e
-          poetry update
+          poetry update --with dev
           poetry run pytest
diff --git a/CODEOWNERS b/CODEOWNERS
@@ -1 +1 @@
-*       @tazarov
+*       @tazarov
diff --git a/chromadbx/embeddings/llamacpp.py b/chromadbx/embeddings/llamacpp.py
@@ -10,7 +10,8 @@ def __init__(self, model_path: str) -> None:
             from llama_embedder import LlamaEmbedder, PoolingType
         except ImportError:
             raise ValueError(
-                "The cohere python package is not installed. Please install it with `pip install llama-embedder`"
+                "The `llama-embedder` python package is not installed. "
+                "Please install it with `pip install llama-embedder`"
             )
 
         if not os.path.exists(model_path):