@salesforce/afv-skills 1.14.0 → 1.16.0

package/skills/investigating-agentforce-architecture/scripts/tests/fixtures/genai_payloads.py

@@ -0,0 +1,400 @@
+"""Synthetic GenAi query payloads for main pipeline integration tests.
+
+Phase 2 Batch 1: these fixtures model the shapes live Salesforce returns
+for each of the 7 Wave A + 3 Wave B SOQL calls. Field names come from
+`references/soql_fields.md` (live-describe verified on my-org-alias +
+my-perf-org-alias, 2026-05-02).
+
+Two shapes covered:
+  * CLASSIC_*  — MyAgent v5-shaped payloads (AiCopilot__* planner,
+    InvocationTarget is DeveloperName strings, PlannerId set on bundle-
+    scope functions).
+  * NGA_*      — MyAgent2 v3-shaped payloads (Atlas__* planner,
+    InvocationTarget is Salesforce Ids, PluginId set on every function).
+
+SEQUENTIAL_* is a third variant: classic PlannerType but 0 plugins, 1
+bundle-scope function — reproduces the SequentialPlannerIntentClassifier
+bot shape which tests often miss.
+"""
+from __future__ import annotations
+
+
+# ===========================================================================
+# Classic shape (MyAgent v5-style)
+# ===========================================================================
+
+CLASSIC_PLANNER = {
+    "Id": "1VxVF0000000001",
+    "DeveloperName": "MyAgent",
+    "MasterLabel": "Service Agent",
+    "Description": "Classic ReAct MyAgent",
+    "PlannerType": "AiCopilot__ReActAiPlannerV1",
+    "Capabilities": None,
+    "AgentGraph": None,
+}
+
+CLASSIC_PLUGINS = [
+    {
+        # Distinct Ids per plugin; the plugin→function join table keys on
+        # this Id, so collapsing them would fan actions across every topic.
+        "Id": f"1VyVF000000TOP{i}",
+        "DeveloperName": f"Topic{i}",
+        "MasterLabel": f"Topic {i}",
+        "Description": f"Topic {i} desc",
+        "PluginType": "Custom",
+        "Scope": f"Topic {i} scope",
+        "IsLocal": True,
+        "CanEscalate": False,
+        "Source": "Declarative",
+        "ParentId": "1VxVF0000000001",
+        "LocalDeveloperName": f"Topic{i}",
+    }
+    for i in range(1, 7)  # 6 topics → matches the "6 topics" assertion
+]
+
+# Classic functions: 2 bundle-scope (PlannerId set, PluginId null) +
+# 2 topic-scope (PluginId set under Topic1). Invocation targets are
+# DeveloperName strings (classic).
+CLASSIC_FUNCTIONS = [
+    # Bundle-scope
+    {
+        "Id": "1VuVF0000000B01",
+        "DeveloperName": "LookupOrderFlow",
+        "MasterLabel": "Lookup Order",
+        "Description": None,
+        "InvocationTargetType": "flow",
+        "InvocationTarget": "LookupOrder",
+        "IsLocal": False,
+        "IsConfirmationRequired": False,
+        "IsIncludeInProgressIndicator": False,
+        "ProgressIndicatorMessage": None,
+        "Source": "Declarative",
+        "PluginId": None,
+        "PlannerId": "1VxVF0000000001",
+        "ParentId": None,
+        "LocalDeveloperName": "LookupOrderFlow",
+    },
+    {
+        "Id": "1VuVF0000000B02",
+        "DeveloperName": "EscalateCase",
+        "MasterLabel": "Escalate Case",
+        "Description": None,
+        "InvocationTargetType": "apex",
+        "InvocationTarget": "CaseEscalationHandler",
+        "IsLocal": False,
+        "IsConfirmationRequired": True,
+        "IsIncludeInProgressIndicator": False,
+        "ProgressIndicatorMessage": None,
+        "Source": "Declarative",
+        "PluginId": None,
+        "PlannerId": "1VxVF0000000001",
+        "ParentId": None,
+        "LocalDeveloperName": "EscalateCase",
+    },
+    # Topic-scope (Topic1)
+    {
+        "Id": "1VuVF0000000T01",
+        "DeveloperName": "GetOrderDetails",
+        "MasterLabel": "Get Order Details",
+        "Description": None,
+        "InvocationTargetType": "flow",
+        "InvocationTarget": "GetOrderDetails",
+        "IsLocal": True,
+        "IsConfirmationRequired": False,
+        "IsIncludeInProgressIndicator": True,
+        "ProgressIndicatorMessage": "Looking up...",
+        "Source": "Declarative",
+        "PluginId": "1VyVF000000TOP1",
+        "PlannerId": None,
+        "ParentId": None,
+        "LocalDeveloperName": "GetOrderDetails",
+    },
+    {
+        "Id": "1VuVF0000000T02",
+        "DeveloperName": "ApplyDiscount",
+        "MasterLabel": "Apply Discount",
+        "Description": None,
+        "InvocationTargetType": "apex",
+        "InvocationTarget": "DiscountApplicator",
+        "IsLocal": True,
+        "IsConfirmationRequired": False,
+        "IsIncludeInProgressIndicator": False,
+        "ProgressIndicatorMessage": None,
+        "Source": "Declarative",
+        "PluginId": "1VyVF000000TOP1",
+        "PlannerId": None,
+        "ParentId": None,
+        "LocalDeveloperName": "ApplyDiscount",
+    },
+]
+
+CLASSIC_PLUGIN_FUNCTIONS = [
+    {"Id": "1VqVF000001", "PluginId": "1VyVF000000TOP1", "Function": "1VuVF0000000T01"},
+    {"Id": "1VqVF000002", "PluginId": "1VyVF000000TOP1", "Function": "1VuVF0000000T02"},
+]
+
+CLASSIC_BUNDLE_FN_JOIN = [
+    {"Id": "1VrVF000001", "PlannerId": "1VxVF0000000001", "Plugin": "LookupOrderFlow"},
+    {"Id": "1VrVF000002", "PlannerId": "1VxVF0000000001", "Plugin": "EscalateCase"},
+]
+
+CLASSIC_INSTRUCTIONS = [
+    {
+        "Id": "1VwVF000001",
+        "GenAiPluginDefinitionId": "1VyVF000000TOP1",
+        "DeveloperName": "Instruction1",
+        "MasterLabel": "Use polite tone",
+        "Description": "Always greet the user first.",
+        "SortOrder": 1,
+    }
+]
+
+CLASSIC_ATTRS = [
+    {
+        "Id": "1VvVF000001",
+        "ParentId": "1VuVF0000000T01",
+        "DeveloperName": "OrderId",
+        "MasterLabel": "Order Id",
+        "Description": None,
+        "MappingType": "input",
+        "ParameterName": "orderId",
+    }
+]
+
+# Wave B classic: 2 flows (LookupOrder, GetOrderDetails), 2 apex
+# (CaseEscalationHandler, DiscountApplicator).
+CLASSIC_FLOW_DEFS = [
+    {"Id": "300VF001", "DeveloperName": "LookupOrder", "ActiveVersionId": "301VF001"},
+    {"Id": "300VF002", "DeveloperName": "GetOrderDetails", "ActiveVersionId": "301VF002"},
+]
+
+CLASSIC_APEX_ROWS = [
+    {
+        "Id": "01pVF001",
+        "Name": "CaseEscalationHandler",
+        "Body": "public class CaseEscalationHandler {}",
+        "SymbolTable": None,
+        "ApiVersion": 60.0,
+        "IsValid": True,
+    },
+    {
+        "Id": "01pVF002",
+        "Name": "DiscountApplicator",
+        "Body": "public class DiscountApplicator {}",
+        "SymbolTable": None,
+        "ApiVersion": 60.0,
+        "IsValid": True,
+    },
+]
+
+CLASSIC_FLOW_METADATA = {
+    "301VF001": {"Id": "301VF001", "FullName": "LookupOrder-1", "Metadata": {}},
+    "301VF002": {"Id": "301VF002", "FullName": "GetOrderDetails-1", "Metadata": {}},
+}
+
+
+# ===========================================================================
+# NGA shape (MyAgent2 v3-style) — InvocationTarget is Ids, PluginId-only
+# ===========================================================================
+
+NGA_PLANNER = {
+    "Id": "1VxVF0000000N01",
+    "DeveloperName": "MyAgent2",
+    "MasterLabel": "NGA Service Agent",
+    "Description": "Atlas MyAgent",
+    "PlannerType": "Atlas__ConcurrentMultiAgentOrchestration",
+    "Capabilities": None,
+    "AgentGraph": None,
+}
+
+NGA_PLUGINS = [
+    {
+        "Id": "1VyVF0000000N01",
+        "DeveloperName": "NGATopic1",
+        "MasterLabel": "NGA Topic 1",
+        "Description": None,
+        "PluginType": "Custom",
+        "Scope": "NGA scope",
+        "IsLocal": True,
+        "CanEscalate": False,
+        "Source": "Declarative",
+        "ParentId": "1VxVF0000000N01",
+        "LocalDeveloperName": "NGATopic1",
+    },
+]
+
+# NGA functions store InvocationTarget as Salesforce Ids — `01p...` for
+# Apex, `300...` for FlowDefinition. PluginId is always set; PlannerId
+# is null.
+NGA_FUNCTIONS = [
+    {
+        "Id": "1VuVF0000000N01",
+        "DeveloperName": "NGAFlowAction",
+        "MasterLabel": "NGA Flow Action",
+        "Description": None,
+        "InvocationTargetType": "flow",
+        "InvocationTarget": "300VF999NGAFLOW",
+        "IsLocal": True,
+        "IsConfirmationRequired": False,
+        "IsIncludeInProgressIndicator": False,
+        "ProgressIndicatorMessage": None,
+        "Source": "Declarative",
+        "PluginId": "1VyVF0000000N01",
+        "PlannerId": None,
+        "ParentId": None,
+        "LocalDeveloperName": "NGAFlowAction",
+    },
+    {
+        "Id": "1VuVF0000000N02",
+        "DeveloperName": "NGAApexAction",
+        "MasterLabel": "NGA Apex Action",
+        "Description": None,
+        "InvocationTargetType": "apex",
+        "InvocationTarget": "01pVF999NGAAPEX",
+        "IsLocal": True,
+        "IsConfirmationRequired": False,
+        "IsIncludeInProgressIndicator": False,
+        "ProgressIndicatorMessage": None,
+        "Source": "Declarative",
+        "PluginId": "1VyVF0000000N01",
+        "PlannerId": None,
+        "ParentId": None,
+        "LocalDeveloperName": "NGAApexAction",
+    },
+]
+
+NGA_PLUGIN_FUNCTIONS = [
+    {"Id": "1VqVF00N01", "PluginId": "1VyVF0000000N01", "Function": "1VuVF0000000N01"},
+    {"Id": "1VqVF00N02", "PluginId": "1VyVF0000000N01", "Function": "1VuVF0000000N02"},
+]
+
+# NGA reverse-lookup rows (B1 NGA variant + B3 NGA variant)
+NGA_FLOW_DEF_BY_ID = [
+    {
+        "Id": "300VF999NGAFLOW",
+        "DeveloperName": "NGAResolvedFlow",
+        "ActiveVersionId": "301VF999NGAVER",
+        "LatestVersionId": "301VF999NGAVER",
+    },
+]
+
+NGA_APEX_BY_ID = [
+    {
+        "Id": "01pVF999NGAAPEX",
+        "Name": "NGAResolvedApex",
+        "Body": "public class NGAResolvedApex {}",
+        "SymbolTable": None,
+        "ApiVersion": 60.0,
+        "IsValid": True,
+    },
+]
+
+
+# ===========================================================================
+# SequentialPlannerIntentClassifier shape
+# ===========================================================================
+# Classic family (AiCopilot__SequentialPlannerIntentClassifier), ZERO
+# plugins, exactly ONE bundle-scope function. Tests the
+# "empty plugin_ids short-circuits" path.
+
+SEQ_PLANNER = {
+    "Id": "1VxVF0000000SEQ",
+    "DeveloperName": "SequentialAgent",
+    "MasterLabel": "Sequential Agent",
+    "Description": None,
+    "PlannerType": "AiCopilot__SequentialPlannerIntentClassifier",
+    "Capabilities": None,
+    "AgentGraph": None,
+}
+
+SEQ_FUNCTIONS = [
+    {
+        "Id": "1VuVF0000000S01",
+        "DeveloperName": "StandardReply",
+        "MasterLabel": "Standard Reply",
+        "Description": None,
+        "InvocationTargetType": "standardInvocableAction",
+        "InvocationTarget": "reply",
+        "IsLocal": False,
+        "IsConfirmationRequired": False,
+        "IsIncludeInProgressIndicator": False,
+        "ProgressIndicatorMessage": None,
+        "Source": "Declarative",
+        "PluginId": None,
+        "PlannerId": "1VxVF0000000SEQ",
+        "ParentId": None,
+        "LocalDeveloperName": "StandardReply",
+    },
+]
+
+
+# ===========================================================================
+# Bot version / definition rows (Data API shape)
+# ===========================================================================
+
+def make_bot_versions(agent_api_name: str, versions=("v5",), active: str = "v5") -> list[dict]:
+    """Build BotVersion records list in the Data API shape."""
+    return [
+        {
+            "Id": f"0Xx{i:012d}",
+            "DeveloperName": v,
+            "Status": "Active" if v == active else "Inactive",
+            "BotDefinitionId": "0Xa000000000ABC",
+            "BotDefinition": {
+                "DeveloperName": agent_api_name,
+                "MasterLabel": f"{agent_api_name} bot",
+            },
+        }
+        for i, v in enumerate(versions)
+    ]
+
+
+BOT_DEFINITION_DETAIL_CLASSIC = {
+    "DeveloperName": "MyAgent",
+    "MasterLabel": "Service Agent",
+    "Description": "Classic bot",
+    "AgentType": "ExternalCopilot",
+    "Type": "Copilot",
+    "AgentTemplate": "EinsteinCopilotForServiceTmpl",
+    "BotSource": "EinsteinCopilot",
+}
+
+BOT_DEFINITION_DETAIL_NGA = {
+    "DeveloperName": "MyAgent2",
+    "MasterLabel": "NGA Service Agent",
+    "Description": "NGA bot",
+    "AgentType": "EinsteinAgentKind",
+    "Type": "Agentforce",
+    "AgentTemplate": "SvcCopilotTmpl__EinsteinAgentKind",
+    "BotSource": "AgentforceService",
+}
+
+
+# ===========================================================================
+# Probe-channels "all OK" + "PROBE_FAILED" payloads
+# ===========================================================================
+
+
+def probe_ok_payload() -> dict:
+    return {
+        "_schema": "channels/1",
+        "_built_at_utc": 0.0,
+        "status": "OK",
+        "channels": {},
+    }
+
+
+def probe_failed_payload(sobject: str = "GenAiPlannerDefinition",
+                        missing: list[str] | None = None) -> dict:
+    return {
+        "_schema": "channels/1",
+        "_built_at_utc": 0.0,
+        "status": "PROBE_FAILED",
+        "channels": {
+            sobject: {
+                "queryable_fields": [],
+                "mandatory_missing": missing or ["DeveloperName"],
+                "describe_error": None,
+            }
+        },
+    }