diff --git a/source/api_definitions.py b/source/api_definitions.py index 66473b2e7..84e0954f7 100644 --- a/source/api_definitions.py +++ b/source/api_definitions.py @@ -143,7 +143,7 @@ class BatchChatCompletionRequest: class Inference(Protocol): - + """Set of methods that can be called on the inference service.""" def post_completion( self, request: CompletionRequest, @@ -301,7 +301,7 @@ class MemoryBanks(Protocol): ) -> List[MemoryBankDocument]: ... @webmethod(route="/memory_bank/delete") - def post_delete_memory_documents( + def remove_memory_documents( self, bank_uuid: str, document_uuids: List[str], @@ -542,11 +542,13 @@ if __name__ == "__main__": spec = Specification( LlamaStackEndpoints, Options( - server=Server(url="http://llama.meta.com"), + server=Server(url="http://any-hosted-llama-stack.com"), info=Info( - title="Llama Stack specification", - version="0.1", - description="This is the llama stack", + title="[DRAFT] Llama Stack Specification", + version="0.0.1", + description="""This is the specification of the llama stack that provides + a set of endpoints and their corresponding interfaces that are tailored to + best leverage Llama Models. The specification is still in draft and subject to change.""", ), ), ) diff --git a/source/openapi.html b/source/openapi.html index f8264aef8..a490bac46 100644 --- a/source/openapi.html +++ b/source/openapi.html @@ -19,13 +19,13 @@ spec = { "openapi": "3.1.0", "info": { - "title": "Llama Stack specification", - "version": "0.1", - "description": "This is the llama stack" + "title": "[DRAFT] Llama Stack Specification RFC", + "version": "0.0.0", + "description": "This is the specification of the llama stack that allows" }, "servers": [ { - "url": "http://llama.meta.com" + "url": "http://any-hosted-llama-stack.com" } ], "paths": { @@ -520,41 +520,6 @@ } } }, - "/memory_bank/delete": { - "post": { - "responses": { - "200": { - "description": "OK" - } - }, - "tags": [ - "MemoryBanks" - ], - "parameters": [ - { - "name": "bank_uuid", - "in": "query", - "required": true, - "schema": { - "type": "string" - } - } - ], - "requestBody": { - "content": { - "application/json": { - "schema": { - "type": "array", - "items": { - "type": "string" - } - } - } - }, - "required": true - } - } - }, "/synthetic_data_generation/generate": { "post": { "responses": { @@ -882,6 +847,41 @@ } ] } + }, + "/memory_bank/delete": { + "post": { + "responses": { + "200": { + "description": "OK" + } + }, + "tags": [ + "MemoryBanks" + ], + "parameters": [ + { + "name": "bank_uuid", + "in": "query", + "required": true, + "schema": { + "type": "string" + } + } + ], + "requestBody": { + "content": { + "application/json": { + "schema": { + "type": "array", + "items": { + "type": "string" + } + } + } + }, + "required": true + } + } } }, "jsonSchemaDialect": "https://json-schema.org/draft/2020-12/schema", @@ -3323,12 +3323,6 @@ { "name": "RewardScoring" }, - { - "name": "AgenticSystem" - }, - { - "name": "MemoryBanks" - }, { "name": "Inference" }, @@ -3338,6 +3332,12 @@ { "name": "SyntheticDataGeneration" }, + { + "name": "MemoryBanks" + }, + { + "name": "AgenticSystem" + }, { "name": "ShieldConfig", "description": "" diff --git a/source/openapi.yaml b/source/openapi.yaml index d05153ca1..be291e84f 100644 --- a/source/openapi.yaml +++ b/source/openapi.yaml @@ -1502,9 +1502,9 @@ components: pattern: ^(https?://|file://|data:) type: string info: - description: This is the llama stack - title: Llama Stack specification - version: '0.1' + description: This is the specification of the llama stack that allows + title: '[DRAFT] Llama Stack Specification RFC' + version: 0.0.0 jsonSchemaDialect: https://json-schema.org/draft/2020-12/schema openapi: 3.1.0 paths: @@ -2010,15 +2010,15 @@ paths: security: - Default: [] servers: -- url: http://llama.meta.com +- url: http://any-hosted-llama-stack.com tags: - name: PostTraining - name: RewardScoring -- name: AgenticSystem -- name: MemoryBanks - name: Inference - name: Datasets - name: SyntheticDataGeneration +- name: MemoryBanks +- name: AgenticSystem - description: name: ShieldConfig - description: