acex-devkit 2.2.0__tar.gz → 2.3.0__tar.gz

{acex_devkit-2.2.0 → acex_devkit-2.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: acex-devkit
-Version: 2.2.0
+Version: 2.3.0
 Summary: ACE-X DevKit - Development kit for building ACE-X drivers and plugins
 License: AGPL-3.0
 Keywords: automation,devkit,sdk,drivers,plugins

{acex_devkit-2.2.0 → acex_devkit-2.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "acex-devkit"
-version = "2.2.0"
+version = "2.3.0"
 description = "ACE-X DevKit - Development kit for building ACE-X drivers and plugins"
 authors = ["Johan Lahti <johan.lahti@acebit.se>"]
 readme = "README.md"

{acex_devkit-2.2.0 → acex_devkit-2.3.0}/src/acex_devkit/models/__init__.py

@@ -6,7 +6,16 @@ from .agent_manifest import (
     CollectionAgentManifest,
     ManifestTarget,
 )
-from .ai_ops import AiAnalysisRequest, AiAnalysisTask, AiAskRequest
+from .ai_ops import (
+    AiAnalysisRequest,
+    AiAnalysisTask,
+    AiAskRequest,
+    AiChainLevel,
+    AiModelInfo,
+    AiModelsResponse,
+    AiProviderInfo,
+    AiProvidersResponse,
+)
 from .asset import (
     Asset,
     AssetClusterAssetResponse,
@@ -236,6 +245,11 @@ __all__ = [
     "AiAskRequest",
     "AiAnalysisTask",
     "AiAnalysisRequest",
+    "AiChainLevel",
+    "AiModelInfo",
+    "AiModelsResponse",
+    "AiProviderInfo",
+    "AiProvidersResponse",
     "TelemetryCapability",
     "TelemetryAgentBase",
     "TelemetryAgentCreate",

acex_devkit-2.3.0/src/acex_devkit/models/ai_ops.py

@@ -0,0 +1,78 @@
+from pydantic import BaseModel
+
+
+class AiAskRequest(BaseModel):
+    """Body for `POST /ai_ops/ai/ask/`. Returns an SSE stream of str chunks."""
+
+    question: str
+    node_instance_id: int | None = None
+    site: str | None = None
+    # Explicit model override ("provider/model" or "model"). Omitted = failover chain.
+    model: str | None = None
+
+
+class AiAnalysisTask(BaseModel):
+    """Task type for AI config-diff analysis."""
+
+    name: str
+    description: str = ""
+
+
+class AiAnalysisRequest(BaseModel):
+    """Body for `POST /ai_ops/ai/config_analysis/`. Returns an SSE stream."""
+
+    task: str  # "explain" | "risk_assessment" | "alignment"
+    diff: str
+    node_instance_id: int | None = None
+    model: str | None = None
+
+
+class AiChainLevel(BaseModel):
+    """One level in a failover chain, as exposed by `GET /ai_ops/providers`."""
+
+    provider: str
+    model: str
+
+
+class AiModelInfo(BaseModel):
+    """One available model with capability/cost metadata.
+
+    Fields are None when the provider does not report them and no
+    override is configured.
+    """
+
+    id: str
+    is_chat_model: bool = True  # False for embeddings/STT/rerank/image models
+    supports_tools: bool | None = None  # function calling / tool use
+    supports_vision: bool | None = None
+    context_window: int | None = None
+    input_cost_per_mtok: float | None = None
+    output_cost_per_mtok: float | None = None
+    currency: str | None = None
+    extra: dict | None = None  # provider-specific passthrough
+
+
+class AiProviderInfo(BaseModel):
+    """A configured AI provider enriched with health + available models."""
+
+    name: str
+    base_url: str
+    status: str  # "ok" | "unreachable"
+    models: list[AiModelInfo] = []
+    has_static_models: bool = False
+
+
+class AiProvidersResponse(BaseModel):
+    """Response for `GET /ai_ops/providers`."""
+
+    providers: list[AiProviderInfo]
+    # Per-task failover chains: task name -> ordered levels. Tasks not
+    # explicitly configured inherit "default".
+    chains: dict[str, list[AiChainLevel]] = {}
+
+
+class AiModelsResponse(BaseModel):
+    """Response for `GET /ai_ops/models?provider=X`."""
+
+    provider: str
+    models: list[AiModelInfo]

acex_devkit-2.2.0/src/acex_devkit/models/ai_ops.py

@@ -1,24 +0,0 @@
-from pydantic import BaseModel
-
-
-class AiAskRequest(BaseModel):
-    """Body for `POST /ai_ops/ai/ask/`. Returns an SSE stream of str chunks."""
-
-    question: str
-    node_instance_id: int | None = None
-    site: str | None = None
-
-
-class AiAnalysisTask(BaseModel):
-    """Task type for AI config-diff analysis."""
-
-    name: str
-    description: str = ""
-
-
-class AiAnalysisRequest(BaseModel):
-    """Body for `POST /ai_ops/ai/config_analysis/`. Returns an SSE stream."""
-
-    task: str  # "explain" | "risk_assessment" | "alignment"
-    diff: str
-    node_instance_id: int | None = None