diff --git a/llama_stack/providers/remote/memory/chroma/__init__.py b/llama_stack/providers/remote/memory/chroma/__init__.py index 9e011498c..581d60e75 100644 --- a/llama_stack/providers/remote/memory/chroma/__init__.py +++ b/llama_stack/providers/remote/memory/chroma/__init__.py @@ -6,11 +6,14 @@ from typing import Dict -from .config import ChromaRemoteImplConfig from llama_stack.providers.datatypes import Api, ProviderSpec +from .config import ChromaRemoteImplConfig -async def get_adapter_impl(config: ChromaRemoteImplConfig, deps: Dict[Api, ProviderSpec]): + +async def get_adapter_impl( + config: ChromaRemoteImplConfig, deps: Dict[Api, ProviderSpec] +): from .chroma import ChromaMemoryAdapter impl = ChromaMemoryAdapter(config, deps[Api.inference])