feati(api): add connectors API

Closes #4235 and #4061 (partially)

Signed-off-by: Jaideep Rao <jrao@redhat.com>
This commit is contained in:
Jaideep Rao 2025-11-28 21:34:00 +05:30
parent 89807dc117
commit 8a5a56f166
11 changed files with 1977 additions and 0 deletions

View 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)