kris/afv-library
1
0
Fork 0
mirror of https://github.com/forcedotcom/afv-library.git synced 2026-07-31 20:12:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
fa3cbb4fa9
afv-library/skills/agentforce-architecture-analyze/scripts/tests/fixtures/genai_payloads.py

401 lines
12 KiB
Python
Raw Blame History

"""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,
}
},
}
Reference in New Issue View Git Blame Copy Permalink