mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 01:48:05 +00:00
68 lines
2.2 KiB
Python
68 lines
2.2 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 datetime import datetime
|
|
from enum import StrEnum
|
|
from typing import Literal
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from llama_stack_api.resource import Resource, ResourceType
|
|
from llama_stack_api.schema_utils import json_schema_type
|
|
|
|
|
|
@json_schema_type
|
|
class RegistryType(StrEnum):
|
|
"""Type of registry."""
|
|
|
|
MCP = "mcp"
|
|
|
|
|
|
@json_schema_type
|
|
class Registry(Resource):
|
|
"""A registry resource representing a registry of connectors.
|
|
|
|
:param type: Type of resource, always 'registry' for registries
|
|
:param identifier: Unique identifier for this resource in llama stack
|
|
:param provider_resource_id: Unique identifier for this resource in the provider
|
|
:param provider_id: ID of the provider that owns this resource
|
|
:param type: Type of resource (e.g. 'model', 'shield', 'vector_store', etc.)
|
|
"""
|
|
|
|
type: Literal[ResourceType.registry] = ResourceType.registry
|
|
registry_type: RegistryType = Field(default=RegistryType.MCP)
|
|
user_registry_id: str | None = Field(default=None, description="User-specified identifier for the registry")
|
|
url: str = Field(..., description="URL of the registry")
|
|
created_at: datetime = Field(..., description="Timestamp of creation")
|
|
updated_at: datetime = Field(..., description="Timestamp of last update")
|
|
|
|
@property
|
|
def registry_id(self) -> str:
|
|
return self.user_registry_id if self.user_registry_id is not None else self.identifier
|
|
|
|
|
|
@json_schema_type
|
|
class ListRegistriesResponse(BaseModel):
|
|
"""Response containing a list of registries.
|
|
|
|
:param data: List of registries
|
|
"""
|
|
|
|
data: list[Registry]
|
|
|
|
|
|
@json_schema_type
|
|
class RegistryInput(BaseModel):
|
|
"""Input for creating a registry.
|
|
|
|
:param url: URL of the registry
|
|
:param user_registry_id: User-specified identifier for the registry
|
|
"""
|
|
|
|
url: str
|
|
user_registry_id: str | None = Field(default=None, description="User-specified identifier for the registry")
|
|
registry_type: RegistryType = Field(default=RegistryType.MCP)
|