From 59830e5a22c60a8b82ea0dc08f94677c0b26c768 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?S=C3=A9bastien=20Han?= Date: Tue, 3 Jun 2025 15:11:54 +0200 Subject: [PATCH] chore: return NotImplementedError instead of ValueError MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Sébastien Han --- .../inference/sentence_transformers/sentence_transformers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py b/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py index 890c526f5..057c9bb50 100644 --- a/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py +++ b/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py @@ -65,7 +65,7 @@ class SentenceTransformersInferenceImpl( stream: bool | None = False, logprobs: LogProbConfig | None = None, ) -> CompletionResponse | AsyncGenerator: - raise ValueError("Sentence transformers don't support completion") + raise NotImplementedError("Sentence transformers don't support completion") async def chat_completion( self, @@ -80,7 +80,7 @@ class SentenceTransformersInferenceImpl( logprobs: LogProbConfig | None = None, tool_config: ToolConfig | None = None, ) -> AsyncGenerator: - raise ValueError("Sentence transformers don't support chat completion") + raise NotImplementedError("Sentence transformers don't support chat completion") async def batch_completion( self,