refactored: to utilise the google adk and production grade agent

app/nodes/design_node.py

@@ -4,6 +4,7 @@ import logging
 from typing import Any, Dict
 from app.states.state import GCPArchitectureState
 from app.skills.loader import SkillLoader
+from app.tools.mcp_developer_knowledge import get_mcp_client
 
 logger = logging.getLogger(__name__)
 
@@ -13,12 +14,18 @@ def design_node(state: GCPArchitectureState, skill_loader: SkillLoader) -> Dict[
     logger.info("Executing design_node")
     skill_prompt = skill_loader.format_skills_for_prompt("design")
 
-    architecture_doc = """# Phase 1 — Architecture & Product Selection
+    # Developer Knowledge MCP Grounding Search
+    mcp_client = get_mcp_client()
+    mcp_knowledge = mcp_client.search_documents(query="Cloud Run PubSub event-driven architecture", category="compute")
+    docs = mcp_knowledge.get("documents", [])
+    citations_str = "\n".join([f"- [{doc['title']}]({doc['uri']})" for doc in docs[:3]]) if docs else "- [Google Cloud Architecture Framework](https://cloud.google.com/architecture/framework)"
+
+    architecture_doc = f"""# Phase 1 — Architecture & Product Selection
 
 ## Selected Products
 - **Compute / Serving**: Google Cloud Run (Fully Managed Container Ingress & Stateless Execution)
 - **Messaging & Eventing**: Google Cloud Pub/Sub (Regional Event Bus for Asynchronous Decoupling)
-- **State & Storage**: Google Cloud Storage (Bucket Storage for Durable Audit Event Replay)
+- **State & Storage**: Google Cloud Storage & Firestore (Database & Bucket Storage for Durable Audit Event Replay)
 - **Security & Identity**: Cloud IAM (Least Privilege Service Accounts) & KMS (Customer-Managed Encryption Keys)
 - **Artifact Registry**: Google Artifact Registry (OCI Container Image Hosting)
 
@@ -31,9 +38,12 @@ def design_node(state: GCPArchitectureState, skill_loader: SkillLoader) -> Dict[
 - HTTPS ingress with TLS 1.3 encryption in transit.
 - Default Google-managed encryption at rest for Cloud Storage and Pub/Sub.
 - Cloud Run service account bound strictly to `roles/pubsub.publisher` and `roles/storage.objectCreator`.
+
+## Grounded Documentation Citations (Google Developer Knowledge MCP)
+{citations_str}
 """
 
-    mermaid_diagram = """graph TD
+    mermaid_diagram = """flowchart TD
     Client[External HTTPS Client] -->|HTTPS POST /events| CloudRun[Google Cloud Run Service]
     CloudRun -->|Publish Event| PubSubTopic[Cloud Pub/Sub Topic]
     CloudRun -->|Write Raw Payload| GCSAudit[Cloud Storage Audit Bucket]
@@ -57,6 +67,21 @@ provider "google" {
   region  = var.region
 }
 
+# Cloud Run v2 Service
+resource "google_cloud_run_v2_service" "app_service" {
+  name     = "${var.environment}-app-service"
+  location = var.region
+
+  template {
+    containers {
+      image = var.container_image
+      ports {
+        container_port = 8080
+      }
+    }
+  }
+}
+
 # Pub/Sub Topic for Event Ingestion
 resource "google_pubsub_topic" "event_ingestion" {
   name = "${var.environment}-event-ingestion-topic"