small rename

This commit is contained in:
Ashwin Bharambe 2024-07-11 00:07:19 -07:00
parent 631328f556
commit 86c2993296
3 changed files with 39 additions and 46 deletions

View file

@ -302,9 +302,7 @@ class KPromptGenerations:
@json_schema_type @json_schema_type
@dataclass @dataclass
class MessageScore: class ScoredMessage:
"""A single message and its score."""
message: Message message: Message
score: float score: float
@ -313,7 +311,7 @@ class MessageScore:
@dataclass @dataclass
class KScoredPromptGenerations: class KScoredPromptGenerations:
prompt: Message prompt: Message
k_scored_generations: List[MessageScore] k_scored_generations: List[ScoredMessage]
@json_schema_type @json_schema_type

View file

@ -2627,7 +2627,7 @@
"k_scored_generations": { "k_scored_generations": {
"type": "array", "type": "array",
"items": { "items": {
"$ref": "#/components/schemas/MessageScore" "$ref": "#/components/schemas/ScoredMessage"
} }
} }
}, },
@ -2637,7 +2637,7 @@
"k_scored_generations" "k_scored_generations"
] ]
}, },
"MessageScore": { "ScoredMessage": {
"type": "object", "type": "object",
"properties": { "properties": {
"message": { "message": {
@ -2651,8 +2651,7 @@
"required": [ "required": [
"message", "message",
"score" "score"
], ]
"title": "A single message and its score."
}, },
"SyntheticDataGenerationResponse": { "SyntheticDataGenerationResponse": {
"type": "object", "type": "object",
@ -3150,26 +3149,26 @@
} }
], ],
"tags": [ "tags": [
{
"name": "AgenticSystem"
},
{
"name": "RewardScoring"
},
{
"name": "Inference"
},
{ {
"name": "SyntheticDataGeneration" "name": "SyntheticDataGeneration"
}, },
{ {
"name": "Datasets" "name": "MemoryBanks"
},
{
"name": "AgenticSystem"
}, },
{ {
"name": "PostTraining" "name": "PostTraining"
}, },
{ {
"name": "MemoryBanks" "name": "Inference"
},
{
"name": "RewardScoring"
},
{
"name": "Datasets"
}, },
{ {
"name": "ShieldConfig", "name": "ShieldConfig",
@ -3280,8 +3279,8 @@
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/KScoredPromptGenerations\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/KScoredPromptGenerations\" />"
}, },
{ {
"name": "MessageScore", "name": "ScoredMessage",
"description": "A single message and its score.\n\n<SchemaDefinition schemaRef=\"#/components/schemas/MessageScore\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/ScoredMessage\" />"
}, },
{ {
"name": "SyntheticDataGenerationResponse", "name": "SyntheticDataGenerationResponse",
@ -3368,7 +3367,6 @@
"LoraFinetuningConfig", "LoraFinetuningConfig",
"MemoryBank", "MemoryBank",
"Message", "Message",
"MessageScore",
"OptimizerConfig", "OptimizerConfig",
"PostTrainingJobArtifactsResponse", "PostTrainingJobArtifactsResponse",
"PostTrainingJobLogStream", "PostTrainingJobLogStream",
@ -3378,6 +3376,7 @@
"QLoraFinetuningConfig", "QLoraFinetuningConfig",
"RewardScoringRequest", "RewardScoringRequest",
"RewardScoringResponse", "RewardScoringResponse",
"ScoredMessage",
"ShieldConfig", "ShieldConfig",
"SyntheticDataGenerationRequest", "SyntheticDataGenerationRequest",
"SyntheticDataGenerationResponse", "SyntheticDataGenerationResponse",

View file

@ -967,7 +967,7 @@ components:
properties: properties:
k_scored_generations: k_scored_generations:
items: items:
$ref: '#/components/schemas/MessageScore' $ref: '#/components/schemas/ScoredMessage'
type: array type: array
prompt: prompt:
$ref: '#/components/schemas/Message' $ref: '#/components/schemas/Message'
@ -1075,18 +1075,6 @@ components:
- tool_calls - tool_calls
- tool_responses - tool_responses
type: object type: object
MessageScore:
additionalProperties: false
properties:
message:
$ref: '#/components/schemas/Message'
score:
type: number
required:
- message
- score
title: A single message and its score.
type: object
OptimizerConfig: OptimizerConfig:
additionalProperties: false additionalProperties: false
properties: properties:
@ -1374,6 +1362,17 @@ components:
title: Response from the reward scoring. Batch of (prompt, response, score) title: Response from the reward scoring. Batch of (prompt, response, score)
tuples that pass the threshold. tuples that pass the threshold.
type: object type: object
ScoredMessage:
additionalProperties: false
properties:
message:
$ref: '#/components/schemas/Message'
score:
type: number
required:
- message
- score
type: object
ShieldConfig: ShieldConfig:
additionalProperties: false additionalProperties: false
properties: properties:
@ -1917,13 +1916,13 @@ security:
servers: servers:
- url: http://llama.meta.com - url: http://llama.meta.com
tags: tags:
- name: AgenticSystem
- name: RewardScoring
- name: Inference
- name: SyntheticDataGeneration - name: SyntheticDataGeneration
- name: Datasets
- name: PostTraining
- name: MemoryBanks - name: MemoryBanks
- name: AgenticSystem
- name: PostTraining
- name: Inference
- name: RewardScoring
- name: Datasets
- description: <SchemaDefinition schemaRef="#/components/schemas/ShieldConfig" /> - description: <SchemaDefinition schemaRef="#/components/schemas/ShieldConfig" />
name: ShieldConfig name: ShieldConfig
- description: <SchemaDefinition schemaRef="#/components/schemas/AgenticSystemCreateRequest" - description: <SchemaDefinition schemaRef="#/components/schemas/AgenticSystemCreateRequest"
@ -2038,11 +2037,8 @@ tags:
- description: <SchemaDefinition schemaRef="#/components/schemas/KScoredPromptGenerations" - description: <SchemaDefinition schemaRef="#/components/schemas/KScoredPromptGenerations"
/> />
name: KScoredPromptGenerations name: KScoredPromptGenerations
- description: 'A single message and its score. - description: <SchemaDefinition schemaRef="#/components/schemas/ScoredMessage" />
name: ScoredMessage
<SchemaDefinition schemaRef="#/components/schemas/MessageScore" />'
name: MessageScore
- 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.
@ -2125,7 +2121,6 @@ x-tagGroups:
- LoraFinetuningConfig - LoraFinetuningConfig
- MemoryBank - MemoryBank
- Message - Message
- MessageScore
- OptimizerConfig - OptimizerConfig
- PostTrainingJobArtifactsResponse - PostTrainingJobArtifactsResponse
- PostTrainingJobLogStream - PostTrainingJobLogStream
@ -2135,6 +2130,7 @@ x-tagGroups:
- QLoraFinetuningConfig - QLoraFinetuningConfig
- RewardScoringRequest - RewardScoringRequest
- RewardScoringResponse - RewardScoringResponse
- ScoredMessage
- ShieldConfig - ShieldConfig
- SyntheticDataGenerationRequest - SyntheticDataGenerationRequest
- SyntheticDataGenerationResponse - SyntheticDataGenerationResponse