Update spec since we changed some types for completion API

This commit is contained in:
Ashwin Bharambe 2024-10-18 20:48:28 -07:00
parent fedc11b726
commit 9fddbdfcdf
2 changed files with 61 additions and 54 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-10-10 15:29:56.831109" "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-10-18 20:48:17.730988"
}, },
"servers": [ "servers": [
{ {
@ -2830,8 +2830,11 @@
"CompletionResponse": { "CompletionResponse": {
"type": "object", "type": "object",
"properties": { "properties": {
"completion_message": { "content": {
"$ref": "#/components/schemas/CompletionMessage" "type": "string"
},
"stop_reason": {
"$ref": "#/components/schemas/StopReason"
}, },
"logprobs": { "logprobs": {
"type": "array", "type": "array",
@ -2842,7 +2845,8 @@
}, },
"additionalProperties": false, "additionalProperties": false,
"required": [ "required": [
"completion_message" "content",
"stop_reason"
], ],
"title": "Completion response." "title": "Completion response."
}, },
@ -6075,49 +6079,49 @@
], ],
"tags": [ "tags": [
{ {
"name": "Evaluations" "name": "Models"
},
{
"name": "Inspect"
}, },
{ {
"name": "RewardScoring" "name": "RewardScoring"
}, },
{ {
"name": "Datasets" "name": "MemoryBanks"
},
{
"name": "Models"
},
{
"name": "Telemetry"
},
{
"name": "PostTraining"
},
{
"name": "SyntheticDataGeneration"
},
{
"name": "BatchInference"
},
{
"name": "Inference"
},
{
"name": "Agents"
},
{
"name": "Memory"
},
{
"name": "Safety"
}, },
{ {
"name": "Shields" "name": "Shields"
}, },
{ {
"name": "MemoryBanks" "name": "SyntheticDataGeneration"
},
{
"name": "Inference"
},
{
"name": "Inspect"
},
{
"name": "BatchInference"
},
{
"name": "Memory"
},
{
"name": "Datasets"
},
{
"name": "Agents"
},
{
"name": "PostTraining"
},
{
"name": "Telemetry"
},
{
"name": "Safety"
},
{
"name": "Evaluations"
}, },
{ {
"name": "BuiltinTool", "name": "BuiltinTool",

View file

@ -501,14 +501,17 @@ components:
CompletionResponse: CompletionResponse:
additionalProperties: false additionalProperties: false
properties: properties:
completion_message: content:
$ref: '#/components/schemas/CompletionMessage' type: string
logprobs: logprobs:
items: items:
$ref: '#/components/schemas/TokenLogProbs' $ref: '#/components/schemas/TokenLogProbs'
type: array type: array
stop_reason:
$ref: '#/components/schemas/StopReason'
required: required:
- completion_message - content
- stop_reason
title: Completion response. title: Completion response.
type: object type: object
CompletionResponseStreamChunk: CompletionResponseStreamChunk:
@ -2507,7 +2510,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-10-10 15:29:56.831109" \ draft and subject to change.\n Generated at 2024-10-18 20:48:17.730988"
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
@ -3712,21 +3715,21 @@ security:
servers: servers:
- url: http://any-hosted-llama-stack.com - url: http://any-hosted-llama-stack.com
tags: tags:
- name: Evaluations
- name: Inspect
- name: RewardScoring
- name: Datasets
- name: Models - name: Models
- name: Telemetry - name: RewardScoring
- name: PostTraining
- name: SyntheticDataGeneration
- name: BatchInference
- name: Inference
- name: Agents
- name: Memory
- name: Safety
- name: Shields
- name: MemoryBanks - name: MemoryBanks
- name: Shields
- name: SyntheticDataGeneration
- name: Inference
- name: Inspect
- name: BatchInference
- name: Memory
- name: Datasets
- name: Agents
- name: PostTraining
- name: Telemetry
- name: Safety
- name: Evaluations
- description: <SchemaDefinition schemaRef="#/components/schemas/BuiltinTool" /> - description: <SchemaDefinition schemaRef="#/components/schemas/BuiltinTool" />
name: BuiltinTool name: BuiltinTool
- description: <SchemaDefinition schemaRef="#/components/schemas/CompletionMessage" - description: <SchemaDefinition schemaRef="#/components/schemas/CompletionMessage"