cloudwright-ai 0.3.0__tar.gz → 0.3.2__tar.gz

{cloudwright_ai-0.3.0 → cloudwright_ai-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cloudwright-ai
-Version: 0.3.0
+Version: 0.3.2
 Summary: Architecture intelligence for cloud engineers — design, cost, compare, export
 Project-URL: Homepage, https://github.com/xmpuspus/cloudwright
 Project-URL: Repository, https://github.com/xmpuspus/cloudwright
@@ -22,19 +22,22 @@ Requires-Dist: openai<3,>=1.50
 Requires-Dist: pydantic<3,>=2.11
 Requires-Dist: pyyaml<7,>=6.0
 Provides-Extra: all
-Requires-Dist: cloudwright-ai-cli==0.3.0; extra == 'all'
-Requires-Dist: cloudwright-ai-web==0.3.0; extra == 'all'
+Requires-Dist: cloudwright-ai-cli==0.3.1; extra == 'all'
+Requires-Dist: cloudwright-ai-mcp==0.3.1; extra == 'all'
+Requires-Dist: cloudwright-ai-web==0.3.1; extra == 'all'
 Requires-Dist: databricks-sdk>=0.38.0; extra == 'all'
 Provides-Extra: cli
-Requires-Dist: cloudwright-ai-cli==0.3.0; extra == 'cli'
+Requires-Dist: cloudwright-ai-cli==0.3.1; extra == 'cli'
 Provides-Extra: databricks
 Requires-Dist: databricks-sdk>=0.38.0; extra == 'databricks'
+Provides-Extra: mcp
+Requires-Dist: cloudwright-ai-mcp==0.3.1; extra == 'mcp'
 Provides-Extra: pdf
 Requires-Dist: markdown2; extra == 'pdf'
 Requires-Dist: weasyprint; extra == 'pdf'
 Provides-Extra: web
-Requires-Dist: cloudwright-ai-cli==0.3.0; extra == 'web'
-Requires-Dist: cloudwright-ai-web==0.3.0; extra == 'web'
+Requires-Dist: cloudwright-ai-cli==0.3.1; extra == 'web'
+Requires-Dist: cloudwright-ai-web==0.3.1; extra == 'web'
 Description-Content-Type: text/markdown
 
 # Cloudwright

{cloudwright_ai-0.3.0 → cloudwright_ai-0.3.2}/cloudwright/__init__.py

@@ -17,7 +17,7 @@ from cloudwright.spec import (
     ValidationResult,
 )
 
-__version__ = "0.3.0"
+__version__ = "0.3.2"
 
 __all__ = [
     "Alternative",

{cloudwright_ai-0.3.0 → cloudwright_ai-0.3.2}/cloudwright/exporter/__init__.py

@@ -9,7 +9,7 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from cloudwright.spec import ArchSpec
 
-FORMATS = ("terraform", "cloudformation", "mermaid", "d2", "svg", "png", "c4", "sbom", "aibom", "compliance")
+FORMATS = ("terraform", "cloudformation", "mermaid", "d2", "ascii", "svg", "png", "c4", "sbom", "aibom", "compliance")
 
 
 class ExporterPlugin(ABC):
@@ -74,6 +74,14 @@ def export_spec(spec: ArchSpec, fmt: str, output: str | None = None, output_dir:
             Path(output).write_text(content)
         return content
 
+    if fmt in ("ascii", "text"):
+        from cloudwright.exporter.ascii import render
+
+        content = render(spec)
+        if output:
+            Path(output).write_text(content)
+        return content
+
     if fmt == "c4":
         from cloudwright.exporter.c4 import render
 

cloudwright_ai-0.3.2/cloudwright/exporter/ascii.py

@@ -0,0 +1,107 @@
+"""Render an ArchSpec as an ASCII architecture diagram for terminal display."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from cloudwright.spec import ArchSpec
+
+_TIER_LABELS = {
+    0: "Edge",
+    1: "Ingress",
+    2: "Compute",
+    3: "Data",
+    4: "Storage",
+}
+
+_WIDTH = 72
+_TIER_INNER = _WIDTH - 8  # content area inside tier boxes
+_MAX_CELL = 20  # max width per component cell
+
+
+def render(spec: "ArchSpec") -> str:
+    """Render an ArchSpec as a boxed ASCII architecture diagram."""
+    tiers: dict[int, list[tuple[str, str]]] = {}
+    for c in spec.components:
+        tiers.setdefault(c.tier, []).append((c.service, c.label))
+
+    lines: list[str] = []
+    inner = _WIDTH - 4  # inside the outer box
+
+    # Header
+    lines.append("+" + "=" * (_WIDTH - 2) + "+")
+    lines.append(_pad(f"  {spec.name}", inner))
+    region = spec.region or ""
+    provider = (spec.provider or "").upper()
+    meta = f"  {provider} / {region}  --  {len(spec.components)} services  --  {len(spec.connections)} connections"
+    lines.append(_pad(meta, inner))
+    lines.append("+" + "=" * (_WIDTH - 2) + "+")
+
+    # Tiers
+    sorted_tiers = sorted(tiers.keys())
+    for i, tier_num in enumerate(sorted_tiers):
+        components = tiers[tier_num]
+        tier_label = _TIER_LABELS.get(tier_num, f"Tier {tier_num}")
+
+        lines.append(_pad("", inner))
+
+        # Tier box top with label
+        box_w = inner - 4  # tier box width
+        label_part = f"-- {tier_label} "
+        dash_rest = "-" * (box_w - len(label_part) - 1)
+        lines.append(_pad(f"  +{label_part}{dash_rest}+", inner))
+
+        # Component rows (fit within tier box)
+        for row_start in range(0, len(components), 3):
+            row = components[row_start : row_start + 3]
+            cells = []
+            for service, _label in row:
+                cells.append(f"[{service}]")
+            row_text = "  |  " + "  ".join(cells)
+            # Pad to tier box width
+            pad_needed = box_w - len(row_text) + 2
+            if pad_needed > 0:
+                row_text += " " * pad_needed + "|"
+            else:
+                row_text = row_text[: box_w + 2] + "|"
+            lines.append(_pad(row_text, inner))
+
+        # Tier box bottom
+        lines.append(_pad("  +" + "-" * box_w + "+", inner))
+
+        # Arrow between tiers
+        if i < len(sorted_tiers) - 1:
+            mid = inner // 2
+            lines.append(_pad(" " * mid + "|", inner))
+            lines.append(_pad(" " * mid + "v", inner))
+
+    lines.append(_pad("", inner))
+
+    # Connection summary
+    if spec.connections:
+        lines.append(_pad("  Connections:", inner))
+        shown = min(6, len(spec.connections))
+        for conn in spec.connections[:shown]:
+            arrow = f"    {conn.source} --> {conn.target}"
+            if conn.label:
+                max_label = inner - len(arrow) - 5
+                lbl = conn.label if len(conn.label) <= max_label else conn.label[: max_label - 2] + ".."
+                arrow += f"  ({lbl})"
+            if len(arrow) > inner:
+                arrow = arrow[: inner - 3] + "..."
+            lines.append(_pad(arrow, inner))
+        if len(spec.connections) > shown:
+            lines.append(_pad(f"    ... +{len(spec.connections) - shown} more", inner))
+
+    lines.append(_pad("", inner))
+    lines.append("+" + "=" * (_WIDTH - 2) + "+")
+
+    return "\n".join(lines)
+
+
+def _pad(text: str, inner: int) -> str:
+    """Wrap text in outer box borders, padded to fixed width."""
+    if len(text) > inner:
+        text = text[:inner]
+    return "| " + text + " " * (inner - len(text)) + " |"

{cloudwright_ai-0.3.0 → cloudwright_ai-0.3.2}/cloudwright/llm/base.py

@@ -24,3 +24,7 @@ class BaseLLM(ABC):
     @abstractmethod
     def generate_stream(self, messages: list[dict], system: str, max_tokens: int = 2000) -> Iterator[str]:
         """Stream generation using the capable model. Yields text chunks."""
+
+    def estimate_tokens(self, text: str) -> int:
+        """Rough token count estimate (~4 chars per token for English)."""
+        return len(text) // 4

{cloudwright_ai-0.3.0 → cloudwright_ai-0.3.2}/pyproject.toml

@@ -28,9 +28,10 @@ dependencies = [
 ]
 
 [project.optional-dependencies]
-cli = ["cloudwright-ai-cli==0.3.0"]
-web = ["cloudwright-ai-cli==0.3.0", "cloudwright-ai-web==0.3.0"]
-all = ["cloudwright-ai-cli==0.3.0", "cloudwright-ai-web==0.3.0", "databricks-sdk>=0.38.0"]
+cli = ["cloudwright-ai-cli==0.3.1"]
+web = ["cloudwright-ai-cli==0.3.1", "cloudwright-ai-web==0.3.1"]
+mcp = ["cloudwright-ai-mcp==0.3.1"]
+all = ["cloudwright-ai-cli==0.3.1", "cloudwright-ai-web==0.3.1", "cloudwright-ai-mcp==0.3.1", "databricks-sdk>=0.38.0"]
 pdf = ["weasyprint", "markdown2"]
 databricks = ["databricks-sdk>=0.38.0"]