phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 22:03:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

vector_dbs

Browse source
This commit is contained in:
Sai Soundararaj 2025-07-01 16:26:13 -07:00
parent fa65de5fb3
commit 441259f2e3
4 changed files with 39 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

17
llama_stack/apis/vector_dbs/vector_dbs.py
View file

@ -15,6 +15,12 @@ from llama_stack.schema_utils import json_schema_type, webmethod
@json_schema_type
class VectorDB(Resource):
"""Vector database resource for storing and querying vector embeddings.
:param type: Type of resource, always 'vector_db' for vector databases
:param embedding_model: Name of the embedding model to use for vector generation
:param embedding_dimension: Dimension of the embedding vectors
"""
type: Literal[ResourceType.vector_db] = ResourceType.vector_db
embedding_model: str
@ -30,6 +36,13 @@ class VectorDB(Resource):
class VectorDBInput(BaseModel):
"""Input parameters for creating or configuring a vector database.
:param vector_db_id: Unique identifier for the vector database
:param embedding_model: Name of the embedding model to use for vector generation
:param embedding_dimension: Dimension of the embedding vectors
:param provider_vector_db_id: (Optional) Provider-specific identifier for the vector database
"""
vector_db_id: str
embedding_model: str
embedding_dimension: int
@ -37,6 +50,10 @@ class VectorDBInput(BaseModel):
class ListVectorDBsResponse(BaseModel):
"""Response from listing vector databases.
:param data: List of vector databases
"""
data: list[VectorDB]