Add a special header per-client call to parser provider data

This commit is contained in:
Ashwin Bharambe 2024-09-18 09:17:59 -07:00 committed by Xi Yan
parent 0b715c063a
commit 9380661ebd
2 changed files with 359 additions and 0 deletions

View file

@ -21,7 +21,11 @@
"info": {
"title": "[DRAFT] Llama Stack Specification",
"version": "0.0.1",
<<<<<<< HEAD
"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-18 19:27:39.955190"
=======
"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-20 13:32:23.841908"
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
},
"servers": [
{
@ -2584,7 +2588,187 @@
"$ref": "#/components/schemas/FunctionCallToolDefinition"
},
{
<<<<<<< HEAD
"$ref": "#/components/schemas/MemoryToolDefinition"
=======
"type": "object",
"properties": {
"input_shields": {
"type": "array",
"items": {
"$ref": "#/components/schemas/ShieldDefinition"
}
},
"output_shields": {
"type": "array",
"items": {
"$ref": "#/components/schemas/ShieldDefinition"
}
},
"type": {
"type": "string",
"const": "memory"
},
"memory_bank_configs": {
"type": "array",
"items": {
"oneOf": [
{
"type": "object",
"properties": {
"bank_id": {
"type": "string"
},
"type": {
"type": "string",
"const": "vector"
}
},
"additionalProperties": false,
"required": [
"bank_id",
"type"
]
},
{
"type": "object",
"properties": {
"bank_id": {
"type": "string"
},
"type": {
"type": "string",
"const": "keyvalue"
},
"keys": {
"type": "array",
"items": {
"type": "string"
}
}
},
"additionalProperties": false,
"required": [
"bank_id",
"type",
"keys"
]
},
{
"type": "object",
"properties": {
"bank_id": {
"type": "string"
},
"type": {
"type": "string",
"const": "keyword"
}
},
"additionalProperties": false,
"required": [
"bank_id",
"type"
]
},
{
"type": "object",
"properties": {
"bank_id": {
"type": "string"
},
"type": {
"type": "string",
"const": "graph"
},
"entities": {
"type": "array",
"items": {
"type": "string"
}
}
},
"additionalProperties": false,
"required": [
"bank_id",
"type",
"entities"
]
}
]
}
},
"query_generator_config": {
"oneOf": [
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "default"
},
"sep": {
"type": "string"
}
},
"additionalProperties": false,
"required": [
"type",
"sep"
]
},
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "llm"
},
"model": {
"type": "string"
},
"template": {
"type": "string"
}
},
"additionalProperties": false,
"required": [
"type",
"model",
"template"
]
},
{
"type": "object",
"properties": {
"type": {
"type": "string",
"const": "custom"
}
},
"additionalProperties": false,
"required": [
"type"
]
}
]
},
"max_tokens_in_context": {
"type": "integer"
},
"max_chunks": {
"type": "integer"
}
},
"additionalProperties": false,
"required": [
"type",
"memory_bank_configs",
"query_generator_config",
"max_tokens_in_context",
"max_chunks"
]
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
}
]
}
@ -2693,6 +2877,7 @@
"parameters"
]
},
<<<<<<< HEAD
"MemoryToolDefinition": {
"type": "object",
"properties": {
@ -2872,6 +3057,8 @@
"max_chunks"
]
},
=======
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
"OnViolationAction": {
"type": "integer",
"enum": [
@ -5589,7 +5776,14 @@
],
"tags": [
{
<<<<<<< HEAD
"name": "Safety"
=======
"name": "BatchInference"
},
{
"name": "PostTraining"
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
},
{
"name": "Memory"
@ -5598,6 +5792,7 @@
"name": "Evaluations"
},
{
<<<<<<< HEAD
"name": "RewardScoring"
},
{
@ -5605,18 +5800,35 @@
},
{
"name": "Evaluations"
=======
"name": "Datasets"
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
},
{
"name": "SyntheticDataGeneration"
},
{
<<<<<<< HEAD
"name": "Memory"
=======
"name": "RewardScoring"
},
{
"name": "Agents"
},
{
"name": "Safety"
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
},
{
"name": "Inference"
},
{
<<<<<<< HEAD
"name": "Datasets"
=======
"name": "Telemetry"
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
},
{
"name": "BuiltinTool",
@ -5755,10 +5967,13 @@
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/FunctionCallToolDefinition\" />"
},
{
<<<<<<< HEAD
"name": "MemoryToolDefinition",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/MemoryToolDefinition\" />"
},
{
=======
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
"name": "OnViolationAction",
"description": "<SchemaDefinition schemaRef=\"#/components/schemas/OnViolationAction\" />"
},
@ -6204,7 +6419,10 @@
"MemoryBank",
"MemoryBankDocument",
"MemoryRetrievalStep",
<<<<<<< HEAD
"MemoryToolDefinition",
=======
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
"MetricEvent",
"OnViolationAction",
"OptimizerConfig",

View file

@ -30,7 +30,127 @@ components:
- $ref: '#/components/schemas/PhotogenToolDefinition'
- $ref: '#/components/schemas/CodeInterpreterToolDefinition'
- $ref: '#/components/schemas/FunctionCallToolDefinition'
<<<<<<< HEAD
- $ref: '#/components/schemas/MemoryToolDefinition'
=======
- additionalProperties: false
properties:
input_shields:
items:
$ref: '#/components/schemas/ShieldDefinition'
type: array
max_chunks:
type: integer
max_tokens_in_context:
type: integer
memory_bank_configs:
items:
oneOf:
- additionalProperties: false
properties:
bank_id:
type: string
type:
const: vector
type: string
required:
- bank_id
- type
type: object
- additionalProperties: false
properties:
bank_id:
type: string
keys:
items:
type: string
type: array
type:
const: keyvalue
type: string
required:
- bank_id
- type
- keys
type: object
- additionalProperties: false
properties:
bank_id:
type: string
type:
const: keyword
type: string
required:
- bank_id
- type
type: object
- additionalProperties: false
properties:
bank_id:
type: string
entities:
items:
type: string
type: array
type:
const: graph
type: string
required:
- bank_id
- type
- entities
type: object
type: array
output_shields:
items:
$ref: '#/components/schemas/ShieldDefinition'
type: array
query_generator_config:
oneOf:
- additionalProperties: false
properties:
sep:
type: string
type:
const: default
type: string
required:
- type
- sep
type: object
- additionalProperties: false
properties:
model:
type: string
template:
type: string
type:
const: llm
type: string
required:
- type
- model
- template
type: object
- additionalProperties: false
properties:
type:
const: custom
type: string
required:
- type
type: object
type:
const: memory
type: string
required:
- type
- memory_bank_configs
- query_generator_config
- max_tokens_in_context
- max_chunks
type: object
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
type: array
required:
- model
@ -1074,6 +1194,7 @@ components:
- memory_bank_ids
- inserted_context
type: object
<<<<<<< HEAD
MemoryToolDefinition:
additionalProperties: false
properties:
@ -1192,6 +1313,8 @@ components:
- max_tokens_in_context
- max_chunks
type: object
=======
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
MetricEvent:
additionalProperties: false
properties:
@ -3469,6 +3592,7 @@ servers:
tags:
- name: BatchInference
- name: PostTraining
<<<<<<< HEAD
- name: Inference
- name: Safety
- name: RewardScoring
@ -3478,6 +3602,17 @@ tags:
- name: Memory
- name: Agents
- name: Datasets
=======
- name: Memory
- name: Evaluations
- name: Datasets
- name: SyntheticDataGeneration
- name: RewardScoring
- name: Agents
- name: Safety
- name: Inference
- name: Telemetry
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
- description: <SchemaDefinition schemaRef="#/components/schemas/BuiltinTool" />
name: BuiltinTool
- description: <SchemaDefinition schemaRef="#/components/schemas/CompletionMessage"
@ -3589,9 +3724,12 @@ tags:
- description: <SchemaDefinition schemaRef="#/components/schemas/FunctionCallToolDefinition"
/>
name: FunctionCallToolDefinition
<<<<<<< HEAD
- description: <SchemaDefinition schemaRef="#/components/schemas/MemoryToolDefinition"
/>
name: MemoryToolDefinition
=======
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
- description: <SchemaDefinition schemaRef="#/components/schemas/OnViolationAction"
/>
name: OnViolationAction
@ -3954,7 +4092,10 @@ x-tagGroups:
- MemoryBank
- MemoryBankDocument
- MemoryRetrievalStep
<<<<<<< HEAD
- MemoryToolDefinition
=======
>>>>>>> 446914e (Add a special header per-client call to parser provider data)
- MetricEvent
- OnViolationAction
- OptimizerConfig