phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-13 22:17:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(api)!: support extra_body to embeddings and vector_stores APIs (#3794)
Some checks failed
Integration Auth Tests / test-matrix (oauth2_token) (push) Failing after 0s
Details
Python Package Build Test / build (3.12) (push) Failing after 1s
Details
Unit Tests / unit-tests (3.13) (push) Failing after 4s
Details
SqlStore Integration Tests / test-postgres (3.12) (push) Failing after 0s
Details
SqlStore Integration Tests / test-postgres (3.13) (push) Failing after 0s
Details
Test External Providers Installed via Module / test-external-providers-from-module (venv) (push) Has been skipped
Details
Python Package Build Test / build (3.13) (push) Failing after 1s
Details
Integration Tests (Replay) / Integration Tests (, , , client=, ) (push) Failing after 3s
Details
Vector IO Integration Tests / test-matrix (push) Failing after 5s
Details
Test External API and Providers / test-external (venv) (push) Failing after 5s
Details
Unit Tests / unit-tests (3.12) (push) Failing after 4s
Details
API Conformance Tests / check-schema-compatibility (push) Successful in 10s
Details
UI Tests / ui-tests (22) (push) Successful in 40s
Details
Pre-commit / pre-commit (push) Successful in 1m23s
Details

Browse source 
Applies the same pattern from
https://github.com/llamastack/llama-stack/pull/3777 to embeddings and
vector_stores.create() endpoints.

This should _not_ be a breaking change since (a) our tests were already
using the `extra_body` parameter when passing in to the backend (b) but
the backend probably wasn't extracting the parameters correctly. This PR
will fix that.

Updated APIs: `openai_embeddings(), openai_create_vector_store(),
openai_create_vector_store_file_batch()`
This commit is contained in:
Ashwin Bharambe 2025-10-12 19:01:52 -07:00 • committed by GitHub
parent 3bb6ef351b
commit ecc8a554d2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
26 changed files with 451 additions and 426 deletions
Download patch file Download diff file Expand all files Collapse all files

54
docs/static/llama-stack-spec.html vendored
View file

@ -765,7 +765,7 @@
"content": {
"application/json": {
"schema": {
"$ref": "#/components/schemas/OpenaiEmbeddingsRequest"
"$ref": "#/components/schemas/OpenAIEmbeddingsRequestWithExtraBody"
}
}
},
@ -3170,13 +3170,13 @@
"VectorIO"
],
"summary": "Creates a vector store.",
"description": "Creates a vector store.",
"description": "Creates a vector store.\nGenerate an OpenAI-compatible vector store with the given parameters.",
"parameters": [],
"requestBody": {
"content": {
"application/json": {
"schema": {
"$ref": "#/components/schemas/OpenaiCreateVectorStoreRequest"
"$ref": "#/components/schemas/OpenAICreateVectorStoreRequestWithExtraBody"
}
}
},
@ -3356,7 +3356,7 @@
"VectorIO"
],
"summary": "Create a vector store file batch.",
"description": "Create a vector store file batch.",
"description": "Create a vector store file batch.\nGenerate an OpenAI-compatible vector store file batch for the given vector store.",
"parameters": [
{
"name": "vector_store_id",
@ -3372,7 +3372,7 @@
"content": {
"application/json": {
"schema": {
"$ref": "#/components/schemas/OpenaiCreateVectorStoreFileBatchRequest"
"$ref": "#/components/schemas/OpenAICreateVectorStoreFileBatchRequestWithExtraBody"
}
}
},
@ -6324,7 +6324,7 @@
"title": "ConversationItemDeletedResource",
"description": "Response for deleted conversation item."
},
"OpenaiEmbeddingsRequest": {
"OpenAIEmbeddingsRequestWithExtraBody": {
"type": "object",
"properties": {
"model": {
@ -6347,6 +6347,7 @@
},
"encoding_format": {
"type": "string",
"default": "float",
"description": "(Optional) The format to return the embeddings in. Can be either \"float\" or \"base64\". Defaults to \"float\"."
},
"dimensions": {
@ -6363,7 +6364,8 @@
"model",
"input"
],
"title": "OpenaiEmbeddingsRequest"
"title": "OpenAIEmbeddingsRequestWithExtraBody",
"description": "Request parameters for OpenAI-compatible embeddings endpoint."
},
"OpenAIEmbeddingData": {
"type": "object",
@ -12587,19 +12589,19 @@
"title": "VectorStoreObject",
"description": "OpenAI Vector Store object."
},
"OpenaiCreateVectorStoreRequest": {
"OpenAICreateVectorStoreRequestWithExtraBody": {
"type": "object",
"properties": {
"name": {
"type": "string",
"description": "A name for the vector store."
"description": "(Optional) A name for the vector store"
},
"file_ids": {
"type": "array",
"items": {
"type": "string"
},
"description": "A list of File IDs that the vector store should use. Useful for tools like `file_search` that can access files."
"description": "List of file IDs to include in the vector store"
},
"expires_after": {
"type": "object",
@ -12625,7 +12627,7 @@
}
]
},
"description": "The expiration policy for a vector store."
"description": "(Optional) Expiration policy for the vector store"
},
"chunking_strategy": {
"type": "object",
@ -12651,7 +12653,7 @@
}
]
},
"description": "The chunking strategy used to chunk the file(s). If not set, will use the `auto` strategy."
"description": "(Optional) Strategy for splitting files into chunks"
},
"metadata": {
"type": "object",
@ -12677,23 +12679,12 @@
}
]
},
"description": "Set of 16 key-value pairs that can be attached to an object."
},
"embedding_model": {
"type": "string",
"description": "The embedding model to use for this vector store."
},
"embedding_dimension": {
"type": "integer",
"description": "The dimension of the embedding vectors (default: 384)."
},
"provider_id": {
"type": "string",
"description": "The ID of the provider to use for this vector store."
"description": "Set of key-value pairs that can be attached to the vector store"
}
},
"additionalProperties": false,
"title": "OpenaiCreateVectorStoreRequest"
"title": "OpenAICreateVectorStoreRequestWithExtraBody",
"description": "Request to create a vector store with extra_body support."
},
"OpenaiUpdateVectorStoreRequest": {
"type": "object",
@ -12863,7 +12854,7 @@
"title": "VectorStoreChunkingStrategyStaticConfig",
"description": "Configuration for static chunking strategy."
},
"OpenaiCreateVectorStoreFileBatchRequest": {
"OpenAICreateVectorStoreFileBatchRequestWithExtraBody": {
"type": "object",
"properties": {
"file_ids": {
@ -12871,7 +12862,7 @@
"items": {
"type": "string"
},
"description": "A list of File IDs that the vector store should use."
"description": "A list of File IDs that the vector store should use"
},
"attributes": {
"type": "object",
@ -12897,18 +12888,19 @@
}
]
},
"description": "(Optional) Key-value attributes to store with the files."
"description": "(Optional) Key-value attributes to store with the files"
},
"chunking_strategy": {
"$ref": "#/components/schemas/VectorStoreChunkingStrategy",
"description": "(Optional) The chunking strategy used to chunk the file(s). Defaults to auto."
"description": "(Optional) The chunking strategy used to chunk the file(s). Defaults to auto"
}
},
"additionalProperties": false,
"required": [
"file_ids"
],
"title": "OpenaiCreateVectorStoreFileBatchRequest"
"title": "OpenAICreateVectorStoreFileBatchRequestWithExtraBody",
"description": "Request to create a vector store file batch with extra_body support."
},
"VectorStoreFileBatchObject": {
"type": "object",