From 9380661ebdc785595f5841afcad671920184846e Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Wed, 18 Sep 2024 09:17:59 -0700 Subject: [PATCH] Add a special header per-client call to parser provider data --- docs/llama-stack-spec.html | 218 +++++++++++++++++++++++++++++++++++++ docs/llama-stack-spec.yaml | 141 ++++++++++++++++++++++++ 2 files changed, 359 insertions(+) diff --git a/docs/llama-stack-spec.html b/docs/llama-stack-spec.html index c7c456dbd..e7b1af16f 100644 --- a/docs/llama-stack-spec.html +++ b/docs/llama-stack-spec.html @@ -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": "" }, { +<<<<<<< HEAD "name": "MemoryToolDefinition", "description": "" }, { +======= +>>>>>>> 446914e (Add a special header per-client call to parser provider data) "name": "OnViolationAction", "description": "" }, @@ -6204,7 +6419,10 @@ "MemoryBank", "MemoryBankDocument", "MemoryRetrievalStep", +<<<<<<< HEAD "MemoryToolDefinition", +======= +>>>>>>> 446914e (Add a special header per-client call to parser provider data) "MetricEvent", "OnViolationAction", "OptimizerConfig", diff --git a/docs/llama-stack-spec.yaml b/docs/llama-stack-spec.yaml index 4d08158e3..d2de426bb 100644 --- a/docs/llama-stack-spec.yaml +++ b/docs/llama-stack-spec.yaml @@ -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: name: BuiltinTool - description: name: FunctionCallToolDefinition +<<<<<<< HEAD - description: name: MemoryToolDefinition +======= +>>>>>>> 446914e (Add a special header per-client call to parser provider data) - description: 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