apipod-registry 0.0.6

Service registry for APIPod clients.

The service registry behind the SocAIty SDK.

Define AI services as typed Pydantic models, store them anywhere, look them up by id or name.
One source of truth for the FastSDK engine and the socaity wrapper.

What it is

apipod_registry is the definitions and registry layer that the SocAIty SDK runs on. It gives you:

• Typed definitions. ServiceDefinition, EndpointDefinition, ModelDefinition, ServiceCategory, and ServiceFamily are Pydantic models. Validated on construction, serializable, no loose dicts.
• A registry. Registry holds services in memory and behaves like a dict. Look up by id or by display name, both resolve to the same typed object.
• Pluggable storage. Pass a store (Supabase, Postgres, filesystem) and the registry persists and loads for you. Leave it out to stay fully in memory.
• Parsers. Turn an OpenAPI, FastTaskAPI, or Cog schema into a ServiceDefinition without writing it by hand.

Install

pip install apipod_registry

Python 3.8 or newer.

Quick start

Create a registry, add a service, get it back typed:

from apipod_registry import Registry, ServiceDefinition, EndpointDefinition

reg = Registry()

reg.add_service(ServiceDefinition(
    id="flux-schnell",
    display_name="Flux Schnell",
    specification="replicate",
    endpoints=[EndpointDefinition(id="predict", path="/predict", method="POST")],
))

svc = reg.get_service("flux-schnell")   # by id
svc = reg.get_service("Flux Schnell")   # or by display name, same object
ep = reg.get_endpoint("flux-schnell", "/predict")

The registry is dict-like: reg["flux-schnell"], len(reg), and for service_id in reg all work. list_services(), filter_services(**attrs), get_services_by_family(...), and get_services_by_category(...) cover lookups across the set.

Persisting to a store

Pass an IServiceRegistry backend and the registry loads on init and saves on write. A Supabase helper ships in the box:

from apipod_registry import create_service_registry_with_supabase

reg = create_service_registry_with_supabase()   # reads SUPABASE_URL / SUPABASE_KEY
reg.add_service(my_service)                      # persisted, not just in memory

Postgres and filesystem stores live under apipod_registry.service_registry. Implement IServiceRegistry to back the registry with anything else.

License

GPLv3. See LICENSE.