From 3cda82be3afe8396e7b97bb382141838db5096f8 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Wed, 23 Jul 2025 16:36:24 -0700 Subject: [PATCH] fix import --- .../inference/sentence_transformers/sentence_transformers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 95b99c65e..fea8a8189 100644 --- a/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py +++ b/llama_stack/providers/inline/inference/sentence_transformers/sentence_transformers.py @@ -20,7 +20,8 @@ from llama_stack.apis.inference import ( ToolDefinition, ToolPromptFormat, ) -from llama_stack.providers.datatypes import Model, ModelsProtocolPrivate, ModelType +from llama_stack.apis.models import ModelType +from llama_stack.providers.datatypes import Model, ModelsProtocolPrivate from llama_stack.providers.utils.inference.embedding_mixin import ( SentenceTransformerEmbeddingMixin, )