mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-08-21 17:33:12 +00:00
914c7be288
Add complete batches API implementation with protocol, providers, and tests: Core Infrastructure: - Add batches API protocol using OpenAI Batch types directly - Add Api.batches enum value and protocol mapping in resolver - Add OpenAI "batch" file purpose support - Include proper error handling (ConflictError, ResourceNotFoundError) Reference Provider: - Add ReferenceBatchesImpl with full CRUD operations (create, retrieve, cancel, list) - Implement background batch processing with configurable concurrency - Add SQLite KVStore backend for persistence - Support /v1/chat/completions endpoint with request validation Comprehensive Test Suite: - Add unit tests for provider implementation with validation - Add integration tests for end-to-end batch processing workflows - Add error handling tests for validation, malformed inputs, and edge cases Configuration: - Add max_concurrent_batches and max_concurrent_requests_per_batch options - Add provider documentation with sample configurations Test with - ``` $ uv run llama stack build --image-type venv --providers inference=YOU_PICK,files=inline::localfs,batches=inline::reference --run & $ LLAMA_STACK_CONFIG=http://localhost:8321 uv run pytest tests/unit/providers/batches tests/integration/batches --text-model YOU_PICK ``` addresses #3066 --------- Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: Ashwin Bharambe <ashwin.bharambe@gmail.com>
41 lines
919 B
Markdown
41 lines
919 B
Markdown
# Inference
|
|
|
|
## Overview
|
|
|
|
Llama Stack Inference API for generating completions, chat completions, and embeddings.
|
|
|
|
This API provides the raw interface to the underlying models. Two kinds of models are supported:
|
|
- LLM models: these models generate "raw" and "chat" (conversational) completions.
|
|
- Embedding models: these models generate embeddings to be used for semantic search.
|
|
|
|
This section contains documentation for all available providers for the **inference** API.
|
|
|
|
## Providers
|
|
|
|
```{toctree}
|
|
:maxdepth: 1
|
|
|
|
inline_meta-reference
|
|
inline_sentence-transformers
|
|
remote_anthropic
|
|
remote_bedrock
|
|
remote_cerebras
|
|
remote_databricks
|
|
remote_fireworks
|
|
remote_gemini
|
|
remote_groq
|
|
remote_hf_endpoint
|
|
remote_hf_serverless
|
|
remote_llama-openai-compat
|
|
remote_nvidia
|
|
remote_ollama
|
|
remote_openai
|
|
remote_passthrough
|
|
remote_runpod
|
|
remote_sambanova
|
|
remote_tgi
|
|
remote_together
|
|
remote_vertexai
|
|
remote_vllm
|
|
remote_watsonx
|
|
```
|