From 73c6f6126f969e630c4d5f0a305308e53dc13183 Mon Sep 17 00:00:00 2001 From: Luis Tomas Bolivar Date: Fri, 28 Feb 2025 01:47:26 +0100 Subject: [PATCH] fix: Avoid unexpected keyword argument for sentence_transformers (#1269) Now that remote-vllm include inline::sentence_transformers there is an issue building the image: Error building stack: SentenceTransformersInferenceConfig.sample_run_config() got an unexpected keyword argument '__distro_dir__' To avoid that issue this fix extends the sample_run_config to accept extra kwargs --- .../providers/inline/inference/sentence_transformers/config.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/inline/inference/sentence_transformers/config.py b/llama_stack/providers/inline/inference/sentence_transformers/config.py index 232e4bf32..93e0afe11 100644 --- a/llama_stack/providers/inline/inference/sentence_transformers/config.py +++ b/llama_stack/providers/inline/inference/sentence_transformers/config.py @@ -11,5 +11,5 @@ from pydantic import BaseModel class SentenceTransformersInferenceConfig(BaseModel): @classmethod - def sample_run_config(cls) -> Dict[str, Any]: + def sample_run_config(cls, **kwargs) -> Dict[str, Any]: return {}