phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 02:53:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add json_schema_type to ParamType deps (#808)

Browse source 
# What does this PR do?

Add missing json_schema_type annotation to ParamType deps
This commit is contained in:
Dinesh Yeduguru 2025-01-17 11:02:25 -08:00 • committed by GitHub
parent c2a072911d
commit 53b5f6b24a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 355 additions and 211 deletions
Download patch file Download diff file Expand all files Collapse all files

314
docs/resources/llama-stack-spec.html
View file

@ -6067,6 +6067,76 @@
"step" "step"
] ]
}, },
"AgentTurnInputType": {
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "agent_turn_input",
"default": "agent_turn_input"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
"ArrayType": {
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "array",
"default": "array"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
"BooleanType": {
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "boolean",
"default": "boolean"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
"ChatCompletionInputType": {
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "chat_completion_input",
"default": "chat_completion_input"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
"CompletionInputType": {
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "completion_input",
"default": "completion_input"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
"Dataset": { "Dataset": {
"type": "object", "type": "object",
"properties": { "properties": {
@ -6130,79 +6200,7 @@
"metadata" "metadata"
] ]
}, },
"ParamType": { "JsonType": {
"oneOf": [
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "string",
"default": "string"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "number",
"default": "number"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "boolean",
"default": "boolean"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "array",
"default": "array"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "object",
"default": "object"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
{
"type": "object", "type": "object",
"properties": { "properties": {
"type": { "type": {
@ -6216,7 +6214,83 @@
"type" "type"
] ]
}, },
"NumberType": {
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "number",
"default": "number"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
"ObjectType": {
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "object",
"default": "object"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
"ParamType": {
"oneOf": [
{ {
"$ref": "#/components/schemas/StringType"
},
{
"$ref": "#/components/schemas/NumberType"
},
{
"$ref": "#/components/schemas/BooleanType"
},
{
"$ref": "#/components/schemas/ArrayType"
},
{
"$ref": "#/components/schemas/ObjectType"
},
{
"$ref": "#/components/schemas/JsonType"
},
{
"$ref": "#/components/schemas/UnionType"
},
{
"$ref": "#/components/schemas/ChatCompletionInputType"
},
{
"$ref": "#/components/schemas/CompletionInputType"
},
{
"$ref": "#/components/schemas/AgentTurnInputType"
}
]
},
"StringType": {
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "string",
"default": "string"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
"UnionType": {
"type": "object", "type": "object",
"properties": { "properties": {
"type": { "type": {
@ -6230,50 +6304,6 @@
"type" "type"
] ]
}, },
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "chat_completion_input",
"default": "chat_completion_input"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "completion_input",
"default": "completion_input"
}
},
"additionalProperties": false,
"required": [
"type"
]
},
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "agent_turn_input",
"default": "agent_turn_input"
}
},
"additionalProperties": false,
"required": [
"type"
]
}
]
},
"EvalTask": { "EvalTask": {
"type": "object", "type": "object",
"properties": { "properties": {
@ -8922,6 +8952,10 @@
"name": "AgentTool", "name": "AgentTool",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/AgentTool\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/AgentTool\" />"
}, },
{
"name": "AgentTurnInputType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/AgentTurnInputType\" />"
},
{ {
"name": "AgentTurnResponseEvent", "name": "AgentTurnResponseEvent",
"description": "Streamed agent execution response.\n\n<SchemaDefinition schemaRef=\"#/components/schemas/AgentTurnResponseEvent\" />" "description": "Streamed agent execution response.\n\n<SchemaDefinition schemaRef=\"#/components/schemas/AgentTurnResponseEvent\" />"
@ -8965,6 +8999,10 @@
"name": "AppendRowsRequest", "name": "AppendRowsRequest",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/AppendRowsRequest\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/AppendRowsRequest\" />"
}, },
{
"name": "ArrayType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/ArrayType\" />"
},
{ {
"name": "BasicScoringFnParams", "name": "BasicScoringFnParams",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/BasicScoringFnParams\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/BasicScoringFnParams\" />"
@ -8992,6 +9030,10 @@
"name": "BenchmarkEvalTaskConfig", "name": "BenchmarkEvalTaskConfig",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/BenchmarkEvalTaskConfig\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/BenchmarkEvalTaskConfig\" />"
}, },
{
"name": "BooleanType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/BooleanType\" />"
},
{ {
"name": "BuiltinTool", "name": "BuiltinTool",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/BuiltinTool\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/BuiltinTool\" />"
@ -9000,6 +9042,10 @@
"name": "CancelTrainingJobRequest", "name": "CancelTrainingJobRequest",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/CancelTrainingJobRequest\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/CancelTrainingJobRequest\" />"
}, },
{
"name": "ChatCompletionInputType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/ChatCompletionInputType\" />"
},
{ {
"name": "ChatCompletionRequest", "name": "ChatCompletionRequest",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/ChatCompletionRequest\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/ChatCompletionRequest\" />"
@ -9024,6 +9070,10 @@
"name": "Checkpoint", "name": "Checkpoint",
"description": "Checkpoint created during training runs\n\n<SchemaDefinition schemaRef=\"#/components/schemas/Checkpoint\" />" "description": "Checkpoint created during training runs\n\n<SchemaDefinition schemaRef=\"#/components/schemas/Checkpoint\" />"
}, },
{
"name": "CompletionInputType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/CompletionInputType\" />"
},
{ {
"name": "CompletionMessage", "name": "CompletionMessage",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/CompletionMessage\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/CompletionMessage\" />"
@ -9166,6 +9216,10 @@
"name": "JobStatus", "name": "JobStatus",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/JobStatus\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/JobStatus\" />"
}, },
{
"name": "JsonType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/JsonType\" />"
},
{ {
"name": "KeyValueMemoryBank", "name": "KeyValueMemoryBank",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/KeyValueMemoryBank\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/KeyValueMemoryBank\" />"
@ -9283,6 +9337,14 @@
{ {
"name": "Models" "name": "Models"
}, },
{
"name": "NumberType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/NumberType\" />"
},
{
"name": "ObjectType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/ObjectType\" />"
},
{ {
"name": "OptimizerConfig", "name": "OptimizerConfig",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/OptimizerConfig\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/OptimizerConfig\" />"
@ -9490,6 +9552,10 @@
"name": "StopReason", "name": "StopReason",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/StopReason\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/StopReason\" />"
}, },
{
"name": "StringType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/StringType\" />"
},
{ {
"name": "StructuredLogEvent", "name": "StructuredLogEvent",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/StructuredLogEvent\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/StructuredLogEvent\" />"
@ -9622,6 +9688,10 @@
"name": "URL", "name": "URL",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/URL\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/URL\" />"
}, },
{
"name": "UnionType",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/UnionType\" />"
},
{ {
"name": "UnstructuredLogEvent", "name": "UnstructuredLogEvent",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/UnstructuredLogEvent\" />" "description": "<SchemaDefinition schemaRef=\"#/components/schemas/UnstructuredLogEvent\" />"
@ -9682,6 +9752,7 @@
"AgentSessionCreateResponse", "AgentSessionCreateResponse",
"AgentStepResponse", "AgentStepResponse",
"AgentTool", "AgentTool",
"AgentTurnInputType",
"AgentTurnResponseEvent", "AgentTurnResponseEvent",
"AgentTurnResponseStepCompletePayload", "AgentTurnResponseStepCompletePayload",
"AgentTurnResponseStepProgressPayload", "AgentTurnResponseStepProgressPayload",
@ -9692,20 +9763,24 @@
"AggregationFunctionType", "AggregationFunctionType",
"AppEvalTaskConfig", "AppEvalTaskConfig",
"AppendRowsRequest", "AppendRowsRequest",
"ArrayType",
"BasicScoringFnParams", "BasicScoringFnParams",
"BatchChatCompletionRequest", "BatchChatCompletionRequest",
"BatchChatCompletionResponse", "BatchChatCompletionResponse",
"BatchCompletionRequest", "BatchCompletionRequest",
"BatchCompletionResponse", "BatchCompletionResponse",
"BenchmarkEvalTaskConfig", "BenchmarkEvalTaskConfig",
"BooleanType",
"BuiltinTool", "BuiltinTool",
"CancelTrainingJobRequest", "CancelTrainingJobRequest",
"ChatCompletionInputType",
"ChatCompletionRequest", "ChatCompletionRequest",
"ChatCompletionResponse", "ChatCompletionResponse",
"ChatCompletionResponseEvent", "ChatCompletionResponseEvent",
"ChatCompletionResponseEventType", "ChatCompletionResponseEventType",
"ChatCompletionResponseStreamChunk", "ChatCompletionResponseStreamChunk",
"Checkpoint", "Checkpoint",
"CompletionInputType",
"CompletionMessage", "CompletionMessage",
"CompletionRequest", "CompletionRequest",
"CompletionResponse", "CompletionResponse",
@ -9737,6 +9812,7 @@
"InvokeToolRequest", "InvokeToolRequest",
"Job", "Job",
"JobStatus", "JobStatus",
"JsonType",
"KeyValueMemoryBank", "KeyValueMemoryBank",
"KeyValueMemoryBankParams", "KeyValueMemoryBankParams",
"KeywordMemoryBank", "KeywordMemoryBank",
@ -9764,6 +9840,8 @@
"Model", "Model",
"ModelCandidate", "ModelCandidate",
"ModelType", "ModelType",
"NumberType",
"ObjectType",
"OptimizerConfig", "OptimizerConfig",
"OptimizerType", "OptimizerType",
"PaginatedRowsResult", "PaginatedRowsResult",
@ -9812,6 +9890,7 @@
"SpanStatus", "SpanStatus",
"SpanWithStatus", "SpanWithStatus",
"StopReason", "StopReason",
"StringType",
"StructuredLogEvent", "StructuredLogEvent",
"SupervisedFineTuneRequest", "SupervisedFineTuneRequest",
"SyntheticDataGenerateRequest", "SyntheticDataGenerateRequest",
@ -9842,6 +9921,7 @@
"TrainingConfig", "TrainingConfig",
"Turn", "Turn",
"URL", "URL",
"UnionType",
"UnstructuredLogEvent", "UnstructuredLogEvent",
"UserMessage", "UserMessage",
"VectorMemoryBank", "VectorMemoryBank",

233
docs/resources/llama-stack-spec.yaml
View file

@ -105,6 +105,16 @@ components:
- name - name
- args - args
type: object type: object
AgentTurnInputType:
additionalProperties: false
properties:
type:
const: agent_turn_input
default: agent_turn_input
type: string
required:
- type
type: object
AgentTurnResponseEvent: AgentTurnResponseEvent:
additionalProperties: false additionalProperties: false
properties: properties:
@ -290,6 +300,16 @@ components:
- dataset_id - dataset_id
- rows - rows
type: object type: object
ArrayType:
additionalProperties: false
properties:
type:
const: array
default: array
type: string
required:
- type
type: object
BasicScoringFnParams: BasicScoringFnParams:
additionalProperties: false additionalProperties: false
properties: properties:
@ -395,6 +415,16 @@ components:
- type - type
- eval_candidate - eval_candidate
type: object type: object
BooleanType:
additionalProperties: false
properties:
type:
const: boolean
default: boolean
type: string
required:
- type
type: object
BuiltinTool: BuiltinTool:
enum: enum:
- brave_search - brave_search
@ -410,6 +440,16 @@ components:
required: required:
- job_uuid - job_uuid
type: object type: object
ChatCompletionInputType:
additionalProperties: false
properties:
type:
const: chat_completion_input
default: chat_completion_input
type: string
required:
- type
type: object
ChatCompletionRequest: ChatCompletionRequest:
additionalProperties: false additionalProperties: false
properties: properties:
@ -492,6 +532,16 @@ components:
type: object type: object
Checkpoint: Checkpoint:
description: Checkpoint created during training runs description: Checkpoint created during training runs
CompletionInputType:
additionalProperties: false
properties:
type:
const: completion_input
default: completion_input
type: string
required:
- type
type: object
CompletionMessage: CompletionMessage:
additionalProperties: false additionalProperties: false
properties: properties:
@ -1007,6 +1057,16 @@ components:
- failed - failed
- scheduled - scheduled
type: string type: string
JsonType:
additionalProperties: false
properties:
type:
const: json
default: json
type: string
required:
- type
type: object
KeyValueMemoryBank: KeyValueMemoryBank:
additionalProperties: false additionalProperties: false
properties: properties:
@ -1440,6 +1500,26 @@ components:
- llm - llm
- embedding - embedding
type: string type: string
NumberType:
additionalProperties: false
properties:
type:
const: number
default: number
type: string
required:
- type
type: object
ObjectType:
additionalProperties: false
properties:
type:
const: object
default: object
type: string
required:
- type
type: object
OptimizerConfig: OptimizerConfig:
additionalProperties: false additionalProperties: false
properties: properties:
@ -1488,96 +1568,16 @@ components:
type: object type: object
ParamType: ParamType:
oneOf: oneOf:
- additionalProperties: false - $ref: '#/components/schemas/StringType'
properties: - $ref: '#/components/schemas/NumberType'
type: - $ref: '#/components/schemas/BooleanType'
const: string - $ref: '#/components/schemas/ArrayType'
default: string - $ref: '#/components/schemas/ObjectType'
type: string - $ref: '#/components/schemas/JsonType'
required: - $ref: '#/components/schemas/UnionType'
- type - $ref: '#/components/schemas/ChatCompletionInputType'
type: object - $ref: '#/components/schemas/CompletionInputType'
- additionalProperties: false - $ref: '#/components/schemas/AgentTurnInputType'
properties:
type:
const: number
default: number
type: string
required:
- type
type: object
- additionalProperties: false
properties:
type:
const: boolean
default: boolean
type: string
required:
- type
type: object
- additionalProperties: false
properties:
type:
const: array
default: array
type: string
required:
- type
type: object
- additionalProperties: false
properties:
type:
const: object
default: object
type: string
required:
- type
type: object
- additionalProperties: false
properties:
type:
const: json
default: json
type: string
required:
- type
type: object
- additionalProperties: false
properties:
type:
const: union
default: union
type: string
required:
- type
type: object
- additionalProperties: false
properties:
type:
const: chat_completion_input
default: chat_completion_input
type: string
required:
- type
type: object
- additionalProperties: false
properties:
type:
const: completion_input
default: completion_input
type: string
required:
- type
type: object
- additionalProperties: false
properties:
type:
const: agent_turn_input
default: agent_turn_input
type: string
required:
- type
type: object
PostTrainingJob: PostTrainingJob:
additionalProperties: false additionalProperties: false
properties: properties:
@ -2479,6 +2479,16 @@ components:
- end_of_message - end_of_message
- out_of_tokens - out_of_tokens
type: string type: string
StringType:
additionalProperties: false
properties:
type:
const: string
default: string
type: string
required:
- type
type: object
StructuredLogEvent: StructuredLogEvent:
additionalProperties: false additionalProperties: false
properties: properties:
@ -3131,6 +3141,16 @@ components:
required: required:
- uri - uri
type: object type: object
UnionType:
additionalProperties: false
properties:
type:
const: union
default: union
type: string
required:
- type
type: object
UnstructuredLogEvent: UnstructuredLogEvent:
additionalProperties: false additionalProperties: false
properties: properties:
@ -5588,6 +5608,9 @@ tags:
name: AgentStepResponse name: AgentStepResponse
- description: <SchemaDefinition schemaRef="#/components/schemas/AgentTool" /> - description: <SchemaDefinition schemaRef="#/components/schemas/AgentTool" />
name: AgentTool name: AgentTool
- description: <SchemaDefinition schemaRef="#/components/schemas/AgentTurnInputType"
/>
name: AgentTurnInputType
- description: 'Streamed agent execution response. - description: 'Streamed agent execution response.
@ -5624,6 +5647,8 @@ tags:
- description: <SchemaDefinition schemaRef="#/components/schemas/AppendRowsRequest" - description: <SchemaDefinition schemaRef="#/components/schemas/AppendRowsRequest"
/> />
name: AppendRowsRequest name: AppendRowsRequest
- description: <SchemaDefinition schemaRef="#/components/schemas/ArrayType" />
name: ArrayType
- description: <SchemaDefinition schemaRef="#/components/schemas/BasicScoringFnParams" - description: <SchemaDefinition schemaRef="#/components/schemas/BasicScoringFnParams"
/> />
name: BasicScoringFnParams name: BasicScoringFnParams
@ -5643,11 +5668,16 @@ tags:
- description: <SchemaDefinition schemaRef="#/components/schemas/BenchmarkEvalTaskConfig" - description: <SchemaDefinition schemaRef="#/components/schemas/BenchmarkEvalTaskConfig"
/> />
name: BenchmarkEvalTaskConfig name: BenchmarkEvalTaskConfig
- description: <SchemaDefinition schemaRef="#/components/schemas/BooleanType" />
name: BooleanType
- description: <SchemaDefinition schemaRef="#/components/schemas/BuiltinTool" /> - description: <SchemaDefinition schemaRef="#/components/schemas/BuiltinTool" />
name: BuiltinTool name: BuiltinTool
- description: <SchemaDefinition schemaRef="#/components/schemas/CancelTrainingJobRequest" - description: <SchemaDefinition schemaRef="#/components/schemas/CancelTrainingJobRequest"
/> />
name: CancelTrainingJobRequest name: CancelTrainingJobRequest
- description: <SchemaDefinition schemaRef="#/components/schemas/ChatCompletionInputType"
/>
name: ChatCompletionInputType
- description: <SchemaDefinition schemaRef="#/components/schemas/ChatCompletionRequest" - description: <SchemaDefinition schemaRef="#/components/schemas/ChatCompletionRequest"
/> />
name: ChatCompletionRequest name: ChatCompletionRequest
@ -5676,6 +5706,9 @@ tags:
<SchemaDefinition schemaRef="#/components/schemas/Checkpoint" />' <SchemaDefinition schemaRef="#/components/schemas/Checkpoint" />'
name: Checkpoint name: Checkpoint
- description: <SchemaDefinition schemaRef="#/components/schemas/CompletionInputType"
/>
name: CompletionInputType
- description: <SchemaDefinition schemaRef="#/components/schemas/CompletionMessage" - description: <SchemaDefinition schemaRef="#/components/schemas/CompletionMessage"
/> />
name: CompletionMessage name: CompletionMessage
@ -5770,6 +5803,8 @@ tags:
name: Job name: Job
- description: <SchemaDefinition schemaRef="#/components/schemas/JobStatus" /> - description: <SchemaDefinition schemaRef="#/components/schemas/JobStatus" />
name: JobStatus name: JobStatus
- description: <SchemaDefinition schemaRef="#/components/schemas/JsonType" />
name: JsonType
- description: <SchemaDefinition schemaRef="#/components/schemas/KeyValueMemoryBank" - description: <SchemaDefinition schemaRef="#/components/schemas/KeyValueMemoryBank"
/> />
name: KeyValueMemoryBank name: KeyValueMemoryBank
@ -5847,6 +5882,10 @@ tags:
- description: <SchemaDefinition schemaRef="#/components/schemas/ModelType" /> - description: <SchemaDefinition schemaRef="#/components/schemas/ModelType" />
name: ModelType name: ModelType
- name: Models - name: Models
- description: <SchemaDefinition schemaRef="#/components/schemas/NumberType" />
name: NumberType
- description: <SchemaDefinition schemaRef="#/components/schemas/ObjectType" />
name: ObjectType
- description: <SchemaDefinition schemaRef="#/components/schemas/OptimizerConfig" - description: <SchemaDefinition schemaRef="#/components/schemas/OptimizerConfig"
/> />
name: OptimizerConfig name: OptimizerConfig
@ -5988,6 +6027,8 @@ tags:
name: SpanWithStatus name: SpanWithStatus
- description: <SchemaDefinition schemaRef="#/components/schemas/StopReason" /> - description: <SchemaDefinition schemaRef="#/components/schemas/StopReason" />
name: StopReason name: StopReason
- description: <SchemaDefinition schemaRef="#/components/schemas/StringType" />
name: StringType
- description: <SchemaDefinition schemaRef="#/components/schemas/StructuredLogEvent" - description: <SchemaDefinition schemaRef="#/components/schemas/StructuredLogEvent"
/> />
name: StructuredLogEvent name: StructuredLogEvent
@ -6080,6 +6121,8 @@ tags:
name: Turn name: Turn
- description: <SchemaDefinition schemaRef="#/components/schemas/URL" /> - description: <SchemaDefinition schemaRef="#/components/schemas/URL" />
name: URL name: URL
- description: <SchemaDefinition schemaRef="#/components/schemas/UnionType" />
name: UnionType
- description: <SchemaDefinition schemaRef="#/components/schemas/UnstructuredLogEvent" - description: <SchemaDefinition schemaRef="#/components/schemas/UnstructuredLogEvent"
/> />
name: UnstructuredLogEvent name: UnstructuredLogEvent
@ -6126,6 +6169,7 @@ x-tagGroups:
- AgentSessionCreateResponse - AgentSessionCreateResponse
- AgentStepResponse - AgentStepResponse
- AgentTool - AgentTool
- AgentTurnInputType
- AgentTurnResponseEvent - AgentTurnResponseEvent
- AgentTurnResponseStepCompletePayload - AgentTurnResponseStepCompletePayload
- AgentTurnResponseStepProgressPayload - AgentTurnResponseStepProgressPayload
@ -6136,20 +6180,24 @@ x-tagGroups:
- AggregationFunctionType - AggregationFunctionType
- AppEvalTaskConfig - AppEvalTaskConfig
- AppendRowsRequest - AppendRowsRequest
- ArrayType
- BasicScoringFnParams - BasicScoringFnParams
- BatchChatCompletionRequest - BatchChatCompletionRequest
- BatchChatCompletionResponse - BatchChatCompletionResponse
- BatchCompletionRequest - BatchCompletionRequest
- BatchCompletionResponse - BatchCompletionResponse
- BenchmarkEvalTaskConfig - BenchmarkEvalTaskConfig
- BooleanType
- BuiltinTool - BuiltinTool
- CancelTrainingJobRequest - CancelTrainingJobRequest
- ChatCompletionInputType
- ChatCompletionRequest - ChatCompletionRequest
- ChatCompletionResponse - ChatCompletionResponse
- ChatCompletionResponseEvent - ChatCompletionResponseEvent
- ChatCompletionResponseEventType - ChatCompletionResponseEventType
- ChatCompletionResponseStreamChunk - ChatCompletionResponseStreamChunk
- Checkpoint - Checkpoint
- CompletionInputType
- CompletionMessage - CompletionMessage
- CompletionRequest - CompletionRequest
- CompletionResponse - CompletionResponse
@ -6181,6 +6229,7 @@ x-tagGroups:
- InvokeToolRequest - InvokeToolRequest
- Job - Job
- JobStatus - JobStatus
- JsonType
- KeyValueMemoryBank - KeyValueMemoryBank
- KeyValueMemoryBankParams - KeyValueMemoryBankParams
- KeywordMemoryBank - KeywordMemoryBank
@ -6208,6 +6257,8 @@ x-tagGroups:
- Model - Model
- ModelCandidate - ModelCandidate
- ModelType - ModelType
- NumberType
- ObjectType
- OptimizerConfig - OptimizerConfig
- OptimizerType - OptimizerType
- PaginatedRowsResult - PaginatedRowsResult
@ -6256,6 +6307,7 @@ x-tagGroups:
- SpanStatus - SpanStatus
- SpanWithStatus - SpanWithStatus
- StopReason - StopReason
- StringType
- StructuredLogEvent - StructuredLogEvent
- SupervisedFineTuneRequest - SupervisedFineTuneRequest
- SyntheticDataGenerateRequest - SyntheticDataGenerateRequest
@ -6286,6 +6338,7 @@ x-tagGroups:
- TrainingConfig - TrainingConfig
- Turn - Turn
- URL - URL
- UnionType
- UnstructuredLogEvent - UnstructuredLogEvent
- UserMessage - UserMessage
- VectorMemoryBank - VectorMemoryBank

13
llama_stack/apis/common/type_system.py
View file

@ -6,54 +6,65 @@
from typing import Literal, Union from typing import Literal, Union
from llama_models.schema_utils import register_schema from llama_models.schema_utils import json_schema_type, register_schema
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
from typing_extensions import Annotated from typing_extensions import Annotated
@json_schema_type
class StringType(BaseModel): class StringType(BaseModel):
type: Literal["string"] = "string" type: Literal["string"] = "string"
@json_schema_type
class NumberType(BaseModel): class NumberType(BaseModel):
type: Literal["number"] = "number" type: Literal["number"] = "number"
@json_schema_type
class BooleanType(BaseModel): class BooleanType(BaseModel):
type: Literal["boolean"] = "boolean" type: Literal["boolean"] = "boolean"
@json_schema_type
class ArrayType(BaseModel): class ArrayType(BaseModel):
type: Literal["array"] = "array" type: Literal["array"] = "array"
@json_schema_type
class ObjectType(BaseModel): class ObjectType(BaseModel):
type: Literal["object"] = "object" type: Literal["object"] = "object"
@json_schema_type
class JsonType(BaseModel): class JsonType(BaseModel):
type: Literal["json"] = "json" type: Literal["json"] = "json"
@json_schema_type
class UnionType(BaseModel): class UnionType(BaseModel):
type: Literal["union"] = "union" type: Literal["union"] = "union"
@json_schema_type
class ChatCompletionInputType(BaseModel): class ChatCompletionInputType(BaseModel):
# expects List[Message] for messages # expects List[Message] for messages
type: Literal["chat_completion_input"] = "chat_completion_input" type: Literal["chat_completion_input"] = "chat_completion_input"
@json_schema_type
class CompletionInputType(BaseModel): class CompletionInputType(BaseModel):
# expects InterleavedTextMedia for content # expects InterleavedTextMedia for content
type: Literal["completion_input"] = "completion_input" type: Literal["completion_input"] = "completion_input"
@json_schema_type
class AgentTurnInputType(BaseModel): class AgentTurnInputType(BaseModel):
# expects List[Message] for messages (may also include attachments?) # expects List[Message] for messages (may also include attachments?)
type: Literal["agent_turn_input"] = "agent_turn_input" type: Literal["agent_turn_input"] = "agent_turn_input"
@json_schema_type
class DialogType(BaseModel): class DialogType(BaseModel):
# expects List[Message] for messages # expects List[Message] for messages
# this type semantically contains the output label whereas ChatCompletionInputType does not # this type semantically contains the output label whereas ChatCompletionInputType does not