mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-06-28 19:04:19 +00:00
0d96070af9
We desperately need to document our APIs. This is the basic requirement of having a Spec :) This PR updates the OpenAPI generator so documentation for request parameters and object fields can be properly added to the OpenAPI specs. From there, this should get picked by Stainless, etc. ## Test Plan: Updated client-sdk (See https://github.com/meta-llama/llama-stack-client-python/pull/104) and then ran: ```bash cd tests/client-sdk LLAMA_STACK_CONFIG=../../llama_stack/templates/fireworks/run.yaml pytest -s -v inference/test_inference.py agents/test_agents.py ```
39 lines
1 KiB
Python
39 lines
1 KiB
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
from enum import Enum
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
class ResourceType(Enum):
|
|
model = "model"
|
|
shield = "shield"
|
|
vector_db = "vector_db"
|
|
dataset = "dataset"
|
|
scoring_function = "scoring_function"
|
|
eval_task = "eval_task"
|
|
tool = "tool"
|
|
tool_group = "tool_group"
|
|
|
|
|
|
class Resource(BaseModel):
|
|
"""Base class for all Llama Stack resources"""
|
|
|
|
identifier: str = Field(
|
|
description="Unique identifier for this resource in llama stack"
|
|
)
|
|
|
|
provider_resource_id: str = Field(
|
|
description="Unique identifier for this resource in the provider",
|
|
default=None,
|
|
)
|
|
|
|
provider_id: str = Field(description="ID of the provider that owns this resource")
|
|
|
|
type: ResourceType = Field(
|
|
description="Type of resource (e.g. 'model', 'shield', 'vector_db', etc.)"
|
|
)
|