refactored: to utilise the google adk and production grade agent

app/workflows/gcp_architecture_graph.py

@@ -6,7 +6,7 @@ from langgraph.graph import END, START, StateGraph
 from langgraph.graph.state import CompiledStateGraph
 
 from app.config import get_settings
-from app.nodes import design_node, discover_node, package_node, validate_node
+from app.nodes import design_node, discover_node, package_node, source_discover_node, validate_node
 from app.skills.loader import SkillLoader
 from app.states.state import GCPArchitectureState
 
@@ -14,7 +14,7 @@ logger = logging.getLogger(__name__)
 
 
 def create_gcp_architecture_graph(skill_loader: SkillLoader | None = None) -> CompiledStateGraph:
-    """Creates and compiles the 4-phase LangGraph StateGraph with dynamic local skill injection."""
+    """Creates and compiles the LangGraph StateGraph with dynamic local skill injection."""
     if skill_loader is None:
         settings = get_settings()
         skill_loader = SkillLoader(settings.SKILLS_DIR)
@@ -23,6 +23,9 @@ def create_gcp_architecture_graph(skill_loader: SkillLoader | None = None) -> Co
     builder = StateGraph(GCPArchitectureState)
 
     # Wrap nodes with skill loader injection
+    def run_source_discover(state: GCPArchitectureState) -> dict[str, Any]:
+        return source_discover_node(state, skill_loader)
+
     def run_discover(state: GCPArchitectureState) -> dict[str, Any]:
         return discover_node(state, skill_loader)
 
@@ -36,13 +39,15 @@ def create_gcp_architecture_graph(skill_loader: SkillLoader | None = None) -> Co
         return package_node(state, skill_loader)
 
     # Add graph nodes
+    builder.add_node("source_discover", run_source_discover)
     builder.add_node("discover", run_discover)
     builder.add_node("design", run_design)
     builder.add_node("validate", run_validate)
     builder.add_node("package", run_package)
 
     # Add sequential workflow edges
-    builder.add_edge(START, "discover")
+    builder.add_edge(START, "source_discover")
+    builder.add_edge("source_discover", "discover")
     builder.add_edge("discover", "design")
     builder.add_edge("design", "validate")
     builder.add_edge("validate", "package")