update script

This commit is contained in:
Xi Yan 2024-09-09 11:19:57 -07:00
parent 84b8a53a34
commit b7b8f5c2c3
3 changed files with 126 additions and 127 deletions

View file

@ -21,7 +21,7 @@
"info": { "info": {
"title": "[DRAFT] Llama Stack Specification", "title": "[DRAFT] Llama Stack Specification",
"version": "0.0.1", "version": "0.0.1",
"description": "This is the specification of the llama stack that provides\n a set of endpoints and their corresponding interfaces that are tailored to\n best leverage Llama Models. The specification is still in draft and subject to change.\n Generated at 2024-09-09 11:14:56.584501" "description": "This is the specification of the llama stack that provides\n a set of endpoints and their corresponding interfaces that are tailored to\n best leverage Llama Models. The specification is still in draft and subject to change.\n Generated at 2024-09-09 11:19:39.855375"
}, },
"servers": [ "servers": [
{ {
@ -277,7 +277,7 @@
"content": { "content": {
"application/json": { "application/json": {
"schema": { "schema": {
"$ref": "#/components/schemas/CreateAgenticSystemTurnRequestWrapper" "$ref": "#/components/schemas/AgenticSystemTurnCreateRequestWrapper"
} }
} }
}, },
@ -1024,7 +1024,7 @@
"content": { "content": {
"application/json": { "application/json": {
"schema": { "schema": {
"$ref": "#/components/schemas/GetLogsRequestWrapper" "$ref": "#/components/schemas/LogSearchRequestWrapper"
} }
} }
}, },
@ -1365,7 +1365,7 @@
"content": { "content": {
"application/json": { "application/json": {
"schema": { "schema": {
"$ref": "#/components/schemas/PreferenceOptimizeRequestWrapper" "$ref": "#/components/schemas/PostTrainingRLHFRequestWrapper"
} }
} }
}, },
@ -1425,7 +1425,7 @@
"content": { "content": {
"application/json": { "application/json": {
"schema": { "schema": {
"$ref": "#/components/schemas/RewardScoreRequestWrapper" "$ref": "#/components/schemas/RewardScoringRequestWrapper"
} }
} }
}, },
@ -1455,7 +1455,7 @@
"content": { "content": {
"application/json": { "application/json": {
"schema": { "schema": {
"$ref": "#/components/schemas/SupervisedFineTuneRequestWrapper" "$ref": "#/components/schemas/PostTrainingSFTRequestWrapper"
} }
} }
}, },
@ -1485,7 +1485,7 @@
"content": { "content": {
"application/json": { "application/json": {
"schema": { "schema": {
"$ref": "#/components/schemas/SyntheticDataGenerateRequestWrapper" "$ref": "#/components/schemas/SyntheticDataGenerationRequestWrapper"
} }
} }
}, },
@ -3229,7 +3229,7 @@
"mime_type" "mime_type"
] ]
}, },
"CreateAgenticSystemTurnRequestWrapper": { "AgenticSystemTurnCreateRequestWrapper": {
"type": "object", "type": "object",
"properties": { "properties": {
"request": { "request": {
@ -4775,7 +4775,7 @@
"query" "query"
] ]
}, },
"GetLogsRequestWrapper": { "LogSearchRequestWrapper": {
"type": "object", "type": "object",
"properties": { "properties": {
"request": { "request": {
@ -5277,7 +5277,7 @@
"fsdp_cpu_offload" "fsdp_cpu_offload"
] ]
}, },
"PreferenceOptimizeRequestWrapper": { "PostTrainingRLHFRequestWrapper": {
"type": "object", "type": "object",
"properties": { "properties": {
"request": { "request": {
@ -5457,7 +5457,7 @@
], ],
"title": "Request to score a reward function. A list of prompts and a list of responses per prompt." "title": "Request to score a reward function. A list of prompts and a list of responses per prompt."
}, },
"RewardScoreRequestWrapper": { "RewardScoringRequestWrapper": {
"type": "object", "type": "object",
"properties": { "properties": {
"request": { "request": {
@ -5754,7 +5754,7 @@
"alpha" "alpha"
] ]
}, },
"SupervisedFineTuneRequestWrapper": { "PostTrainingSFTRequestWrapper": {
"type": "object", "type": "object",
"properties": { "properties": {
"request": { "request": {
@ -5811,7 +5811,7 @@
], ],
"title": "Request to generate synthetic data. A small batch of prompts and a filtering function" "title": "Request to generate synthetic data. A small batch of prompts and a filtering function"
}, },
"SyntheticDataGenerateRequestWrapper": { "SyntheticDataGenerationRequestWrapper": {
"type": "object", "type": "object",
"properties": { "properties": {
"request": { "request": {
@ -6063,34 +6063,34 @@
], ],
"tags": [ "tags": [
{ {
"name": "Memory" "name": "Inference"
}, },
{ {
"name": "Telemetry" "name": "Datasets"
},
{
"name": "PostTraining"
},
{
"name": "BatchInference"
},
{
"name": "AgenticSystem"
},
{
"name": "RewardScoring"
},
{
"name": "SyntheticDataGeneration"
}, },
{ {
"name": "Evaluations" "name": "Evaluations"
}, },
{ {
"name": "Inference" "name": "Memory"
}, },
{ {
"name": "Datasets" "name": "SyntheticDataGeneration"
},
{
"name": "PostTraining"
},
{
"name": "RewardScoring"
},
{
"name": "BatchInference"
},
{
"name": "Telemetry"
},
{
"name": "AgenticSystem"
}, },
{ {
"name": "BatchChatCompletionRequest", "name": "BatchChatCompletionRequest",
@ -6293,8 +6293,8 @@
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/Attachment\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/Attachment\" />"
}, },
{ {
"name": "CreateAgenticSystemTurnRequestWrapper", "name": "AgenticSystemTurnCreateRequestWrapper",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/CreateAgenticSystemTurnRequestWrapper\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/AgenticSystemTurnCreateRequestWrapper\" />"
}, },
{ {
"name": "AgenticSystemTurnResponseEvent", "name": "AgenticSystemTurnResponseEvent",
@ -6509,8 +6509,8 @@
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/LogSearchRequest\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/LogSearchRequest\" />"
}, },
{ {
"name": "GetLogsRequestWrapper", "name": "LogSearchRequestWrapper",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/GetLogsRequestWrapper\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/LogSearchRequestWrapper\" />"
}, },
{ {
"name": "Log", "name": "Log",
@ -6585,8 +6585,8 @@
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/TrainingConfig\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/TrainingConfig\" />"
}, },
{ {
"name": "PreferenceOptimizeRequestWrapper", "name": "PostTrainingRLHFRequestWrapper",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/PreferenceOptimizeRequestWrapper\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/PostTrainingRLHFRequestWrapper\" />"
}, },
{ {
"name": "QueryDocumentsRequest", "name": "QueryDocumentsRequest",
@ -6605,8 +6605,8 @@
"description": "Request to score a reward function. A list of prompts and a list of responses per prompt.\n\n<SchemaDefinition schemaRef=\"#/components/schemas/RewardScoringRequest\" />" "description": "Request to score a reward function. A list of prompts and a list of responses per prompt.\n\n<SchemaDefinition schemaRef=\"#/components/schemas/RewardScoringRequest\" />"
}, },
{ {
"name": "RewardScoreRequestWrapper", "name": "RewardScoringRequestWrapper",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/RewardScoreRequestWrapper\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/RewardScoringRequestWrapper\" />"
}, },
{ {
"name": "RewardScoringResponse", "name": "RewardScoringResponse",
@ -6641,16 +6641,16 @@
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/QLoraFinetuningConfig\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/QLoraFinetuningConfig\" />"
}, },
{ {
"name": "SupervisedFineTuneRequestWrapper", "name": "PostTrainingSFTRequestWrapper",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/SupervisedFineTuneRequestWrapper\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/PostTrainingSFTRequestWrapper\" />"
}, },
{ {
"name": "SyntheticDataGenerationRequest", "name": "SyntheticDataGenerationRequest",
"description": "Request to generate synthetic data. A small batch of prompts and a filtering function\n\n<SchemaDefinition schemaRef=\"#/components/schemas/SyntheticDataGenerationRequest\" />" "description": "Request to generate synthetic data. A small batch of prompts and a filtering function\n\n<SchemaDefinition schemaRef=\"#/components/schemas/SyntheticDataGenerationRequest\" />"
}, },
{ {
"name": "SyntheticDataGenerateRequestWrapper", "name": "SyntheticDataGenerationRequestWrapper",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/SyntheticDataGenerateRequestWrapper\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/SyntheticDataGenerationRequestWrapper\" />"
}, },
{ {
"name": "SyntheticDataGenerationResponse", "name": "SyntheticDataGenerationResponse",
@ -6709,6 +6709,7 @@
"AgenticSystemSessionCreateResponse", "AgenticSystemSessionCreateResponse",
"AgenticSystemStepResponse", "AgenticSystemStepResponse",
"AgenticSystemTurnCreateRequest", "AgenticSystemTurnCreateRequest",
"AgenticSystemTurnCreateRequestWrapper",
"AgenticSystemTurnResponseEvent", "AgenticSystemTurnResponseEvent",
"AgenticSystemTurnResponseStepCompletePayload", "AgenticSystemTurnResponseStepCompletePayload",
"AgenticSystemTurnResponseStepProgressPayload", "AgenticSystemTurnResponseStepProgressPayload",
@ -6743,7 +6744,6 @@
"CompletionResponseStreamChunk", "CompletionResponseStreamChunk",
"CreateAgenticSystemRequest", "CreateAgenticSystemRequest",
"CreateAgenticSystemSessionRequest", "CreateAgenticSystemSessionRequest",
"CreateAgenticSystemTurnRequestWrapper",
"CreateDatasetRequest", "CreateDatasetRequest",
"CreateDatasetRequestWrapper", "CreateDatasetRequestWrapper",
"CreateExperimentRequest", "CreateExperimentRequest",
@ -6777,7 +6777,6 @@
"FunctionCallToolDefinition", "FunctionCallToolDefinition",
"GetAgenticSystemSessionRequest", "GetAgenticSystemSessionRequest",
"GetDocumentsRequest", "GetDocumentsRequest",
"GetLogsRequestWrapper",
"InferenceStep", "InferenceStep",
"InsertDocumentsRequest", "InsertDocumentsRequest",
"ListArtifactsRequest", "ListArtifactsRequest",
@ -6787,6 +6786,7 @@
"LogMetricsRequest", "LogMetricsRequest",
"LogMetricsRequestWrapper", "LogMetricsRequestWrapper",
"LogSearchRequest", "LogSearchRequest",
"LogSearchRequestWrapper",
"LoraFinetuningConfig", "LoraFinetuningConfig",
"MemoryBank", "MemoryBank",
"MemoryBankDocument", "MemoryBankDocument",
@ -6801,16 +6801,17 @@
"PostTrainingJobStatus", "PostTrainingJobStatus",
"PostTrainingJobStatusResponse", "PostTrainingJobStatusResponse",
"PostTrainingRLHFRequest", "PostTrainingRLHFRequest",
"PostTrainingRLHFRequestWrapper",
"PostTrainingSFTRequest", "PostTrainingSFTRequest",
"PreferenceOptimizeRequestWrapper", "PostTrainingSFTRequestWrapper",
"QLoraFinetuningConfig", "QLoraFinetuningConfig",
"QueryDocumentsRequest", "QueryDocumentsRequest",
"QueryDocumentsResponse", "QueryDocumentsResponse",
"RLHFAlgorithm", "RLHFAlgorithm",
"RestAPIExecutionConfig", "RestAPIExecutionConfig",
"RestAPIMethod", "RestAPIMethod",
"RewardScoreRequestWrapper",
"RewardScoringRequest", "RewardScoringRequest",
"RewardScoringRequestWrapper",
"RewardScoringResponse", "RewardScoringResponse",
"Run", "Run",
"SamplingParams", "SamplingParams",
@ -6822,9 +6823,8 @@
"ShieldDefinition", "ShieldDefinition",
"ShieldResponse", "ShieldResponse",
"StopReason", "StopReason",
"SupervisedFineTuneRequestWrapper",
"SyntheticDataGenerateRequestWrapper",
"SyntheticDataGenerationRequest", "SyntheticDataGenerationRequest",
"SyntheticDataGenerationRequestWrapper",
"SyntheticDataGenerationResponse", "SyntheticDataGenerationResponse",
"SystemMessage", "SystemMessage",
"TokenLogProbs", "TokenLogProbs",

View file

@ -346,6 +346,14 @@ components:
- session_id - session_id
- messages - messages
type: object type: object
AgenticSystemTurnCreateRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/AgenticSystemTurnCreateRequest'
required:
- request
type: object
AgenticSystemTurnResponseEvent: AgenticSystemTurnResponseEvent:
additionalProperties: false additionalProperties: false
properties: properties:
@ -870,14 +878,6 @@ components:
- agent_id - agent_id
- session_name - session_name
type: object type: object
CreateAgenticSystemTurnRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/AgenticSystemTurnCreateRequest'
required:
- request
type: object
CreateDatasetRequest: CreateDatasetRequest:
additionalProperties: false additionalProperties: false
properties: properties:
@ -1373,14 +1373,6 @@ components:
required: required:
- document_ids - document_ids
type: object type: object
GetLogsRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/LogSearchRequest'
required:
- request
type: object
InferenceStep: InferenceStep:
additionalProperties: false additionalProperties: false
properties: properties:
@ -1513,6 +1505,14 @@ components:
required: required:
- query - query
type: object type: object
LogSearchRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/LogSearchRequest'
required:
- request
type: object
LoraFinetuningConfig: LoraFinetuningConfig:
additionalProperties: false additionalProperties: false
properties: properties:
@ -1853,6 +1853,14 @@ components:
- logger_config - logger_config
title: Request to finetune a model. title: Request to finetune a model.
type: object type: object
PostTrainingRLHFRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/PostTrainingRLHFRequest'
required:
- request
type: object
PostTrainingSFTRequest: PostTrainingSFTRequest:
additionalProperties: false additionalProperties: false
properties: properties:
@ -1908,11 +1916,11 @@ components:
- logger_config - logger_config
title: Request to finetune a model. title: Request to finetune a model.
type: object type: object
PreferenceOptimizeRequestWrapper: PostTrainingSFTRequestWrapper:
additionalProperties: false additionalProperties: false
properties: properties:
request: request:
$ref: '#/components/schemas/PostTrainingRLHFRequest' $ref: '#/components/schemas/PostTrainingSFTRequest'
required: required:
- request - request
type: object type: object
@ -2028,14 +2036,6 @@ components:
- PUT - PUT
- DELETE - DELETE
type: string type: string
RewardScoreRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/RewardScoringRequest'
required:
- request
type: object
RewardScoringRequest: RewardScoringRequest:
additionalProperties: false additionalProperties: false
properties: properties:
@ -2051,6 +2051,14 @@ components:
title: Request to score a reward function. A list of prompts and a list of responses title: Request to score a reward function. A list of prompts and a list of responses
per prompt. per prompt.
type: object type: object
RewardScoringRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/RewardScoringRequest'
required:
- request
type: object
RewardScoringResponse: RewardScoringResponse:
additionalProperties: false additionalProperties: false
properties: properties:
@ -2244,22 +2252,6 @@ components:
- end_of_message - end_of_message
- out_of_tokens - out_of_tokens
type: string type: string
SupervisedFineTuneRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/PostTrainingSFTRequest'
required:
- request
type: object
SyntheticDataGenerateRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/SyntheticDataGenerationRequest'
required:
- request
type: object
SyntheticDataGenerationRequest: SyntheticDataGenerationRequest:
additionalProperties: false additionalProperties: false
properties: properties:
@ -2289,6 +2281,14 @@ components:
title: Request to generate synthetic data. A small batch of prompts and a filtering title: Request to generate synthetic data. A small batch of prompts and a filtering
function function
type: object type: object
SyntheticDataGenerationRequestWrapper:
additionalProperties: false
properties:
request:
$ref: '#/components/schemas/SyntheticDataGenerationRequest'
required:
- request
type: object
SyntheticDataGenerationResponse: SyntheticDataGenerationResponse:
additionalProperties: false additionalProperties: false
properties: properties:
@ -2777,7 +2777,7 @@ info:
description: "This is the specification of the llama stack that provides\n \ description: "This is the specification of the llama stack that provides\n \
\ a set of endpoints and their corresponding interfaces that are tailored\ \ a set of endpoints and their corresponding interfaces that are tailored\
\ to\n best leverage Llama Models. The specification is still in\ \ to\n best leverage Llama Models. The specification is still in\
\ draft and subject to change.\n Generated at 2024-09-09 11:14:56.584501" \ draft and subject to change.\n Generated at 2024-09-09 11:19:39.855375"
title: '[DRAFT] Llama Stack Specification' title: '[DRAFT] Llama Stack Specification'
version: 0.0.1 version: 0.0.1
jsonSchemaDialect: https://json-schema.org/draft/2020-12/schema jsonSchemaDialect: https://json-schema.org/draft/2020-12/schema
@ -2909,7 +2909,7 @@ paths:
content: content:
application/json: application/json:
schema: schema:
$ref: '#/components/schemas/CreateAgenticSystemTurnRequestWrapper' $ref: '#/components/schemas/AgenticSystemTurnCreateRequestWrapper'
required: true required: true
responses: responses:
'200': '200':
@ -3351,7 +3351,7 @@ paths:
content: content:
application/json: application/json:
schema: schema:
$ref: '#/components/schemas/GetLogsRequestWrapper' $ref: '#/components/schemas/LogSearchRequestWrapper'
required: true required: true
responses: responses:
'200': '200':
@ -3610,7 +3610,7 @@ paths:
content: content:
application/json: application/json:
schema: schema:
$ref: '#/components/schemas/PreferenceOptimizeRequestWrapper' $ref: '#/components/schemas/PostTrainingRLHFRequestWrapper'
required: true required: true
responses: responses:
'200': '200':
@ -3628,7 +3628,7 @@ paths:
content: content:
application/json: application/json:
schema: schema:
$ref: '#/components/schemas/SupervisedFineTuneRequestWrapper' $ref: '#/components/schemas/PostTrainingSFTRequestWrapper'
required: true required: true
responses: responses:
'200': '200':
@ -3646,7 +3646,7 @@ paths:
content: content:
application/json: application/json:
schema: schema:
$ref: '#/components/schemas/RewardScoreRequestWrapper' $ref: '#/components/schemas/RewardScoringRequestWrapper'
required: true required: true
responses: responses:
'200': '200':
@ -3713,7 +3713,7 @@ paths:
content: content:
application/json: application/json:
schema: schema:
$ref: '#/components/schemas/SyntheticDataGenerateRequestWrapper' $ref: '#/components/schemas/SyntheticDataGenerationRequestWrapper'
required: true required: true
responses: responses:
'200': '200':
@ -3729,16 +3729,16 @@ security:
servers: servers:
- url: http://any-hosted-llama-stack.com - url: http://any-hosted-llama-stack.com
tags: tags:
- name: Memory
- name: Telemetry
- name: PostTraining
- name: BatchInference
- name: AgenticSystem
- name: RewardScoring
- name: SyntheticDataGeneration
- name: Evaluations
- name: Inference - name: Inference
- name: Datasets - name: Datasets
- name: Evaluations
- name: Memory
- name: SyntheticDataGeneration
- name: PostTraining
- name: RewardScoring
- name: BatchInference
- name: Telemetry
- name: AgenticSystem
- description: <SchemaDefinition schemaRef="#/components/schemas/BatchChatCompletionRequest" - description: <SchemaDefinition schemaRef="#/components/schemas/BatchChatCompletionRequest"
/> />
name: BatchChatCompletionRequest name: BatchChatCompletionRequest
@ -3891,9 +3891,9 @@ tags:
name: AgenticSystemTurnCreateRequest name: AgenticSystemTurnCreateRequest
- description: <SchemaDefinition schemaRef="#/components/schemas/Attachment" /> - description: <SchemaDefinition schemaRef="#/components/schemas/Attachment" />
name: Attachment name: Attachment
- description: <SchemaDefinition schemaRef="#/components/schemas/CreateAgenticSystemTurnRequestWrapper" - description: <SchemaDefinition schemaRef="#/components/schemas/AgenticSystemTurnCreateRequestWrapper"
/> />
name: CreateAgenticSystemTurnRequestWrapper name: AgenticSystemTurnCreateRequestWrapper
- description: 'Streamed agent execution response. - description: 'Streamed agent execution response.
@ -4068,9 +4068,9 @@ tags:
- description: <SchemaDefinition schemaRef="#/components/schemas/LogSearchRequest" - description: <SchemaDefinition schemaRef="#/components/schemas/LogSearchRequest"
/> />
name: LogSearchRequest name: LogSearchRequest
- description: <SchemaDefinition schemaRef="#/components/schemas/GetLogsRequestWrapper" - description: <SchemaDefinition schemaRef="#/components/schemas/LogSearchRequestWrapper"
/> />
name: GetLogsRequestWrapper name: LogSearchRequestWrapper
- description: <SchemaDefinition schemaRef="#/components/schemas/Log" /> - description: <SchemaDefinition schemaRef="#/components/schemas/Log" />
name: Log name: Log
- description: <SchemaDefinition schemaRef="#/components/schemas/Metric" /> - description: <SchemaDefinition schemaRef="#/components/schemas/Metric" />
@ -4131,9 +4131,9 @@ tags:
name: RLHFAlgorithm name: RLHFAlgorithm
- description: <SchemaDefinition schemaRef="#/components/schemas/TrainingConfig" /> - description: <SchemaDefinition schemaRef="#/components/schemas/TrainingConfig" />
name: TrainingConfig name: TrainingConfig
- description: <SchemaDefinition schemaRef="#/components/schemas/PreferenceOptimizeRequestWrapper" - description: <SchemaDefinition schemaRef="#/components/schemas/PostTrainingRLHFRequestWrapper"
/> />
name: PreferenceOptimizeRequestWrapper name: PostTrainingRLHFRequestWrapper
- description: <SchemaDefinition schemaRef="#/components/schemas/QueryDocumentsRequest" - description: <SchemaDefinition schemaRef="#/components/schemas/QueryDocumentsRequest"
/> />
name: QueryDocumentsRequest name: QueryDocumentsRequest
@ -4149,9 +4149,9 @@ tags:
<SchemaDefinition schemaRef="#/components/schemas/RewardScoringRequest" />' <SchemaDefinition schemaRef="#/components/schemas/RewardScoringRequest" />'
name: RewardScoringRequest name: RewardScoringRequest
- description: <SchemaDefinition schemaRef="#/components/schemas/RewardScoreRequestWrapper" - description: <SchemaDefinition schemaRef="#/components/schemas/RewardScoringRequestWrapper"
/> />
name: RewardScoreRequestWrapper name: RewardScoringRequestWrapper
- description: 'Response from the reward scoring. Batch of (prompt, response, score) - description: 'Response from the reward scoring. Batch of (prompt, response, score)
tuples that pass the threshold. tuples that pass the threshold.
@ -4180,9 +4180,9 @@ tags:
- description: <SchemaDefinition schemaRef="#/components/schemas/QLoraFinetuningConfig" - description: <SchemaDefinition schemaRef="#/components/schemas/QLoraFinetuningConfig"
/> />
name: QLoraFinetuningConfig name: QLoraFinetuningConfig
- description: <SchemaDefinition schemaRef="#/components/schemas/SupervisedFineTuneRequestWrapper" - description: <SchemaDefinition schemaRef="#/components/schemas/PostTrainingSFTRequestWrapper"
/> />
name: SupervisedFineTuneRequestWrapper name: PostTrainingSFTRequestWrapper
- description: 'Request to generate synthetic data. A small batch of prompts and a - description: 'Request to generate synthetic data. A small batch of prompts and a
filtering function filtering function
@ -4190,9 +4190,9 @@ tags:
<SchemaDefinition schemaRef="#/components/schemas/SyntheticDataGenerationRequest" <SchemaDefinition schemaRef="#/components/schemas/SyntheticDataGenerationRequest"
/>' />'
name: SyntheticDataGenerationRequest name: SyntheticDataGenerationRequest
- description: <SchemaDefinition schemaRef="#/components/schemas/SyntheticDataGenerateRequestWrapper" - description: <SchemaDefinition schemaRef="#/components/schemas/SyntheticDataGenerationRequestWrapper"
/> />
name: SyntheticDataGenerateRequestWrapper name: SyntheticDataGenerationRequestWrapper
- description: 'Response from the synthetic data generation. Batch of (prompt, response, - description: 'Response from the synthetic data generation. Batch of (prompt, response,
score) tuples that pass the threshold. score) tuples that pass the threshold.
@ -4241,6 +4241,7 @@ x-tagGroups:
- AgenticSystemSessionCreateResponse - AgenticSystemSessionCreateResponse
- AgenticSystemStepResponse - AgenticSystemStepResponse
- AgenticSystemTurnCreateRequest - AgenticSystemTurnCreateRequest
- AgenticSystemTurnCreateRequestWrapper
- AgenticSystemTurnResponseEvent - AgenticSystemTurnResponseEvent
- AgenticSystemTurnResponseStepCompletePayload - AgenticSystemTurnResponseStepCompletePayload
- AgenticSystemTurnResponseStepProgressPayload - AgenticSystemTurnResponseStepProgressPayload
@ -4275,7 +4276,6 @@ x-tagGroups:
- CompletionResponseStreamChunk - CompletionResponseStreamChunk
- CreateAgenticSystemRequest - CreateAgenticSystemRequest
- CreateAgenticSystemSessionRequest - CreateAgenticSystemSessionRequest
- CreateAgenticSystemTurnRequestWrapper
- CreateDatasetRequest - CreateDatasetRequest
- CreateDatasetRequestWrapper - CreateDatasetRequestWrapper
- CreateExperimentRequest - CreateExperimentRequest
@ -4309,7 +4309,6 @@ x-tagGroups:
- FunctionCallToolDefinition - FunctionCallToolDefinition
- GetAgenticSystemSessionRequest - GetAgenticSystemSessionRequest
- GetDocumentsRequest - GetDocumentsRequest
- GetLogsRequestWrapper
- InferenceStep - InferenceStep
- InsertDocumentsRequest - InsertDocumentsRequest
- ListArtifactsRequest - ListArtifactsRequest
@ -4319,6 +4318,7 @@ x-tagGroups:
- LogMetricsRequest - LogMetricsRequest
- LogMetricsRequestWrapper - LogMetricsRequestWrapper
- LogSearchRequest - LogSearchRequest
- LogSearchRequestWrapper
- LoraFinetuningConfig - LoraFinetuningConfig
- MemoryBank - MemoryBank
- MemoryBankDocument - MemoryBankDocument
@ -4333,16 +4333,17 @@ x-tagGroups:
- PostTrainingJobStatus - PostTrainingJobStatus
- PostTrainingJobStatusResponse - PostTrainingJobStatusResponse
- PostTrainingRLHFRequest - PostTrainingRLHFRequest
- PostTrainingRLHFRequestWrapper
- PostTrainingSFTRequest - PostTrainingSFTRequest
- PreferenceOptimizeRequestWrapper - PostTrainingSFTRequestWrapper
- QLoraFinetuningConfig - QLoraFinetuningConfig
- QueryDocumentsRequest - QueryDocumentsRequest
- QueryDocumentsResponse - QueryDocumentsResponse
- RLHFAlgorithm - RLHFAlgorithm
- RestAPIExecutionConfig - RestAPIExecutionConfig
- RestAPIMethod - RestAPIMethod
- RewardScoreRequestWrapper
- RewardScoringRequest - RewardScoringRequest
- RewardScoringRequestWrapper
- RewardScoringResponse - RewardScoringResponse
- Run - Run
- SamplingParams - SamplingParams
@ -4354,9 +4355,8 @@ x-tagGroups:
- ShieldDefinition - ShieldDefinition
- ShieldResponse - ShieldResponse
- StopReason - StopReason
- SupervisedFineTuneRequestWrapper
- SyntheticDataGenerateRequestWrapper
- SyntheticDataGenerationRequest - SyntheticDataGenerationRequest
- SyntheticDataGenerationRequestWrapper
- SyntheticDataGenerationResponse - SyntheticDataGenerationResponse
- SystemMessage - SystemMessage
- TokenLogProbs - TokenLogProbs

View file

@ -473,8 +473,7 @@ class Generator:
if len(op.request_params) == 1 and "Request" in first[1].__name__: if len(op.request_params) == 1 and "Request" in first[1].__name__:
# TODO(ashwin): Undo the "Request" hack and this entire block eventually # TODO(ashwin): Undo the "Request" hack and this entire block eventually
op_name = "".join(word.capitalize() for word in op.name.split("_")) request_name = first[1].__name__ + "Wrapper"
request_name = f"{op_name}RequestWrapper"
request_type = make_dataclass(request_name, op.request_params) request_type = make_dataclass(request_name, op.request_params)
else: else:
op_name = "".join(word.capitalize() for word in op.name.split("_")) op_name = "".join(word.capitalize() for word in op.name.split("_"))