mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 18:00:36 +00:00
feati(api): add connectors API
Closes #4235 and #4061 (partially) Signed-off-by: Jaideep Rao <jrao@redhat.com>
This commit is contained in:
parent
89807dc117
commit
8a5a56f166
11 changed files with 1977 additions and 0 deletions
68
src/llama_stack_api/registries.py
Normal file
68
src/llama_stack_api/registries.py
Normal file
|
|
@ -0,0 +1,68 @@
|
|||
# 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)
|
||||
Loading…
Add table
Add a link
Reference in a new issue