diff --git a/llama_stack/providers/remote/vector_io/chroma/chroma.py b/llama_stack/providers/remote/vector_io/chroma/chroma.py index 4c5d3b1b0..ffe2cba44 100644 --- a/llama_stack/providers/remote/vector_io/chroma/chroma.py +++ b/llama_stack/providers/remote/vector_io/chroma/chroma.py @@ -217,7 +217,6 @@ class ChromaVectorIOAdapter(VectorIO, VectorDBsProtocolPrivate): embedding_model: str | None = None, embedding_dimension: int | None = 384, provider_id: str | None = None, - provider_vector_db_id: str = "", ) -> VectorStoreObject: raise NotImplementedError("OpenAI Vector Stores API is not supported in Chroma") diff --git a/llama_stack/providers/remote/vector_io/pgvector/pgvector.py b/llama_stack/providers/remote/vector_io/pgvector/pgvector.py index 38ec3279b..c3cdef9b8 100644 --- a/llama_stack/providers/remote/vector_io/pgvector/pgvector.py +++ b/llama_stack/providers/remote/vector_io/pgvector/pgvector.py @@ -247,6 +247,7 @@ class PGVectorVectorIOAdapter(VectorIO, VectorDBsProtocolPrivate): embedding_model: str | None = None, embedding_dimension: int | None = 384, provider_id: str | None = None, + provider_vector_db_id: str | None = None, ) -> VectorStoreObject: raise NotImplementedError("OpenAI Vector Stores API is not supported in PGVector") diff --git a/llama_stack/providers/utils/memory/openai_vector_store_mixin.py b/llama_stack/providers/utils/memory/openai_vector_store_mixin.py index bc13e9bf7..7d47bf747 100644 --- a/llama_stack/providers/utils/memory/openai_vector_store_mixin.py +++ b/llama_stack/providers/utils/memory/openai_vector_store_mixin.py @@ -143,7 +143,7 @@ class OpenAIVectorStoreMixin(ABC): embedding_model: str | None = None, embedding_dimension: int | None = 384, provider_id: str | None = None, - provider_vector_db_id: str = "", + provider_vector_db_id: str | None = None, ) -> VectorStoreObject: """Creates a vector store.""" created_at = int(time.time())