optional chroma api key

zzstoatzz · Oct 15, 2024 · 9ca4ba1 · 9ca4ba1
1 parent 78e0d9b
commit 9ca4ba1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/raggy/settings.py b/src/raggy/settings.py
@@ -44,7 +44,8 @@ class ChromaSettings(BaseSettings):
         default="default",
         description="The database to use for the Chroma Cloud client.",
     )
-    cloud_api_key: SecretStr = Field(
+    cloud_api_key: SecretStr | None = Field(
+        default=None,
         description="The API key to use for the Chroma Cloud client.",
     )
 

diff --git a/src/raggy/vectorstores/chroma.py b/src/raggy/vectorstores/chroma.py
@@ -33,6 +33,7 @@ def get_client(client_type: ChromaClientType) -> ClientAPI:
     elif client_type == "http":
         return HttpClient()
     elif client_type == "cloud":
+        assert settings.chroma.cloud_api_key is not None
         return CloudClient(
             tenant=settings.chroma.cloud_tenant,
             database=settings.chroma.cloud_database,