glaip-sdk 0.0.1b5__py3-none-any.whl

glaip_sdk/config/constants.py

@@ -0,0 +1,28 @@
+"""Configuration constants for the AIP SDK.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+# Default language model configuration
+DEFAULT_MODEL = "gpt-4.1"
+DEFAULT_MODEL_PROVIDER = "openai"
+
+# Default timeout values
+DEFAULT_TIMEOUT = 30.0
+DEFAULT_AGENT_TIMEOUT = 300.0
+
+# User agent
+SDK_NAME = "glaip-sdk"
+SDK_VERSION = "0.1.0"
+
+# Reserved names that cannot be used for agents/tools
+RESERVED_NAMES = {
+    "system",
+    "admin",
+    "root",
+    "test",
+    "example",
+    "demo",
+    "sample",
+}

glaip_sdk/exceptions.py

@@ -0,0 +1,93 @@
+#!/usr/bin/env python3
+"""Custom exceptions for AIP SDK.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from typing import Any
+
+
+class AIPError(Exception):
+    """Base exception for AIP SDK."""
+
+    pass
+
+
+class APIError(AIPError):
+    """Base API exception with rich context."""
+
+    def __init__(
+        self,
+        message: str,
+        *,
+        status_code: int | None = None,
+        error_type: str | None = None,
+        payload: Any = None,
+        request_id: str | None = None,
+    ):
+        super().__init__(message)
+        self.status_code = status_code
+        self.error_type = error_type
+        self.payload = payload
+        self.request_id = request_id
+
+
+class AuthenticationError(APIError):
+    """Authentication failed."""
+
+    pass
+
+
+class ValidationError(APIError):
+    """Validation failed."""
+
+    pass
+
+
+class ForbiddenError(APIError):
+    """Access forbidden."""
+
+    pass
+
+
+class NotFoundError(APIError):
+    """Resource not found."""
+
+    pass
+
+
+class ConflictError(APIError):
+    """Resource conflict."""
+
+    pass
+
+
+class AmbiguousResourceError(APIError):
+    """Multiple resources match the query."""
+
+    pass
+
+
+class ServerError(APIError):
+    """Server error."""
+
+    pass
+
+
+class RateLimitError(APIError):
+    """Rate limit exceeded."""
+
+    pass
+
+
+class TimeoutError(APIError):
+    """Request timeout."""
+
+    pass
+
+
+class ClientError(APIError):
+    """Client-side error (e.g., invalid request format, missing parameters)."""
+
+    pass

glaip_sdk/models.py

@@ -0,0 +1,190 @@
+#!/usr/bin/env python3
+"""Data models for AIP SDK.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+from typing import Any
+
+from pydantic import BaseModel
+
+
+class Agent(BaseModel):
+    """Agent model for API responses."""
+
+    id: str
+    name: str
+    instruction: str | None = None
+    description: str | None = None  # Add missing description field
+    type: str | None = None
+    framework: str | None = None
+    version: str | None = None
+    tools: list[dict[str, Any]] | None = None  # Backend returns ToolReference objects
+    agents: list[dict[str, Any]] | None = None  # Backend returns AgentReference objects
+    agent_config: dict[str, Any] | None = None
+    timeout: int = 300
+    _client: Any = None  # Will hold client reference
+
+    def _set_client(self, client):
+        """Set the client reference for this resource."""
+        self._client = client
+        return self
+
+    def run(self, message: str, **kwargs) -> str:
+        """Run the agent with a message."""
+        if not self._client:
+            raise RuntimeError(
+                "No client available. Use client.get_agent_by_id() to get a client-connected agent."
+            )
+        # Automatically pass the agent name for better renderer display
+        kwargs.setdefault("agent_name", self.name)
+        return self._client.run_agent(self.id, message, **kwargs)
+
+    def update(self, **kwargs) -> "Agent":
+        """Update agent attributes."""
+        if not self._client:
+            raise RuntimeError(
+                "No client available. Use client.get_agent_by_id() to get a client-connected agent."
+            )
+        updated_agent = self._client.update_agent(self.id, **kwargs)
+        # Update current instance with new data
+        for key, value in updated_agent.model_dump().items():
+            if hasattr(self, key):
+                setattr(self, key, value)
+        return self
+
+    def delete(self) -> None:
+        """Delete the agent."""
+        if not self._client:
+            raise RuntimeError(
+                "No client available. Use client.get_agent_by_id() to get a client-connected agent."
+            )
+        self._client.delete_agent(self.id)
+
+
+class Tool(BaseModel):
+    """Tool model for API responses."""
+
+    id: str
+    name: str
+    tool_type: str | None = None
+    description: str | None = None
+    framework: str | None = None
+    version: str | None = None
+    tool_script: str | None = None
+    tool_file: str | None = None
+    _client: Any = None  # Will hold client reference
+
+    def _set_client(self, client):
+        """Set the client reference for this resource."""
+        self._client = client
+        return self
+
+    def get_script(self) -> str:
+        """Get the tool script content."""
+        if self.tool_script:
+            return self.tool_script
+        elif self.tool_file:
+            return f"Script content from file: {self.tool_file}"
+        else:
+            return "No script content available"
+
+    def update(self, **kwargs) -> "Tool":
+        """Update tool attributes."""
+        if not self._client:
+            raise RuntimeError(
+                "No client available. Use client.get_tool_by_id() to get a client-connected tool."
+            )
+        updated_tool = self._client.tools.update_tool(self.id, **kwargs)
+        # Update current instance with new data
+        for key, value in updated_tool.model_dump().items():
+            if hasattr(self, key):
+                setattr(self, key, value)
+        return self
+
+    def delete(self) -> None:
+        """Delete the tool."""
+        if not self._client:
+            raise RuntimeError(
+                "No client available. Use client.get_tool_by_id() to get a client-connected tool."
+            )
+        self._client.tools.delete_tool(self.id)
+
+
+class MCP(BaseModel):
+    """MCP model for API responses."""
+
+    id: str
+    name: str
+    description: str | None = None
+    config: dict[str, Any] | None = None
+    framework: str | None = None
+    version: str | None = None
+    transport: str | None = None  # "sse" or "http"
+    authentication: dict[str, Any] | None = None
+    metadata: dict[str, Any] | None = None
+    _client: Any = None  # Will hold client reference
+
+    def _set_client(self, client):
+        """Set the client reference for this resource."""
+        self._client = client
+        return self
+
+    def get_tools(self) -> list[dict[str, Any]]:
+        """Get tools available from this MCP."""
+        if not self._client:
+            raise RuntimeError(
+                "No client available. Use client.get_mcp_by_id() to get a client-connected MCP."
+            )
+        # This would delegate to the client's MCP tools endpoint
+        # For now, return empty list as placeholder
+        return []
+
+    def update(self, **kwargs) -> "MCP":
+        """Update MCP attributes."""
+        if not self._client:
+            raise RuntimeError(
+                "No client available. Use client.get_mcp_by_id() to get a client-connected MCP."
+            )
+        updated_mcp = self._client.update_mcp(self.id, **kwargs)
+        # Update current instance with new data
+        for key, value in updated_mcp.model_dump().items():
+            if hasattr(self, key):
+                setattr(self, key, value)
+        return self
+
+    def delete(self) -> None:
+        """Delete the MCP."""
+        if not self._client:
+            raise RuntimeError(
+                "No client available. Use client.get_mcp_by_id() to get a client-connected MCP."
+            )
+        self._client.delete_mcp(self.id)
+
+
+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):
+        self.use_color = use_color
+
+    def render_message(self, message: str, event_type: str = "message"):
+        """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/utils/__init__.py

@@ -0,0 +1,95 @@
+"""Utility modules for AIP SDK.
+
+Authors:
+    Raymond Christopher (raymond.christopher@gdplabs.id)
+"""
+
+import re
+from uuid import UUID
+
+from .run_renderer import (
+    RichStreamRenderer,
+    RunStats,
+    Step,
+    StepManager,
+)
+
+__all__ = [
+    "RichStreamRenderer",
+    "RunStats",
+    "Step",
+    "StepManager",
+    "is_uuid",
+    "sanitize_name",
+    "format_file_size",
+    "progress_bar",
+]
+
+
+def is_uuid(value: str) -> bool:
+    """Check if a string is a valid UUID.
+
+    Args:
+        value: String to check
+
+    Returns:
+        True if value is a valid UUID, False otherwise
+    """
+    try:
+        UUID(value)
+        return True
+    except (ValueError, TypeError):
+        return False
+
+
+def sanitize_name(name: str) -> str:
+    """Sanitize a name for resource creation.
+
+    Args:
+        name: Raw name input
+
+    Returns:
+        Sanitized name suitable for resource creation
+    """
+    # Remove special characters and normalize
+    sanitized = re.sub(r"[^a-zA-Z0-9\-_]", "-", name.strip())
+    sanitized = re.sub(r"-+", "-", sanitized)  # Collapse multiple dashes
+    return sanitized.lower().strip("-")
+
+
+def format_file_size(size_bytes: int) -> str:
+    """Format file size in human readable format.
+
+    Args:
+        size_bytes: Size in bytes
+
+    Returns:
+        Human readable size string (e.g., "1.5 MB")
+    """
+    for unit in ["B", "KB", "MB", "GB"]:
+        if size_bytes < 1024.0:
+            return f"{size_bytes:.1f} {unit}"
+        size_bytes /= 1024.0
+    return f"{size_bytes:.1f} TB"
+
+
+def progress_bar(iterable, description: str = "Processing"):
+    """Simple progress bar using click.
+
+    Args:
+        iterable: Iterable to process
+        description: Progress description
+
+    Yields:
+        Items from iterable with progress display
+    """
+    try:
+        import click
+
+        with click.progressbar(iterable, label=description) as bar:
+            for item in bar:
+                yield item
+    except ImportError:
+        # Fallback if click not available
+        for item in iterable:
+            yield item