chore: re-add missing decorator

Signed-off-by: Sébastien Han <seb@redhat.com>

client-sdks/stainless/openapi.yml

@@ -7180,6 +7180,17 @@ components:
       - scoring_functions
       title: Benchmark
       type: object
+    ListBenchmarksResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Benchmark'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListBenchmarksResponse
+      type: object
     ImageDelta:
       description: An image content delta for streaming responses.
       properties:
@@ -7630,6 +7641,18 @@ components:
       - source
       title: Dataset
       type: object
+    ListDatasetsResponse:
+      description: Response from listing datasets.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Dataset'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListDatasetsResponse
+      type: object
     Error:
       description: Error response from the API. Roughly follows RFC 7807.
       properties:
@@ -9612,6 +9635,17 @@ components:
       - owned_by
       title: OpenAIModel
       type: object
+    OpenAIListModelsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/OpenAIModel'
+          title: Data
+          type: array
+      required:
+      - data
+      title: OpenAIListModelsResponse
+      type: object
     DPOLossType:
       enum:
       - sigmoid
@@ -9701,6 +9735,26 @@ components:
           default: false
       title: EfficiencyConfig
       type: object
+    PostTrainingJob:
+      properties:
+        job_uuid:
+          title: Job Uuid
+          type: string
+      required:
+      - job_uuid
+      title: PostTrainingJob
+      type: object
+    ListPostTrainingJobsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/PostTrainingJob'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPostTrainingJobsResponse
+      type: object
     OptimizerType:
       description: Available optimizer algorithms for training.
       enum:
@@ -9935,6 +9989,18 @@ components:
       - prompt_id
       title: Prompt
       type: object
+    ListPromptsResponse:
+      description: Response model to list prompts.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Prompt'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPromptsResponse
+      type: object
     ProviderInfo:
       description: Information about a registered provider including its configuration and health status.
       properties:
@@ -9963,6 +10029,18 @@ components:
       - health
       title: ProviderInfo
       type: object
+    ListProvidersResponse:
+      description: Response containing a list of all available providers.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ProviderInfo'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListProvidersResponse
+      type: object
     ModerationObjectResults:
       description: A moderation object.
       properties:
@@ -10092,6 +10170,28 @@ components:
       - results
       title: ScoreResponse
       type: object
+    ListScoringFunctionsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ScoringFn'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListScoringFunctionsResponse
+      type: object
+    ListShieldsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Shield'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListShieldsResponse
+      type: object
     ToolDef:
       description: Tool definition used in runtime contexts.
       properties:
@@ -10142,6 +10242,18 @@ components:
       - data
       title: ListToolDefsResponse
       type: object
+    ListToolGroupsResponse:
+      description: Response containing a list of tool groups.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ToolGroup'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListToolGroupsResponse
+      type: object
     ToolGroupInput:
       description: Input data for registering a tool group.
       properties:

docs/static/deprecated-llama-stack-spec.yaml

@@ -4923,6 +4923,17 @@ components:
       - scoring_functions
       title: Benchmark
       type: object
+    ListBenchmarksResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Benchmark'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListBenchmarksResponse
+      type: object
     ImageDelta:
       description: An image content delta for streaming responses.
       properties:
@@ -5373,6 +5384,18 @@ components:
       - source
       title: Dataset
       type: object
+    ListDatasetsResponse:
+      description: Response from listing datasets.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Dataset'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListDatasetsResponse
+      type: object
     Error:
       description: Error response from the API. Roughly follows RFC 7807.
       properties:
@@ -7355,6 +7378,17 @@ components:
       - owned_by
       title: OpenAIModel
       type: object
+    OpenAIListModelsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/OpenAIModel'
+          title: Data
+          type: array
+      required:
+      - data
+      title: OpenAIListModelsResponse
+      type: object
     DPOLossType:
       enum:
       - sigmoid
@@ -7444,6 +7478,26 @@ components:
           default: false
       title: EfficiencyConfig
       type: object
+    PostTrainingJob:
+      properties:
+        job_uuid:
+          title: Job Uuid
+          type: string
+      required:
+      - job_uuid
+      title: PostTrainingJob
+      type: object
+    ListPostTrainingJobsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/PostTrainingJob'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPostTrainingJobsResponse
+      type: object
     OptimizerType:
       description: Available optimizer algorithms for training.
       enum:
@@ -7678,6 +7732,18 @@ components:
       - prompt_id
       title: Prompt
       type: object
+    ListPromptsResponse:
+      description: Response model to list prompts.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Prompt'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPromptsResponse
+      type: object
     ProviderInfo:
       description: Information about a registered provider including its configuration and health status.
       properties:
@@ -7706,6 +7772,18 @@ components:
       - health
       title: ProviderInfo
       type: object
+    ListProvidersResponse:
+      description: Response containing a list of all available providers.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ProviderInfo'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListProvidersResponse
+      type: object
     ModerationObjectResults:
       description: A moderation object.
       properties:
@@ -7835,6 +7913,28 @@ components:
       - results
       title: ScoreResponse
       type: object
+    ListScoringFunctionsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ScoringFn'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListScoringFunctionsResponse
+      type: object
+    ListShieldsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Shield'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListShieldsResponse
+      type: object
     ToolDef:
       description: Tool definition used in runtime contexts.
       properties:
@@ -7885,6 +7985,18 @@ components:
       - data
       title: ListToolDefsResponse
       type: object
+    ListToolGroupsResponse:
+      description: Response containing a list of tool groups.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ToolGroup'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListToolGroupsResponse
+      type: object
     ToolGroupInput:
       description: Input data for registering a tool group.
       properties:

docs/static/experimental-llama-stack-spec.yaml

@@ -4775,6 +4775,17 @@ components:
       - scoring_functions
       title: Benchmark
       type: object
+    ListBenchmarksResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Benchmark'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListBenchmarksResponse
+      type: object
     ImageDelta:
       description: An image content delta for streaming responses.
       properties:
@@ -5225,6 +5236,18 @@ components:
       - source
       title: Dataset
       type: object
+    ListDatasetsResponse:
+      description: Response from listing datasets.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Dataset'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListDatasetsResponse
+      type: object
     Error:
       description: Error response from the API. Roughly follows RFC 7807.
       properties:
@@ -7207,6 +7230,17 @@ components:
       - owned_by
       title: OpenAIModel
       type: object
+    OpenAIListModelsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/OpenAIModel'
+          title: Data
+          type: array
+      required:
+      - data
+      title: OpenAIListModelsResponse
+      type: object
     DPOLossType:
       enum:
       - sigmoid
@@ -7296,6 +7330,26 @@ components:
           default: false
       title: EfficiencyConfig
       type: object
+    PostTrainingJob:
+      properties:
+        job_uuid:
+          title: Job Uuid
+          type: string
+      required:
+      - job_uuid
+      title: PostTrainingJob
+      type: object
+    ListPostTrainingJobsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/PostTrainingJob'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPostTrainingJobsResponse
+      type: object
     OptimizerType:
       description: Available optimizer algorithms for training.
       enum:
@@ -7530,6 +7584,18 @@ components:
       - prompt_id
       title: Prompt
       type: object
+    ListPromptsResponse:
+      description: Response model to list prompts.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Prompt'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPromptsResponse
+      type: object
     ProviderInfo:
       description: Information about a registered provider including its configuration and health status.
       properties:
@@ -7558,6 +7624,18 @@ components:
       - health
       title: ProviderInfo
       type: object
+    ListProvidersResponse:
+      description: Response containing a list of all available providers.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ProviderInfo'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListProvidersResponse
+      type: object
     ModerationObjectResults:
       description: A moderation object.
       properties:
@@ -7687,6 +7765,28 @@ components:
       - results
       title: ScoreResponse
       type: object
+    ListScoringFunctionsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ScoringFn'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListScoringFunctionsResponse
+      type: object
+    ListShieldsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Shield'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListShieldsResponse
+      type: object
     ToolDef:
       description: Tool definition used in runtime contexts.
       properties:
@@ -7737,6 +7837,18 @@ components:
       - data
       title: ListToolDefsResponse
       type: object
+    ListToolGroupsResponse:
+      description: Response containing a list of tool groups.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ToolGroup'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListToolGroupsResponse
+      type: object
     ToolGroupInput:
       description: Input data for registering a tool group.
       properties:

docs/static/llama-stack-spec.yaml

@@ -6336,6 +6336,17 @@ components:
       - scoring_functions
       title: Benchmark
       type: object
+    ListBenchmarksResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Benchmark'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListBenchmarksResponse
+      type: object
     ImageDelta:
       description: An image content delta for streaming responses.
       properties:
@@ -6786,6 +6797,18 @@ components:
       - source
       title: Dataset
       type: object
+    ListDatasetsResponse:
+      description: Response from listing datasets.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Dataset'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListDatasetsResponse
+      type: object
     Error:
       description: Error response from the API. Roughly follows RFC 7807.
       properties:
@@ -8768,6 +8791,17 @@ components:
       - owned_by
       title: OpenAIModel
       type: object
+    OpenAIListModelsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/OpenAIModel'
+          title: Data
+          type: array
+      required:
+      - data
+      title: OpenAIListModelsResponse
+      type: object
     DPOLossType:
       enum:
       - sigmoid
@@ -8857,6 +8891,26 @@ components:
           default: false
       title: EfficiencyConfig
       type: object
+    PostTrainingJob:
+      properties:
+        job_uuid:
+          title: Job Uuid
+          type: string
+      required:
+      - job_uuid
+      title: PostTrainingJob
+      type: object
+    ListPostTrainingJobsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/PostTrainingJob'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPostTrainingJobsResponse
+      type: object
     OptimizerType:
       description: Available optimizer algorithms for training.
       enum:
@@ -9091,6 +9145,18 @@ components:
       - prompt_id
       title: Prompt
       type: object
+    ListPromptsResponse:
+      description: Response model to list prompts.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Prompt'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPromptsResponse
+      type: object
     ProviderInfo:
       description: Information about a registered provider including its configuration and health status.
       properties:
@@ -9119,6 +9185,18 @@ components:
       - health
       title: ProviderInfo
       type: object
+    ListProvidersResponse:
+      description: Response containing a list of all available providers.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ProviderInfo'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListProvidersResponse
+      type: object
     ModerationObjectResults:
       description: A moderation object.
       properties:
@@ -9248,6 +9326,28 @@ components:
       - results
       title: ScoreResponse
       type: object
+    ListScoringFunctionsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ScoringFn'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListScoringFunctionsResponse
+      type: object
+    ListShieldsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Shield'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListShieldsResponse
+      type: object
     ToolDef:
       description: Tool definition used in runtime contexts.
       properties:
@@ -9298,6 +9398,18 @@ components:
       - data
       title: ListToolDefsResponse
       type: object
+    ListToolGroupsResponse:
+      description: Response containing a list of tool groups.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ToolGroup'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListToolGroupsResponse
+      type: object
     ToolGroupInput:
       description: Input data for registering a tool group.
       properties:

docs/static/stainless-llama-stack-spec.yaml

@@ -7180,6 +7180,17 @@ components:
       - scoring_functions
       title: Benchmark
       type: object
+    ListBenchmarksResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Benchmark'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListBenchmarksResponse
+      type: object
     ImageDelta:
       description: An image content delta for streaming responses.
       properties:
@@ -7630,6 +7641,18 @@ components:
       - source
       title: Dataset
       type: object
+    ListDatasetsResponse:
+      description: Response from listing datasets.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Dataset'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListDatasetsResponse
+      type: object
     Error:
       description: Error response from the API. Roughly follows RFC 7807.
       properties:
@@ -9612,6 +9635,17 @@ components:
       - owned_by
       title: OpenAIModel
       type: object
+    OpenAIListModelsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/OpenAIModel'
+          title: Data
+          type: array
+      required:
+      - data
+      title: OpenAIListModelsResponse
+      type: object
     DPOLossType:
       enum:
       - sigmoid
@@ -9701,6 +9735,26 @@ components:
           default: false
       title: EfficiencyConfig
       type: object
+    PostTrainingJob:
+      properties:
+        job_uuid:
+          title: Job Uuid
+          type: string
+      required:
+      - job_uuid
+      title: PostTrainingJob
+      type: object
+    ListPostTrainingJobsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/PostTrainingJob'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPostTrainingJobsResponse
+      type: object
     OptimizerType:
       description: Available optimizer algorithms for training.
       enum:
@@ -9935,6 +9989,18 @@ components:
       - prompt_id
       title: Prompt
       type: object
+    ListPromptsResponse:
+      description: Response model to list prompts.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Prompt'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListPromptsResponse
+      type: object
     ProviderInfo:
       description: Information about a registered provider including its configuration and health status.
       properties:
@@ -9963,6 +10029,18 @@ components:
       - health
       title: ProviderInfo
       type: object
+    ListProvidersResponse:
+      description: Response containing a list of all available providers.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ProviderInfo'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListProvidersResponse
+      type: object
     ModerationObjectResults:
       description: A moderation object.
       properties:
@@ -10092,6 +10170,28 @@ components:
       - results
       title: ScoreResponse
       type: object
+    ListScoringFunctionsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ScoringFn'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListScoringFunctionsResponse
+      type: object
+    ListShieldsResponse:
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/Shield'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListShieldsResponse
+      type: object
     ToolDef:
       description: Tool definition used in runtime contexts.
       properties:
@@ -10142,6 +10242,18 @@ components:
       - data
       title: ListToolDefsResponse
       type: object
+    ListToolGroupsResponse:
+      description: Response containing a list of tool groups.
+      properties:
+        data:
+          items:
+            $ref: '#/components/schemas/ToolGroup'
+          title: Data
+          type: array
+      required:
+      - data
+      title: ListToolGroupsResponse
+      type: object
     ToolGroupInput:
       description: Input data for registering a tool group.
       properties:

src/llama_stack_api/benchmarks.py

@@ -48,6 +48,7 @@ class BenchmarkInput(CommonBenchmarkFields, BaseModel):
     provider_benchmark_id: str | None = None
 
 
+@json_schema_type
 class ListBenchmarksResponse(BaseModel):
     data: list[Benchmark]
 

src/llama_stack_api/datasets.py

@@ -136,6 +136,7 @@ class DatasetInput(CommonDatasetFields, BaseModel):
     dataset_id: str
 
 
+@json_schema_type
 class ListDatasetsResponse(BaseModel):
     """Response from listing datasets.
 

src/llama_stack_api/models.py

@@ -100,6 +100,7 @@ class OpenAIModel(BaseModel):
     custom_metadata: dict[str, Any] | None = None
 
 
+@json_schema_type
 class OpenAIListModelsResponse(BaseModel):
     data: list[OpenAIModel]
 

src/llama_stack_api/post_training.py

@@ -236,6 +236,7 @@ class PostTrainingRLHFRequest(BaseModel):
     logger_config: dict[str, Any]
 
 
+@json_schema_type
 class PostTrainingJob(BaseModel):
     job_uuid: str
 
@@ -265,6 +266,7 @@ class PostTrainingJobStatusResponse(BaseModel):
     checkpoints: list[Checkpoint] = Field(default_factory=list)
 
 
+@json_schema_type
 class ListPostTrainingJobsResponse(BaseModel):
     data: list[PostTrainingJob]
 

src/llama_stack_api/prompts.py

@@ -85,6 +85,7 @@ class Prompt(BaseModel):
         return f"pmpt_{hex_string}"
 
 
+@json_schema_type
 class ListPromptsResponse(BaseModel):
     """Response model to list prompts."""
 

src/llama_stack_api/providers.py

@@ -31,6 +31,7 @@ class ProviderInfo(BaseModel):
     health: HealthResponse
 
 
+@json_schema_type
 class ListProvidersResponse(BaseModel):
     """Response containing a list of all available providers.
 

src/llama_stack_api/scoring_functions.py

@@ -155,6 +155,7 @@ class ScoringFnInput(CommonScoringFnFields, BaseModel):
     provider_scoring_fn_id: str | None = None
 
 
+@json_schema_type
 class ListScoringFunctionsResponse(BaseModel):
     data: list[ScoringFn]
 

src/llama_stack_api/shields.py

@@ -43,6 +43,7 @@ class ShieldInput(CommonShieldFields):
     provider_shield_id: str | None = None
 
 
+@json_schema_type
 class ListShieldsResponse(BaseModel):
     data: list[Shield]
 

src/llama_stack_api/tools.py

@@ -88,6 +88,7 @@ class ToolStore(Protocol):
     async def get_tool_group(self, toolgroup_id: str) -> ToolGroup: ...
 
 
+@json_schema_type
 class ListToolGroupsResponse(BaseModel):
     """Response containing a list of tool groups.