glaip-sdk 0.1.2__py3-none-any.whl → 0.7.17__py3-none-any.whl

glaip_sdk/models/agent.py

@@ -0,0 +1,48 @@
+"""Agent response model for AIP SDK.
+
+This module contains the Pydantic model for Agent API responses.
+This is a pure data model with no runtime behavior.
+
+For the runtime Agent class with deploy/run methods, use glaip_sdk.agents.Agent.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+from datetime import datetime
+from typing import Any
+
+from pydantic import BaseModel
+
+from glaip_sdk.config.constants import DEFAULT_AGENT_RUN_TIMEOUT
+
+
+class AgentResponse(BaseModel):
+    """Pydantic model for Agent API responses.
+
+    This is a pure data model for deserializing API responses.
+    It does NOT have runtime methods (run, update, delete).
+
+    For the runtime Agent class, use glaip_sdk.agents.Agent.
+    """
+
+    id: str
+    name: str
+    instruction: str | None = None
+    description: str | None = None
+    type: str | None = None
+    framework: str | None = None
+    version: str | None = None
+    tools: list[dict[str, Any]] | None = None
+    agents: list[dict[str, Any]] | None = None
+    mcps: list[dict[str, Any]] | None = None
+    tool_configs: dict[str, Any] | None = None
+    mcp_configs: dict[str, Any] | None = None
+    agent_config: dict[str, Any] | None = None
+    timeout: int = DEFAULT_AGENT_RUN_TIMEOUT
+    metadata: dict[str, Any] | None = None
+    language_model_id: str | None = None
+    a2a_profile: dict[str, Any] | None = None
+    created_at: datetime | None = None
+    updated_at: datetime | None = None

glaip_sdk/models/agent_runs.py

@@ -0,0 +1,117 @@
+#!/usr/bin/env python3
+"""Agent run models for AIP SDK.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from datetime import datetime, timedelta
+from typing import Any, Literal
+from uuid import UUID
+
+from pydantic import BaseModel, Field, field_validator, model_validator
+
+# Type alias for SSE event dictionaries
+RunOutputChunk = dict[str, Any]
+RunStatus = Literal["started", "success", "failed", "cancelled", "aborted", "unavailable"]
+
+
+class RunSummary(BaseModel):
+    """Represents a single agent run in list/table views with metadata only."""
+
+    id: UUID
+    agent_id: UUID
+    run_type: Literal["manual", "schedule"]
+    schedule_id: UUID | None = None
+    status: RunStatus
+    started_at: datetime
+    completed_at: datetime | None = None
+    input: str | None = None
+    config: dict[str, Any] | None = None
+    created_at: datetime
+    updated_at: datetime
+
+    @field_validator("completed_at")
+    @classmethod
+    def validate_completed_after_started(cls, v: datetime | None, info) -> datetime | None:
+        """Validate that completed_at is after started_at if present."""
+        if v is not None and "started_at" in info.data:
+            started_at = info.data["started_at"]
+            if v < started_at:
+                raise ValueError("completed_at must be after started_at")
+        return v
+
+    def duration(self) -> timedelta | None:
+        """Calculate duration from started_at to completed_at.
+
+        Returns:
+            Duration as timedelta if completed_at exists, None otherwise
+        """
+        if self.completed_at is not None:
+            return self.completed_at - self.started_at
+        return None
+
+    def duration_formatted(self) -> str:
+        """Format duration as HH:MM:SS string.
+
+        Returns:
+            Formatted duration string or "—" if not completed
+        """
+        duration = self.duration()
+        if duration is None:
+            return "—"
+        total_seconds = int(duration.total_seconds())
+        hours = total_seconds // 3600
+        minutes = (total_seconds % 3600) // 60
+        seconds = total_seconds % 60
+        return f"{hours:02d}:{minutes:02d}:{seconds:02d}"
+
+    def input_preview(self, max_length: int = 120) -> str:
+        """Generate truncated input preview for table display.
+
+        Args:
+            max_length: Maximum length of preview string
+
+        Returns:
+            Truncated input string or "—" if input is None or empty
+        """
+        if not self.input:
+            return "—"
+        # Strip newlines and collapse whitespace
+        preview = " ".join(self.input.split())
+        if len(preview) > max_length:
+            return preview[:max_length] + "…"
+        return preview
+
+
+class RunsPage(BaseModel):
+    """Represents a paginated collection of run summaries from the list endpoint."""
+
+    data: list[RunSummary]
+    total: int = Field(ge=0)
+    page: int = Field(ge=1)
+    limit: int = Field(ge=1, le=100)
+    has_next: bool
+    has_prev: bool
+
+    @model_validator(mode="after")
+    def validate_pagination_consistency(self) -> "RunsPage":
+        """Validate pagination consistency."""
+        # If has_next is True, then page * limit < total
+        if self.has_next and self.page * self.limit >= self.total:
+            raise ValueError("has_next inconsistency: page * limit must be < total when has_next is True")
+        return self
+
+
+class RunWithOutput(RunSummary):
+    """Extends RunSummary with the complete SSE event stream for detailed viewing."""
+
+    output: list[RunOutputChunk] = Field(default_factory=list)
+
+    @field_validator("output", mode="before")
+    @classmethod
+    def normalize_output(cls, v: Any) -> list[RunOutputChunk]:
+        """Normalize output field to empty list when null."""
+        if v is None:
+            return []
+        return v

glaip_sdk/models/common.py

@@ -0,0 +1,42 @@
+"""Common models for AIP SDK.
+
+This module contains common models that don't fit into specific categories.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+from pydantic import BaseModel
+
+
+class LanguageModelResponse(BaseModel):
+    """Language model response model."""
+
+    name: str
+    provider: str
+    description: str | None = None
+    capabilities: list[str] | None = None
+    max_tokens: int | None = None
+    supports_streaming: bool = False
+
+
+class TTYRenderer:
+    """Simple TTY renderer for non-Rich environments."""
+
+    def __init__(self, use_color: bool = True):
+        """Initialize the TTY renderer.
+
+        Args:
+            use_color: Whether to use color output
+        """
+        self.use_color = use_color
+
+    def render_message(self, message: str, event_type: str = "message") -> None:
+        """Render a message with optional color."""
+        if event_type == "error":
+            print(f"ERROR: {message}", flush=True)
+        elif event_type == "done":
+            print(f"\n✅ {message}", flush=True)
+        else:
+            print(message, flush=True)

glaip_sdk/models/constants.py

@@ -0,0 +1,141 @@
+# pylint: disable=duplicate-code
+"""Model constants for GL AIP SDK.
+
+Typed constants for popular AI models matching AIP Server's language_models.yaml keys.
+"""
+
+from __future__ import annotations
+
+__all__ = [
+    "ModelProvider",
+    "OpenAI",
+    "Anthropic",
+    "Google",
+    "AzureOpenAI",
+    "DeepInfra",
+    "DeepSeek",
+    "Bedrock",
+    "DEFAULT_MODEL",
+]
+
+# Note: DeepInfra provider changed to 'openai-compatible' for aip_agents compatibility
+
+
+class ModelProvider:
+    """Supported model providers."""
+
+    OPENAI = "openai"
+    ANTHROPIC = "anthropic"
+    AZURE_OPENAI = "azure-openai"
+    GOOGLE = "google"
+    DEEPINFRA = "deepinfra"
+    DEEPSEEK = "deepseek"
+    OPENAI_COMPATIBLE = "openai-compatible"
+    BEDROCK = "bedrock"
+
+
+class OpenAI:
+    """OpenAI model constants."""
+
+    GPT_4O_MINI = "openai/gpt-4o-mini"
+    GPT_4O = "openai/gpt-4o"
+    GPT_4_1 = "openai/gpt-4.1"
+    GPT_4_1_NANO = "openai/gpt-4.1-nano"
+    GPT_4_1_MINI = "openai/gpt-4.1-mini"
+    GPT_5 = "openai/gpt-5"
+    GPT_5_MINI = "openai/gpt-5-mini"
+    GPT_5_NANO = "openai/gpt-5-nano"
+    GPT_5_LOW = "openai/gpt-5-low"
+    GPT_5_MINIMAL = "openai/gpt-5-minimal"
+    GPT_5_MEDIUM = "openai/gpt-5-medium"
+    GPT_5_HIGH = "openai/gpt-5-high"
+    GPT_5_1 = "openai/gpt-5.1"
+    GPT_5_1_LOW = "openai/gpt-5.1-low"
+    GPT_5_1_MEDIUM = "openai/gpt-5.1-medium"
+    GPT_5_1_HIGH = "openai/gpt-5.1-high"
+    GPT_5_2 = "openai/gpt-5.2"
+    GPT_5_2_LOW = "openai/gpt-5.2-low"
+    GPT_5_2_MEDIUM = "openai/gpt-5.2-medium"
+    GPT_5_2_HIGH = "openai/gpt-5.2-high"
+    GPT_5_2_XHIGH = "openai/gpt-5.2-xhigh"
+    GPT_4O_2024_11_20 = "openai/gpt-4o-2024-11-20"
+    O4_MINI = "openai/o4-mini"
+
+
+class Anthropic:
+    """Anthropic model constants."""
+
+    CLAUDE_3_5_SONNET_LATEST = "anthropic/claude-3-5-sonnet-latest"
+    CLAUDE_3_7_SONNET_LATEST = "anthropic/claude-3-7-sonnet-latest"
+    CLAUDE_SONNET_4_0 = "anthropic/claude-sonnet-4-0"
+    CLAUDE_OPUS_4_0 = "anthropic/claude-opus-4-0"
+    CLAUDE_OPUS_4_1 = "anthropic/claude-opus-4-1"
+    CLAUDE_SONNET_4_5 = "anthropic/claude-sonnet-4-5"
+    # DX Alias
+    CLAUDE_3_5_SONNET = CLAUDE_3_5_SONNET_LATEST
+
+
+class Google:
+    """Google model constants."""
+
+    GEMINI_2_5_FLASH = "google/gemini-2.5-flash"
+    GEMINI_3_FLASH_PREVIEW = "google/gemini-3-flash-preview"
+    GEMINI_3_PRO_PREVIEW = "google/gemini-3-pro-preview"
+    GEMINI_2_5_PRO = "google/gemini-2.5-pro"
+
+
+class AzureOpenAI:
+    """Azure OpenAI model constants."""
+
+    GPT_4O = "azure-openai/gpt-4o"
+    GPT_4O_MINI = "azure-openai/gpt-4o-mini"
+    GPT_4_1 = "azure-openai/gpt-4.1"
+
+
+class DeepInfra:
+    """DeepInfra model constants.
+
+    Note: DeepInfra models use the openai-compatible driver internally,
+    but we expose them with the 'deepinfra/' prefix for better DX.
+    The SDK handles the mapping to 'openai-compatible/' automatically.
+    """
+
+    QWEN3_235B_A22B_INSTRUCT_2507 = "deepinfra/Qwen/Qwen3-235B-A22B-Instruct-2507"
+    KIMI_K2_INSTRUCT = "deepinfra/moonshotai/Kimi-K2-Instruct"
+    QWEN3_30B_A3B = "deepinfra/Qwen/Qwen3-30B-A3B"
+    GLM_4_5_AIR = "deepinfra/zai-org/GLM-4.5-Air"
+    QWEN3_32B = "deepinfra/Qwen/Qwen3-32B"
+    QWEN3_NEXT_80B_A3B_THINKING = "deepinfra/Qwen/Qwen3-Next-80B-A3B-Thinking"
+    QWEN3_NEXT_80B_A3B_INSTRUCT = "deepinfra/Qwen/Qwen3-Next-80B-A3B-Instruct"
+    QWEN3_14B = "deepinfra/Qwen/Qwen3-14B"
+    GLM_4_5 = "deepinfra/zai-org/GLM-4.5"
+
+
+class DeepSeek:
+    """DeepSeek model constants.
+
+    Note: DeepSeek models use the openai-compatible driver internally,
+    but we expose them with the 'deepseek/' prefix for better DX.
+    The SDK handles the mapping to 'openai-compatible/' automatically.
+    """
+
+    DEEPSEEK_CHAT = "deepseek/deepseek-chat"
+    DEEPSEEK_V3_1 = "deepseek/deepseek-ai/DeepSeek-V3.1"
+    DEEPSEEK_V3_1_TERMINUS = "deepseek/deepseek-ai/DeepSeek-V3.1-Terminus"
+
+
+class Bedrock:
+    """AWS Bedrock model constants."""
+
+    CLAUDE_SONNET_4_20250514_V1_0 = "bedrock/us.anthropic.claude-sonnet-4-20250514-v1:0"
+    CLAUDE_3_5_HAIKU_20241022_V1_0 = "bedrock/us.anthropic.claude-3-5-haiku-20241022-v1:0"
+    CLAUDE_SONNET_4_5_20250929_V1_0 = "bedrock/us.anthropic.claude-sonnet-4-5-20250929-v1:0"
+
+    # DX Aliases
+    ANTHROPIC_CLAUDE_SONNET_4 = CLAUDE_SONNET_4_20250514_V1_0
+    ANTHROPIC_CLAUDE_3_5_HAIKU = CLAUDE_3_5_HAIKU_20241022_V1_0
+    ANTHROPIC_CLAUDE_SONNET_4_5 = CLAUDE_SONNET_4_5_20250929_V1_0
+
+
+# Default model selection
+DEFAULT_MODEL = OpenAI.GPT_5_NANO

glaip_sdk/models/mcp.py

@@ -0,0 +1,33 @@
+"""MCP response model for AIP SDK.
+
+This module contains the Pydantic model for MCP API responses.
+This is a pure data model with no runtime behavior.
+
+For the runtime MCP class with update/delete methods, use glaip_sdk.mcps.MCP.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+    Christian Trisno Sen Long Chen (christian.t.s.l.chen@gdplabs.id)
+"""
+
+from typing import Any
+
+from pydantic import BaseModel
+
+
+class MCPResponse(BaseModel):
+    """Pydantic model for MCP API responses.
+
+    This is a pure data model for deserializing API responses.
+    It does NOT have runtime methods (update, delete, get_tools).
+
+    For the runtime MCP class, use glaip_sdk.mcps.MCP.
+    """
+
+    id: str
+    name: str
+    description: str | None = None
+    config: dict[str, Any] | None = None
+    transport: str | None = None  # "sse" or "http"
+    authentication: dict[str, Any] | None = None
+    metadata: dict[str, Any] | None = None

glaip_sdk/models/model.py

@@ -0,0 +1,170 @@
+"""Model configuration class for GL AIP SDK.
+
+Provides a structured way to specify models, credentials, and hyperparameters
+for local execution.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+from pydantic import BaseModel, field_validator
+
+
+class Model(BaseModel):
+    """Model configuration class for local execution.
+
+    Bundles model identity with credentials and hyperparameters.
+    """
+
+    id: str
+    credentials: dict[str, Any] | str | None = None
+    hyperparameters: dict[str, Any] | None = None
+    base_url: str | None = None
+
+    @field_validator("id")
+    @classmethod
+    def validate_id(cls, v: str) -> str:
+        """Validate model ID format.
+
+        Args:
+            v: Model identifier string.
+
+        Returns:
+            Validated model identifier in 'provider/model' format.
+
+        Raises:
+            ValueError: If model format is invalid.
+        """
+        from glaip_sdk.models._validation import _validate_model  # noqa: PLC0415
+
+        return _validate_model(v)
+
+    def to_aip_agents_format(self) -> tuple[str, dict[str, Any]]:
+        """Convert Model to aip_agents format.
+
+        Converts provider/model format to driver/model format for aip_agents.
+        For custom models with base_url, uses format: openai-compatible/base_url:model_name
+
+        Returns:
+            Tuple of (model_string, agent_config_dict).
+        """
+        if "/" not in self.id:
+            return self.id, {}
+
+        parts = self.id.split("/", 1)
+        provider = parts[0]
+        model_name = parts[1]
+
+        config = self._build_agent_config(provider)
+        model_string = self._build_model_string(provider, model_name, config)
+
+        return model_string, config
+
+    def _build_agent_config(self, provider: str) -> dict[str, Any]:
+        """Build agent config dict from Model attributes.
+
+        Args:
+            provider: Provider name extracted from model id.
+
+        Returns:
+            Configuration dict with credentials, hyperparameters, and base_url.
+        """
+        config: dict[str, Any] = {}
+
+        if self.credentials:
+            if isinstance(self.credentials, str):
+                config["lm_api_key"] = self.credentials
+            elif isinstance(self.credentials, dict):
+                config["lm_credentials"] = self.credentials
+
+        if self.hyperparameters:
+            config["lm_hyperparameters"] = self.hyperparameters
+
+        base_url = self._resolve_base_url(provider)
+        if base_url:
+            config["lm_base_url"] = base_url
+
+        return config
+
+    def _resolve_base_url(self, provider: str) -> str | None:
+        """Resolve base URL for the provider.
+
+        Uses centralized provider configurations to determine base_url.
+        Users can override by explicitly setting base_url attribute.
+
+        Args:
+            provider: Provider name from model ID (e.g., "deepinfra").
+
+        Returns:
+            Base URL string or None.
+
+        Examples:
+            >>> model = Model(id="deepinfra/Qwen/Qwen3-30B")
+            >>> model._resolve_base_url("deepinfra")
+            "https://api.deepinfra.com/v1/openai"
+        """
+        if self.base_url:
+            return self.base_url
+
+        # Get base_url from provider config
+        from glaip_sdk.models._provider_mappings import get_base_url  # noqa: PLC0415
+
+        base_url = get_base_url(provider)
+        if base_url:
+            return base_url
+
+        return None
+
+    def _build_model_string(self, provider: str, model_name: str, config: dict[str, Any]) -> str:
+        """Build normalized model string for aip_agents.
+
+        Converts provider names to their driver implementations for local execution.
+
+        Conversion strategy:
+        1. Custom models with base_url: Use colon format (openai-compatible/{base_url}:{model_name})
+           This allows aip_agents to parse base_url and model_name separately.
+        2. Standard providers: Map to driver and use slash format (driver/{model_name})
+           aip_agents will handle provider/model format internally.
+
+        Args:
+            provider: Provider name from model ID (e.g., "deepinfra", "openai").
+            model_name: Model name after provider prefix.
+            config: Agent config dict (may contain base_url).
+
+        Returns:
+            Normalized model string in provider/model format with driver name.
+
+        Examples:
+            >>> _build_model_string("deepinfra", "Qwen/Qwen3-30B", {})
+            "openai-compatible/Qwen/Qwen3-30B"
+            >>> _build_model_string("openai", "gpt-4o", {})
+            "openai/gpt-4o"
+        """
+        from glaip_sdk.models._provider_mappings import get_driver  # noqa: PLC0415
+
+        # Map provider to driver (e.g., deepinfra → openai-compatible)
+        driver = get_driver(provider)
+
+        if provider == "custom":
+            base_url = config.get("lm_base_url")
+            if base_url:
+                return f"{driver}/{base_url}:{model_name}"
+
+        # Standard case: driver with slash format
+        # aip_agents will handle provider/model format internally
+        return f"{driver}/{model_name}"
+
+    def __repr__(self) -> str:
+        """Return string representation of Model.
+
+        Note: Credentials are masked to avoid leaking secrets in logs.
+        Hyperparameters and base_url are shown if present.
+        """
+        creds_repr = "***" if self.credentials else "None"
+        return (
+            f"Model(id={self.id!r}, "
+            f"credentials={creds_repr}, "
+            f"hyperparameters={self.hyperparameters!r}, "
+            f"base_url={self.base_url!r})"
+        )