LCORE-4696: Added initial providers registry structure

src/providers/__init__.py

@@ -0,0 +1 @@
+"""Inference provider registry package."""

src/providers/inference/__init__.py

@@ -0,0 +1 @@
+"""Inference providers, catalog models, and shared registry types."""

src/providers/inference/azure/__init__.py

@@ -0,0 +1 @@
+"""Azure inference provider package."""

src/providers/inference/bedrock/__init__.py

@@ -0,0 +1 @@
+"""Bedrock inference provider package."""

src/providers/inference/catalog.py

@@ -0,0 +1,16 @@
+"""Internal inference catalog models."""
+
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+from providers.inference.types import ModelType
+
+
+class CatalogModel(BaseModel):
+    """Cached catalog row for inference providers and the registry."""
+
+    name: str
+    provider_resource_id: str
+    model_type: ModelType
+    metadata: dict[str, Any] = Field(default_factory=dict)

src/providers/inference/openai/__init__.py

@@ -0,0 +1 @@
+"""OpenAI inference provider package."""

src/providers/inference/sentence_transformers/__init__.py

@@ -0,0 +1 @@
+"""Sentence-transformers inference provider package."""

src/providers/inference/types.py

@@ -0,0 +1,19 @@
+"""Shared literals and aliases for the inference provider registry."""
+
+from typing import Any, Final, Literal
+
+ProviderType = Literal[
+    "openai",
+    "azure",
+    "vertexai",
+    "bedrock",
+    "watsonx",
+    "vllm",
+    "sentence_transformers",
+]
+
+ModelType = Literal["llm", "embedding", "rerank"]
+
+type ProviderId = str
+
+PROVIDER_CLASSES: Final[dict[ProviderType, Any]] = {}

src/providers/inference/vertexai/__init__.py

@@ -0,0 +1 @@
+"""Vertex AI inference provider package."""

src/providers/inference/vllm/__init__.py

@@ -0,0 +1 @@
+"""vLLM inference provider package."""

src/providers/inference/watsonx/__init__.py

@@ -0,0 +1 @@
+"""WatsonX inference provider package."""

tests/unit/providers/__init__.py

@@ -0,0 +1 @@
+"""Unit tests for the providers package."""

tests/unit/providers/inference/__init__.py

@@ -0,0 +1 @@
+"""Unit tests for providers.inference."""

tests/unit/providers/inference/test_catalog.py

@@ -0,0 +1,17 @@
+"""Unit tests for internal inference catalog models."""
+
+from providers.inference.catalog import CatalogModel
+
+
+def test_catalog_model_construction() -> None:
+    """CatalogModel stores name, wire id, type, and metadata."""
+    row = CatalogModel(
+        name="fast-chat",
+        provider_resource_id="gpt-4o-mini",
+        model_type="llm",
+        metadata={"context_window": 128000},
+    )
+    assert row.name == "fast-chat"
+    assert row.provider_resource_id == "gpt-4o-mini"
+    assert row.model_type == "llm"
+    assert row.metadata["context_window"] == 128000