Move control plane to distribution server for now

This commit is contained in:
Ashwin Bharambe 2024-09-17 12:55:01 -07:00
parent 099ac81bc7
commit a0bf20f19a
14 changed files with 22 additions and 26 deletions

View file

@ -6,7 +6,6 @@
from llama_models.llama3.api.datatypes import * # noqa: F403 from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.apis.agents import * # noqa: F403 from llama_stack.apis.agents import * # noqa: F403
from llama_stack.apis.control_plane import * # noqa: F403
from llama_stack.apis.dataset import * # noqa: F403 from llama_stack.apis.dataset import * # noqa: F403
from llama_stack.apis.evals import * # noqa: F403 from llama_stack.apis.evals import * # noqa: F403
from llama_stack.apis.inference import * # noqa: F403 from llama_stack.apis.inference import * # noqa: F403
@ -22,7 +21,6 @@ from llama_stack.apis.safety import * # noqa: F403
class LlamaStack( class LlamaStack(
Inference, Inference,
BatchInference, BatchInference,
ControlPlane,
Agents, Agents,
RewardScoring, RewardScoring,
Safety, Safety,

View file

@ -9,7 +9,6 @@ import inspect
from typing import Dict, List from typing import Dict, List
from llama_stack.apis.agents import Agents from llama_stack.apis.agents import Agents
from llama_stack.apis.control_plane import ControlPlane
from llama_stack.apis.inference import Inference from llama_stack.apis.inference import Inference
from llama_stack.apis.memory import Memory from llama_stack.apis.memory import Memory
from llama_stack.apis.safety import Safety from llama_stack.apis.safety import Safety
@ -38,7 +37,6 @@ def api_endpoints() -> Dict[Api, List[ApiEndpoint]]:
Api.agents: Agents, Api.agents: Agents,
Api.memory: Memory, Api.memory: Memory,
Api.telemetry: Telemetry, Api.telemetry: Telemetry,
Api.control_plane: ControlPlane,
} }
for api, protocol in protocols.items(): for api, protocol in protocols.items():

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-09-16 21:21:00.554769" "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-17 12:55:45.538053"
}, },
"servers": [ "servers": [
{ {
@ -5140,23 +5140,29 @@
} }
], ],
"tags": [ "tags": [
{
"name": "Agents"
},
{ {
"name": "Safety" "name": "Safety"
}, },
{ {
"name": "Memory" "name": "SyntheticDataGeneration"
}, },
{ {
"name": "PostTraining" "name": "Telemetry"
},
{
"name": "Datasets"
},
{
"name": "RewardScoring"
}, },
{ {
"name": "Evaluations" "name": "Evaluations"
}, },
{ {
"name": "Agents" "name": "PostTraining"
},
{
"name": "Datasets"
}, },
{ {
"name": "Inference" "name": "Inference"
@ -5165,13 +5171,7 @@
"name": "BatchInference" "name": "BatchInference"
}, },
{ {
"name": "Telemetry" "name": "Memory"
},
{
"name": "SyntheticDataGeneration"
},
{
"name": "RewardScoring"
}, },
{ {
"name": "BuiltinTool", "name": "BuiltinTool",

View file

@ -2356,7 +2356,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-09-16 21:21:00.554769" \ draft and subject to change.\n Generated at 2024-09-17 12:55:45.538053"
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
@ -3144,17 +3144,17 @@ security:
servers: servers:
- url: http://any-hosted-llama-stack.com - url: http://any-hosted-llama-stack.com
tags: tags:
- name: Safety
- name: Memory
- name: PostTraining
- name: Evaluations
- name: Agents - name: Agents
- name: Safety
- name: SyntheticDataGeneration
- name: Telemetry
- name: Datasets - name: Datasets
- name: RewardScoring
- name: Evaluations
- name: PostTraining
- name: Inference - name: Inference
- name: BatchInference - name: BatchInference
- name: Telemetry - name: Memory
- name: SyntheticDataGeneration
- name: RewardScoring
- 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"