cloudwright-ai-cli 0.3.1__tar.gz → 0.3.2__tar.gz

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cloudwright-ai-cli
-Version: 0.3.1
+Version: 0.3.2
 Summary: CLI for Cloudwright architecture intelligence
 Project-URL: Homepage, https://github.com/xmpuspus/cloudwright
 Project-URL: Repository, https://github.com/xmpuspus/cloudwright

cloudwright_ai_cli-0.3.2/cloudwright_cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.3.2"

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/commands/adr.py

@@ -71,13 +71,16 @@ def adr(
             from cloudwright.llm.anthropic import GENERATE_MODEL
 
             spec_json = spec.model_dump_json(indent=2, exclude_none=True)
-            emit_dry_run(ctx, {
-                "model": GENERATE_MODEL,
-                "estimated_tokens": len(spec_json + _ADR_SYSTEM) // 4,
-                "max_tokens": 2000,
-                "system_prompt_preview": _ADR_SYSTEM,
-                "user_prompt_preview": f"Generate ADR for: {spec.name}",
-            })
+            emit_dry_run(
+                ctx,
+                {
+                    "model": GENERATE_MODEL,
+                    "estimated_tokens": len(spec_json + _ADR_SYSTEM) // 4,
+                    "max_tokens": 2000,
+                    "system_prompt_preview": _ADR_SYSTEM,
+                    "user_prompt_preview": f"Generate ADR for: {spec.name}",
+                },
+            )
 
         text = _generate_adr(spec, title=title, decision=decision)
 

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/commands/cost.py

@@ -21,14 +21,33 @@ def cost(
     pricing_tier: Annotated[
         str | None, typer.Option(help="Pricing tier (on_demand, reserved_1yr, reserved_3yr, spot)")
     ] = None,
+    workload_profile: Annotated[
+        str | None,
+        typer.Option(
+            "--workload-profile",
+            "-w",
+            help="Workload sizing profile (small, medium, large, enterprise). "
+            "Sets realistic defaults for request volumes, storage, node counts, and data transfer.",
+        ),
+    ] = None,
 ) -> None:
     """Show cost breakdown for an architecture spec."""
+    if workload_profile:
+        from cloudwright.cost import VALID_WORKLOAD_PROFILES
+
+        if workload_profile not in VALID_WORKLOAD_PROFILES:
+            console.print(
+                f"[red]Invalid workload profile:[/red] {workload_profile!r}. "
+                f"Choose from: {', '.join(sorted(VALID_WORKLOAD_PROFILES))}"
+            )
+            raise typer.Exit(1)
+
     spec = ArchSpec.from_file(spec_file)
 
     # Compute cost estimate if not present
     if not spec.cost_estimate:
         engine = CostEngine()
-        spec.cost_estimate = engine.estimate(spec)
+        spec.cost_estimate = engine.estimate(spec, workload_profile=workload_profile)
 
     if is_json_mode(ctx):
         emit_success(ctx, {"estimate": spec.cost_estimate.model_dump(exclude_none=True)})

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/commands/design.py

@@ -45,14 +45,17 @@ def design(
         system = Architect._select_system_prompt(description)
         if constraints:
             system += _build_constraint_prompt(constraints)
-        emit_dry_run(ctx, {
-            "model": GENERATE_MODEL,
-            "estimated_tokens": len(system + description) // 4,
-            "max_tokens": 10000,
-            "system_prompt_preview": system[:200],
-            "user_prompt_preview": description,
-            "constraints": constraints.model_dump(exclude_none=True),
-        })
+        emit_dry_run(
+            ctx,
+            {
+                "model": GENERATE_MODEL,
+                "estimated_tokens": len(system + description) // 4,
+                "max_tokens": 10000,
+                "system_prompt_preview": system[:200],
+                "user_prompt_preview": description,
+                "constraints": constraints.model_dump(exclude_none=True),
+            },
+        )
 
     try:
         architect = Architect()

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/commands/lint_cmd.py

@@ -32,13 +32,15 @@ def lint(
         if is_json_mode(ctx):
             if should_stream(ctx):
                 for w in warnings:
-                    emit_stream({
-                        "rule": w.rule,
-                        "severity": w.severity,
-                        "component": w.component,
-                        "message": w.message,
-                        "recommendation": w.recommendation,
-                    })
+                    emit_stream(
+                        {
+                            "rule": w.rule,
+                            "severity": w.severity,
+                            "component": w.component,
+                            "message": w.message,
+                            "recommendation": w.recommendation,
+                        }
+                    )
                 return
             result = [
                 {

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/commands/mcp_cmd.py

@@ -21,8 +21,7 @@ def mcp_serve(
         from rich.console import Console
 
         Console(stderr=True).print(
-            "[red]Error:[/red] cloudwright-ai-mcp not installed.\n"
-            "  Install: pip install cloudwright-ai-mcp"
+            "[red]Error:[/red] cloudwright-ai-mcp not installed.\n  Install: pip install cloudwright-ai-mcp"
         )
         raise typer.Exit(1) from None
 

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/commands/modify_cmd.py

@@ -50,13 +50,16 @@ def modify(
             from cloudwright.llm.anthropic import GENERATE_MODEL
 
             spec_text = original.to_yaml()
-            emit_dry_run(ctx, {
-                "model": GENERATE_MODEL,
-                "estimated_tokens": len(spec_text + instruction) // 4,
-                "max_tokens": 8000,
-                "user_prompt_preview": f"Modify: {instruction}",
-                "constraints": {"spec_file": spec_file, "instruction": instruction},
-            })
+            emit_dry_run(
+                ctx,
+                {
+                    "model": GENERATE_MODEL,
+                    "estimated_tokens": len(spec_text + instruction) // 4,
+                    "max_tokens": 8000,
+                    "user_prompt_preview": f"Modify: {instruction}",
+                    "constraints": {"spec_file": spec_file, "instruction": instruction},
+                },
+            )
 
         with console.status("Applying modification..."):
             modified = architect.modify(original, instruction)
@@ -73,11 +76,14 @@ def modify(
         diff_result = Differ().diff(original_costed, modified_costed)
 
         if is_json_mode(ctx):
-            emit_success(ctx, {
-                "original": original.model_dump(),
-                "modified": modified.model_dump(),
-                "diff": diff_result.model_dump(),
-            })
+            emit_success(
+                ctx,
+                {
+                    "original": original.model_dump(),
+                    "modified": modified.model_dump(),
+                    "diff": diff_result.model_dump(),
+                },
+            )
             return
 
         console.print(Rule("[bold]Changes[/bold]"))

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/commands/schema_cmd.py

@@ -47,8 +47,7 @@ def _show_service(ctx: typer.Context, query: str) -> None:
     if not svc_def:
         available = [s.service_key for s in registry.list_services(provider)[:15]]
         raise ValueError(
-            f"Service '{service_key}' not found for provider '{provider}'. "
-            f"Available: {', '.join(available)}"
+            f"Service '{service_key}' not found for provider '{provider}'. Available: {', '.join(available)}"
         )
 
     equivalents = {}
@@ -80,7 +79,9 @@ def _show_service(ctx: typer.Context, query: str) -> None:
         Panel(
             f"[bold]{svc_def.name}[/bold] ({provider}.{service_key})\n"
             f"Category: {svc_def.category}  |  Pricing: {svc_def.pricing_formula}\n"
-            f"{svc_def.description}" if svc_def.description else "",
+            f"{svc_def.description}"
+            if svc_def.description
+            else "",
             title="Service Schema",
         )
     )
@@ -147,10 +148,7 @@ def _show_compliance(ctx: typer.Context, framework: str) -> None:
     data = {
         "framework": result.framework,
         "total_checks": len(result.checks),
-        "checks": [
-            {"name": c.name, "category": c.category, "severity": c.severity}
-            for c in result.checks
-        ],
+        "checks": [{"name": c.name, "category": c.category, "severity": c.severity} for c in result.checks],
     }
 
     if is_json_mode(ctx):

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/commands/security_cmd.py

@@ -32,13 +32,15 @@ def security_scan(
         if is_json_mode(ctx):
             if should_stream(ctx):
                 for f in report.findings:
-                    emit_stream({
-                        "severity": f.severity,
-                        "rule": f.rule,
-                        "component_id": f.component_id,
-                        "message": f.message,
-                        "remediation": f.remediation,
-                    })
+                    emit_stream(
+                        {
+                            "severity": f.severity,
+                            "rule": f.rule,
+                            "component_id": f.component_id,
+                            "message": f.message,
+                            "remediation": f.remediation,
+                        }
+                    )
             else:
                 result = {
                     "passed": report.passed,

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/commands/validate.py

@@ -68,13 +68,15 @@ def validate(
         if should_stream(ctx):
             for result in results:
                 for check in result.checks:
-                    emit_stream({
-                        "framework": result.framework,
-                        "check": check.name,
-                        "passed": check.passed,
-                        "detail": check.detail,
-                        "recommendation": check.recommendation,
-                    })
+                    emit_stream(
+                        {
+                            "framework": result.framework,
+                            "check": check.name,
+                            "passed": check.passed,
+                            "detail": check.detail,
+                            "recommendation": check.recommendation,
+                        }
+                    )
         else:
             emit_success(ctx, {"results": [r.model_dump(exclude_none=True) for r in results]})
         return

{cloudwright_ai_cli-0.3.1 → cloudwright_ai_cli-0.3.2}/cloudwright_cli/completions.py

@@ -24,6 +24,26 @@ def complete_compliance(incomplete: str) -> list[tuple[str, str]]:
     return [(f, h) for f, h in frameworks if f.startswith(incomplete)]
 
 
+def complete_pricing_tier(incomplete: str) -> list[tuple[str, str]]:
+    tiers = [
+        ("on_demand", "Standard on-demand pricing"),
+        ("reserved_1yr", "1-year reserved (40% savings)"),
+        ("reserved_3yr", "3-year reserved (60% savings)"),
+        ("spot", "Spot/preemptible (70% savings)"),
+    ]
+    return [(t, h) for t, h in tiers if t.startswith(incomplete)]
+
+
+def complete_workload_profile(incomplete: str) -> list[tuple[str, str]]:
+    profiles = [
+        ("small", "Startup/dev — low traffic, minimal redundancy"),
+        ("medium", "Production — moderate traffic, multi-AZ databases"),
+        ("large", "Scale — high traffic, large clusters and storage"),
+        ("enterprise", "Enterprise — very high traffic, full redundancy"),
+    ]
+    return [(p, h) for p, h in profiles if p.startswith(incomplete)]
+
+
 def complete_export_format(incomplete: str) -> list[tuple[str, str]]:
     formats = [
         ("terraform", "HashiCorp Terraform HCL"),

cloudwright_ai_cli-0.3.1/cloudwright_cli/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.3.1"