ai-accelerator 0.1.0__py3-none-any.whl

aiac/governance/commands.py

@@ -0,0 +1,150 @@
+import typer
+from aiac.client import AIACClient
+from aiac.console import console
+from rich.table import Table
+
+
+
+table = Table(title="Governance Policies")
+
+governance_api_app = typer.Typer(help="Governance related commands for AIAC.")
+
+@governance_api_app.command("create-policy")
+def create_policy(name: str, policy_type: str, description: str = "", rules: str = "{}"):
+    """Create a new governance policy."""
+
+    client = AIACClient()
+
+    try:
+        # Parse rules if it's a string
+        if isinstance(rules, str):
+            import json
+            rules = json.loads(rules)
+
+        data = {
+            "name": name,
+            "policy_type": policy_type,
+            "description": description,
+            "rules": rules
+        }
+
+        response = client.api_request("/policies/", method="POST", data=data)
+        typer.echo(f"Policy '{name}' created successfully.")
+    except Exception as e:
+        typer.echo(f"Failed to create policy: {str(e)}")
+
+@governance_api_app.command("list-policies")
+def list_policies():
+    """List all governance policies."""
+
+    client = AIACClient()
+    try:
+        policies = client.api_request("/policies/")
+
+        table = Table(title="Governance Policies")
+        table.add_column("ID", style="cyan", no_wrap=True)
+        table.add_column("Name", style="magenta")
+        table.add_column("Type", style="green")
+        table.add_column("Description")
+
+        for policy in policies:
+            table.add_row(
+                str(policy['id']),
+                policy['name'],
+                policy['policy_type'],
+                policy.get('description', '')
+            )
+
+        console.print(table)
+    except Exception as e:
+        typer.echo(f"Failed to retrieve policies: {str(e)}")
+
+@governance_api_app.command("delete-policy")
+def delete_policy(policy_id: int):
+    
+    """Delete a governance policy by its ID."""
+
+    client = AIACClient()
+    response = client.delete(f"/policies/{policy_id}/")
+
+    if response.status_code == 204:
+        typer.echo(f"Policy with ID '{policy_id}' deleted successfully.")
+    else:
+        typer.echo(f"Failed to delete policy. Status code: {response.status_code}, Response: {response.text}")
+
+@governance_api_app.command("view-violations")
+def view_violations():
+    """View all policy violations."""
+
+    client = AIACClient()
+    response = client.get("/policy-violations/")
+
+    if response.status_code == 200:
+        violations = response.json()
+        for violation in violations:
+            table.add_column("Deployment_ID", "Policy_Name", "Violation_Type", "Severity","Resolved", justify="right", style="cyan", no_wrap=True)
+            table.add_row(str(violation['deployment_id']), str(violation['policy_name']), str(violation['violation_type']), str(violation['severity']), str(violation['resolved']))
+        console.print(table)
+    else:
+        typer.echo(f"Failed to retrieve violations. Status code: {response.status_code}, Response: {response.text}")
+
+@governance_api_app.command("metrics")
+def violation_metrics():
+    """View metrics related to policy violations."""
+
+    client = AIACClient()
+    response = client.get("/policy-violations/")
+
+    if response.status_code == 200:
+        violations = response.json()
+        if violations:
+            for metrics in violations:
+                typer.echo("Violation Metrics:{metrics[policy_name]} --- {metrics[violation_type]}")
+                for key, value in metrics.items():
+                    typer.echo(f"{key}: {value}")
+                typer.echo("-------------------------------------------------------------------------")
+        else:
+            typer.echo("No violations found.")
+    else:
+        typer.echo(f"Failed to retrieve violation metrics. Status code: {response.status_code}, Response: {response.text}")
+
+
+@governance_api_app.command("apply-policy")
+def apply_policy(policy_id: int, deployment_id: int):
+    """Apply a policy to a deployment."""
+    
+
+    client = AIACClient()
+    data = {
+        "policy": policy_id,
+        "deployment": deployment_id
+    }
+
+    response = client.post("/policy-assignments/", json=data)
+
+    if response.status_code == 201:
+        typer.echo(f"Policy '{policy_id}' applied to deployment '{deployment_id}' successfully.")
+        typer.echo(f"applied_at: {response.json().get('applied_at')}")
+        typer.echo(f"applied_by: {response.json().get('applied_by')}")
+    else:
+        typer.echo(f"Failed to apply policy. Status code: {response.status_code}, Response: {response.text}")
+
+
+
+
+@governance_api_app.command("alert-logs")
+def alert_logs():
+    """View alert logs for policy violations."""
+
+    client = AIACClient()
+    response = client.get("/alerts/")
+
+    if response.status_code == 200:
+        typer.echo("Alert Logs:")
+        logs = response.json()
+        for log in logs:
+            table.add_column("Policy_Violation", "Message","Sent", justify="right", style="cyan", no_wrap=True)
+            table.add_row(str(log['policy_violation']), str(log['message']), str(log['sent']))
+        console.print(table)
+    else:
+        typer.echo(f"Failed to retrieve alert logs. Status code: {response.status_code}, Response: {response.text}")

aiac/main.py

@@ -0,0 +1,8 @@
+from aiac.deployment.commands import api_app_deployment as deployment_app
+from aiac.monitoring.commands import monitoring_api_app as monitoring_app
+from aiac.governance.commands import governance_api_app as governance_app
+
+def register_commands(main_app):
+    main_app.add_typer(governance_app, name="governance", help="Governance related commands for AIAC.")
+    main_app.add_typer(deployment_app, name="deployment", help="Deployment related commands for AIAC.")
+    main_app.add_typer(monitoring_app , name="monitoring", help="Monitoring related commands for AIAC.")

aiac/models.py

@@ -0,0 +1,3 @@
+from django.db import models
+
+# Create your models here.

aiac/monitoring/commands.py

@@ -0,0 +1,141 @@
+import typer
+from aiac.client import AIACClient
+from  aiac.console import console , print_message , print_warning  , print_info
+from rich.table import Table
+
+
+table = Table()
+
+monitoring_api_app = typer.Typer(help = "monitoring commands")
+
+@monitoring_api_app.command("deploy-stats")
+def deployment_stats(deployment_id: int = typer.Option(..., prompt=True, help="deployment stats")):
+    client = AIACClient()
+
+    try:
+        response = client.api_request(f"deployments/{deployment_id}/stats/", method="GET")
+        print_message(f"Fetching stats for deployment_id: {deployment_id}")
+
+        table = Table(title="Deployment Stats")
+        table.add_column("Deployment", style="cyan")
+        table.add_column("CPU Usage", style="green")
+        table.add_column("RAM Usage", style="yellow")
+        table.add_column("Latency (ms)", style="red")
+        table.add_column("Request Count", style="blue")
+        table.add_column("Error Count", style="magenta")
+        table.add_column("Updated At", style="white")
+
+        # Assuming response is a single stats object
+        table.add_row(
+            str(deployment_id),
+            str(response.get('cpu_usage', 'N/A')),
+            str(response.get('ram_usage', 'N/A')),
+            str(response.get('latency_ms', 'N/A')),
+            str(response.get('request_count', 'N/A')),
+            str(response.get('error_count', 'N/A')),
+            str(response.get('updated_at', 'N/A'))
+        )
+
+        console.print(table)
+    except Exception as e:
+        typer.echo(f"Failed to fetch stats for deployment_id {deployment_id}: {str(e)}")
+
+@monitoring_api_app.command("deploy-records")
+def deployment_records(deployment_id: int = typer.Option(..., prompt=True, help="deployment records")):
+    client = AIACClient()
+
+    try:
+        records = client.api_request(f"deployments/{deployment_id}/records/", method="GET")
+        print_message(f"Fetching records for deployment_id: {deployment_id}")
+
+        table = Table(title="Deployment Records")
+        table.add_column("ID", style="cyan", no_wrap=True)
+        table.add_column("Deployment", style="magenta")
+        table.add_column("CPU Usage", style="green")
+        table.add_column("RAM Usage", style="yellow")
+        table.add_column("Latency (ms)", style="red")
+        table.add_column("Request Count", style="blue")
+        table.add_column("Error Count", style="magenta")
+        table.add_column("Created At", style="white")
+
+        # Assuming records is a list of record objects
+        for record in records if isinstance(records, list) else [records]:
+            table.add_row(
+                str(record.get('id', 'N/A')),
+                str(deployment_id),
+                str(record.get('cpu_usage', 'N/A')),
+                str(record.get('ram_usage', 'N/A')),
+                str(record.get('latency_ms', 'N/A')),
+                str(record.get('request_count', 'N/A')),
+                str(record.get('error_count', 'N/A')),
+                str(record.get('created_at', 'N/A'))
+            )
+
+        console.print(table)
+    except Exception as e:
+        typer.echo(f"Failed to fetch records for deployment_id {deployment_id}: {str(e)}")
+
+@monitoring_api_app.command("alert")
+def receive_metrics(deployment_id: int = typer.Option(..., prompt=True, help="deployment alerts")):
+    client = AIACClient()
+
+    try:
+        alerts = client.api_request(f"deployments/{deployment_id}/alerts/", method="GET")
+        print_warning(f"Fetching alerts for deployment_id: {deployment_id}")
+
+        table = Table(title="Deployment Alerts")
+        table.add_column("ID", style="cyan", no_wrap=True)
+        table.add_column("Deployment", style="magenta")
+        table.add_column("Alert Type", style="red")
+        table.add_column("Message", style="yellow")
+        table.add_column("Created At", style="green")
+        table.add_column("Resolved", style="blue")
+
+        # Assuming alerts is a list of alert objects
+        for alert in alerts if isinstance(alerts, list) else [alerts]:
+            table.add_row(
+                str(alert.get('id', 'N/A')),
+                str(deployment_id),
+                str(alert.get('alert_type', 'N/A')),
+                str(alert.get('message', 'N/A')),
+                str(alert.get('created_at', 'N/A')),
+                str(alert.get('resolved', 'N/A'))
+            )
+
+        console.print(table)
+    except Exception as e:
+        typer.echo(f"Failed to fetch alerts for deployment_id {deployment_id}: {str(e)}")
+
+@monitoring_api_app.command("resolve-alert")
+def resolve_alert(alert_id: str = typer.Option(..., prompt=True, help="resolve alert by id")):
+    client = AIACClient()
+
+    try:
+        response = client.api_request(f"alerts/{alert_id}/resolve/", method="POST")
+        print_info(f"Resolving alert with alert_id: {alert_id}")
+        typer.echo(f"Alert resolved successfully: {response}")
+    except Exception as e:
+        typer.echo(f"Failed to resolve alert with alert_id {alert_id}: {str(e)}")
+
+@monitoring_api_app.command("detect-drift")
+def detect_drift(model_version_id: int = typer.Option(..., prompt=True, help="detect data drift for model version")):
+    client = AIACClient()
+
+    try:
+        response = client.api_request(f"drifts/{model_version_id}/", method="GET")
+        print_info(f"Detecting drift for model_version_id: {model_version_id}")
+        typer.echo(f"Drift detection result: {response}")
+    except Exception as e:
+        typer.echo(f"Failed to detect drift for model_version_id {model_version_id}: {str(e)}")
+
+@monitoring_api_app.command("samples")
+def get_samples(model_version_id: int = typer.Option(..., prompt=True, help="post samples for model version")):
+    client = AIACClient()
+
+    try:
+        response = client.api_request("deployments/samples/", method="POST", data={"model_version_id": model_version_id})
+        print_info(f"Posting samples for model_version_id: {model_version_id}")
+        typer.echo(f"Samples posted successfully: {response}")
+    except Exception as e:
+        typer.echo(f"Failed to post samples for model_version_id {model_version_id}: {str(e)}")
+       

aiac/requirments.txt

@@ -0,0 +1,4 @@
+typer>=0.9.0
+rich>=13.0.0
+requests>=2.28.0
+click>=8.0

aiac/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

aiac/urls.py

@@ -0,0 +1,4 @@
+from django.urls import path
+urlpatterns = [
+    # Define your user-related URL patterns here
+]

aiac/user/auth.py

@@ -0,0 +1,52 @@
+import typer , requests
+from aiac.config import get_config , save_config
+
+auth_app = typer.Typer(help="user authentication commands")
+
+@auth_app.command("register")
+def register(username: str = typer.Option(..., prompt=True, help="Username for registration"),
+             password: str = typer.Option(..., prompt=True, hide_input=True, help="Password for registration"),
+             email: str = typer.Option(..., prompt=True, help="Email address for registration"),
+             role: str = typer.Option(..., prompt=True, help="Role of the user (client or admin)")):
+
+    "Register a new user"
+    config = get_config()
+    url = f"{config.api_base_url}/register/"
+    data = {"username": username, "password": password, "email": email, "role": role}
+    response = requests.post(url, json=data)
+    if response.status_code == 201:
+        typer.echo("Registration successful! Please check your email to activate your account.")
+    else:
+        typer.echo(f"Registration failed: {response.text}")
+
+@auth_app.command("login")
+def login(email: str = typer.Option(..., prompt=True, help="Email address for login"),
+          password: str = typer.Option(..., prompt=True, hide_input=True, help="Password for login")):
+
+    "Login a user"
+    config = get_config()
+    url = f"{config.api_base_url}/login/"
+    data = {"email": email, "password": password}
+    response = requests.post(url, json=data)
+
+    if response.status_code == 200:
+        access = response.json().get("access")
+        refresh = response.json().get("refresh")
+        save_config({"access": access, "refresh": refresh}, config.api_base_url)
+        typer.echo(f"Login successful! Tokens saved.")
+    else:
+        typer.echo(f"Login failed: {response.text}")
+
+@auth_app.command("logout")
+def logout(refresh_token: str = typer.Option(..., prompt=True, hide_input=True, help="Refresh token to logout")):
+
+    "Logout a user"
+    config = get_config()
+    url = f"{config.api_base_url}/logout/"
+    data = {"refresh": refresh_token}
+    response = requests.post(url, json=data)
+    if response.status_code == 200:
+        typer.echo("Logout successful!")
+    else:
+        typer.echo(f"Logout failed: {response.text}")
+

aiac/views.py

@@ -0,0 +1,3 @@
+from django.shortcuts import render
+
+# Create your views here.