From 0e3348b394133420405f9246756423a998bf6b7d Mon Sep 17 00:00:00 2001 From: Francisco Javier Arceo Date: Tue, 1 Jul 2025 10:56:10 -0400 Subject: [PATCH] updating Signed-off-by: Francisco Javier Arceo --- llama_stack/providers/remote/vector_io/milvus/milvus.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/vector_io/milvus/milvus.py b/llama_stack/providers/remote/vector_io/milvus/milvus.py index a4ed21e64..be0c65bcf 100644 --- a/llama_stack/providers/remote/vector_io/milvus/milvus.py +++ b/llama_stack/providers/remote/vector_io/milvus/milvus.py @@ -11,7 +11,7 @@ import os from typing import Any from numpy.typing import NDArray -from pymilvus import MilvusClient +from pymilvus import DataType, MilvusClient from llama_stack.apis.files.files import Files from llama_stack.apis.inference import Inference, InterleavedContent @@ -41,6 +41,7 @@ VECTOR_INDEX_PREFIX = f"vector_index:milvus:{VERSION}::" OPENAI_VECTOR_STORES_PREFIX = f"openai_vector_stores:milvus:{VERSION}::" OPENAI_VECTOR_STORES_FILES_PREFIX = f"openai_vector_stores_files:milvus:{VERSION}::" OPENAI_VECTOR_STORES_FILES_CONTENTS_PREFIX = f"openai_vector_stores_files_contents:milvus:{VERSION}::" +from llama_stack.providers.utils.vector_io.chunk_utils import generate_chunk_id class MilvusIndex(EmbeddingIndex):