More work towards making remote stacks usable from tests

2025-10-19 07:37:18 +00:00 · 2024-11-12 17:09:31 -08:00 · 2024-11-12 17:09:31 -08:00 · 8b7be87bec
commit 8b7be87bec
parent 8645f8bc9e
7 changed files with 91 additions and 99 deletions
--- a/llama_stack/providers/registry/memory.py
+++ b/llama_stack/providers/registry/memory.py
@ -53,6 +53,7 @@ def available_providers() -> List[ProviderSpec]:
                adapter_type="chromadb",
                pip_packages=EMBEDDING_DEPS + ["chromadb-client"],
                module="llama_stack.providers.remote.memory.chroma",
+                config_class="llama_stack.distribution.datatypes.RemoteProviderConfig",
            ),
        ),
        remote_provider_spec(