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

fix: move models.py to top-level init

Browse source 
All batch models are now exported from the top level for better
discoverability and IDE support.

Signed-off-by: Sébastien Han <seb@redhat.com>
This commit is contained in:
Sébastien Han 2025-11-21 15:56:44 +01:00
parent 6f552e0a31
commit ac816a6b25
No known key found for this signature in database
1 changed files with 13 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

14
src/llama_stack_api/__init__.py
View file

@ -26,7 +26,15 @@ from . import common # noqa: F401
# Import all public API symbols # Import all public API symbols
from .agents import Agents, ResponseGuardrail, ResponseGuardrailSpec from .agents import Agents, ResponseGuardrail, ResponseGuardrailSpec
from .batches import Batches, BatchObject, ListBatchesResponse from .batches import (
Batches,
BatchObject,
CancelBatchRequest,
CreateBatchRequest,
ListBatchesRequest,
ListBatchesResponse,
RetrieveBatchRequest,
)
from .benchmarks import ( from .benchmarks import (
Benchmark, Benchmark,
BenchmarkInput, BenchmarkInput,
@ -462,6 +470,9 @@ __all__ = [
"BasicScoringFnParams", "BasicScoringFnParams",
"Batches", "Batches",
"BatchObject", "BatchObject",
"CancelBatchRequest",
"CreateBatchRequest",
"ListBatchesRequest",
"Benchmark", "Benchmark",
"BenchmarkConfig", "BenchmarkConfig",
"BenchmarkInput", "BenchmarkInput",
@ -555,6 +566,7 @@ __all__ = [
"LLMAsJudgeScoringFnParams", "LLMAsJudgeScoringFnParams",
"LLMRAGQueryGeneratorConfig", "LLMRAGQueryGeneratorConfig",
"ListBatchesResponse", "ListBatchesResponse",
"RetrieveBatchRequest",
"ListBenchmarksResponse", "ListBenchmarksResponse",
"ListDatasetsResponse", "ListDatasetsResponse",
"ListModelsResponse", "ListModelsResponse",