first

2026-09-08 07:21:34 -04:00
commit a4b224da8e
169 changed files with 27483 additions and 0 deletions
@@ -0,0 +1,68 @@
+"""AIExecution — immutable record of a single inference call.
+
+Stored after every real (or mock) AI invocation so that:
+- usage/cost is attributable to a tenant + project + job
+- inputs/outputs can be audited / replayed
+- the AI Gateway can train / evaluate routing policies later
+"""
+
+from __future__ import annotations
+
+from typing import Any, Optional, Dict
+from pydantic import Field
+
+from asf_domain.base import (
+    ASFBaseModel,
+    TenantScopedModel,
+    ASFFieldID,
+    UTCDatetime,
+    generate_id,
+)
+
+
+class AIExecutionBase(ASFBaseModel):
+    capability: str = Field(
+        ...,
+        max_length=128,
+        description="ASF capability requested (e.g. chat, analyze_code, generate_tests).",
+    )
+    provider: str = Field(
+        ..., max_length=64, description="Adapter used (e.g. mock, openai, nvidia, local)."
+    )
+    model: str = Field(
+        ..., max_length=128, description="Concrete model identifier reported by the adapter."
+    )
+    success: bool = True
+
+
+class AIExecutionCreate(AIExecutionBase):
+    job_id: Optional[ASFFieldID] = None
+    project_id: Optional[ASFFieldID] = None
+    input_ref: Optional[str] = Field(
+        None, max_length=512,
+        description="MinIO ref or short fingerprint of the input payload (never raw secrets).",
+    )
+    output_ref: Optional[str] = Field(
+        None, max_length=512, description="MinIO ref / fingerprint of the response."
+    )
+    input_tokens: Optional[int] = Field(None, ge=0)
+    output_tokens: Optional[int] = Field(None, ge=0)
+    duration_ms: Optional[int] = Field(None, ge=0)
+    estimated_cost: Optional[float] = Field(None, ge=0)
+    error_message: Optional[str] = Field(None, max_length=2000)
+    metadata: Dict[str, Any] = Field(default_factory=dict)
+
+
+class AIExecution(TenantScopedModel, AIExecutionBase):
+    id: ASFFieldID = Field(default_factory=generate_id)
+    job_id: Optional[ASFFieldID] = None
+    project_id: Optional[ASFFieldID] = None
+    input_ref: Optional[str] = None
+    output_ref: Optional[str] = None
+    input_tokens: Optional[int] = None
+    output_tokens: Optional[int] = None
+    duration_ms: Optional[int] = None
+    estimated_cost: Optional[float] = None
+    error_message: Optional[str] = None
+    metadata: Dict[str, Any] = Field(default_factory=dict)
+    created_at: UTCDatetime = ...