additional bug fixes identified

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: 2 people

apps/ui/package.json

@@ -2,6 +2,7 @@
   "name": "@agentscript/ui",
   "private": true,
   "version": "2.0.0",
+  "license": "Apache-2.0",
   "type": "module",
   "scripts": {
     "dev": "vite",

dialect/agentfabric/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@agentscript/agentfabric-dialect",
-  "version": "0.1.10",
+  "version": "0.1.13",
   "description": "AgentFabric dialect — schema, lint rules, compiler, and dialect config",
   "type": "module",
   "main": "dist/index.js",

dialect/agentfabric/src/lint/passes/index.ts

@@ -17,9 +17,34 @@ import {
   unreachableCodePass,
   emptyBlockPass,
   expressionValidationPass,
+  spreadContextPass,
 } from '@agentscript/language';
 import { agentFabricSemanticPass } from './agentfabric-semantic.js';
 import { suppressActionsNamespaceUndefinedReferencePass } from './suppress-tools-namespace-undefined-reference.js';
+import type { ExpressionValidationOptions } from '@agentscript/language/lint';
+import { AgentFabricSchemaInfo } from '../../schema.js';
+
+const expressionOptions: ExpressionValidationOptions = {
+  functions: new Set([
+    'len',
+    'max',
+    'min',
+    'uuid',
+    'now',
+    'strip',
+    'startswith',
+    'endswith',
+    'abs',
+    'round',
+    'sum',
+    'parse_json',
+    'capitalize',
+    'join',
+    'split',
+    'splitlines',
+  ]),
+  namespacedFunctions: AgentFabricSchemaInfo.namespacedFunctions,
+};
 
 /** All AgentFabric lint passes in engine execution order. */
 export function defaultRules(): LintPass[] {
@@ -33,7 +58,8 @@ export function defaultRules(): LintPass[] {
     positionIndexPass(),
     unreachableCodePass(),
     emptyBlockPass(),
-    expressionValidationPass(),
+    expressionValidationPass(expressionOptions),
+    spreadContextPass(),
     agentFabricSemanticPass(),
     // Validation
     undefinedReferencePass(),

dialect/agentfabric/src/schema.ts

@@ -501,6 +501,17 @@ export const AgentFabricSchemaInfo: SchemaInfo = {
   globalScopes: {
     request: new Set(['payload', 'interface', 'headers']),
   },
+  namespacedFunctions: {
+    a2a: new Set([
+      'task',
+      'message',
+      'textPart',
+      'parts',
+      'dataPart',
+      'filePart',
+      'artifact',
+    ]),
+  },
 };
 
 export const agentFabricSchemaContext: SchemaContext = createSchemaContext(

dialect/agentfabric/src/tests/lint.test.ts

@@ -437,6 +437,42 @@ echo done:
     ).toBe(false);
   });
 
+  it('does not accept A2A global calls with @', () => {
+    const source = `
+echo successResponse:
+  kind: "a2a:response"
+  task: @a2a.task({ state: "completed", message: @a2a.message()})
+`;
+    const result = parseAndLintSource(source);
+    expect(
+      result.diagnostics.filter(
+        d =>
+          d.code === 'namespace-function-call' &&
+          d.message.includes('Only direct namespace function calls are allowed')
+      ).length
+    ).toBe(2);
+  });
+
+  it('allows namespaced A2A helper calls in expression fields (a2a.message, a2a.textPart, …)', () => {
+    const source = `
+echo out:
+  kind: "a2a:response"
+  message: a2a.message(a2a.textPart("hello"))
+`;
+    const result = parseAndLintSource(source);
+    expect(result.diagnostics.length).toBe(0);
+  });
+
+  it('allows namespaced A2A helper calls when assigning value to variable', () => {
+    const source = `
+executor step:
+  do: ->
+    set @variables.t = a2a.task({ state: "completed" })
+`;
+    const result = parseAndLintSource(source);
+    expect(result.diagnostics.length).toBe(0);
+  });
+
   it('accepts generator prompt in procedure form', () => {
     const source = `
 config:

dialect/agentfabric/src/tests/resources/agentfabric-customer-support-network.yaml

@@ -72,11 +72,6 @@ unifiedAgentSpec:
         data-type: string
         description: ""
         default: ""
-      - name: output_response
-        label: output_response
-        data-type: string
-        description: ""
-        default: ""
     initial-node: set_context
     nodes:
       - name: general_response
@@ -86,6 +81,7 @@ unifiedAgentSpec:
         llm:
           ref: openai_gpt4
           configuration:
+            max_output_tokens: "1024"
             model: gpt-4o
         on-init: null
         system-prompt: >-
@@ -168,14 +164,15 @@ unifiedAgentSpec:
             enabled: true
         after-reasoning:
           - type: handoff
-            target: set_output_response_from_general_agent
+            target: send_response
       - name: technical_handler
         label: null
         description: Handles technical support issues using knowledge-base tools.
         type: agent
         llm:
           ref: openai_gpt4
           configuration:
+            max_output_tokens: "1024"
             model: gpt-4o
         on-init: null
         system-prompt: >-
@@ -200,15 +197,15 @@ unifiedAgentSpec:
             enabled: true
         after-reasoning:
           - type: handoff
-            target: set_output_response_from_technical_agent
+            target: send_response
       - name: analyze_request
         label: null
         description: null
         type: agent
         llm:
-          ref: openai_gpt4
+          ref: gemini_2_5_flash
           configuration:
-            model: gpt-4o
+            model: gemini-2.5-flash
           output-structure-ref: os_analyze_request
         on-init: null
         system-prompt: >-
@@ -231,7 +228,7 @@ unifiedAgentSpec:
           - type: action
             ref: IdentityAction
             state-updates:
-              - requestTimestamp: '"2026-03-06T10:23:24.371729+00:00"'
+              - requestTimestamp: 'now()'
               - customerMessage: state.request.payload.message
         on-init:
           - ref: IdentityAction
@@ -255,48 +252,6 @@ unifiedAgentSpec:
             state-updates:
               - outputs: add(state.outputs, "billing_handler", result["result"])
         on-init: null
-        on-exit:
-          - type: handoff
-            target: set_output_response_from_billing_agent
-        add-tool-result-to-chat-history: false
-        output-template: null
-      - name: set_output_response_from_billing_agent
-        type: action
-        label: null
-        tools:
-          - type: action
-            ref: IdentityAction
-            state-updates:
-              - output_response: state.outputs['billing_handler']
-        on-init: null
-        on-exit:
-          - type: handoff
-            target: send_response
-        add-tool-result-to-chat-history: false
-        output-template: null
-      - name: set_output_response_from_technical_agent
-        type: action
-        label: null
-        tools:
-          - type: action
-            ref: IdentityAction
-            state-updates:
-              - output_response: system.node_outputs['technical_handler']
-        on-init: null
-        on-exit:
-          - type: handoff
-            target: send_response
-        add-tool-result-to-chat-history: false
-        output-template: null
-      - name: set_output_response_from_general_agent
-        type: action
-        label: null
-        tools:
-          - type: action
-            ref: IdentityAction
-            state-updates:
-              - output_response: system.node_outputs['general_response']
-        on-init: null
         on-exit:
           - type: handoff
             target: send_response
@@ -311,7 +266,6 @@ unifiedAgentSpec:
             state-updates:
               - customerMessage: '""'
               - requestTimestamp: '""'
-              - output_response: '""'
         on-init: null
         on-exit: null
         add-tool-result-to-chat-history: false
@@ -324,12 +278,10 @@ unifiedAgentSpec:
         on-exit:
           - type: handoff
             target: billing_handler
-            enabled: template::{{system.node_outputs['analyze_request'].category ==
-              "billing"}}
+            enabled: "system.node_outputs['analyze_request'].category == \"billing\""
           - type: handoff
             target: technical_handler
-            enabled: template::{{system.node_outputs['analyze_request'].category ==
-              "technical"}}
+            enabled: "system.node_outputs['analyze_request'].category == \"technical\""
           - type: handoff
             target: general_response
         add-tool-result-to-chat-history: false
@@ -340,7 +292,7 @@ unifiedAgentSpec:
         tools:
           - ref: IdentityAction
             state-updates:
-              - __send_response_value: template::{"state":"completed","message":{"kind":"text","role":"agent","parts":[{"kind":"text","text":"{{state.output_response}}"}]}}
+              - __send_response_value: "a2a_task(state=\"completed\", message=a2a_message(parts=[a2a_textPart(state._node_input)]))"
               - outputs: add(state.outputs, "send_response", state.__send_response_value)
         on-init: null
         on-exit:
@@ -354,6 +306,11 @@ llmProviders:
     metadata:
       platform: openai
       connection: openai_connection
+  - name: gemini_2_5_flash
+    description: "LLM provider: gemini_2_5_flash"
+    metadata:
+      platform: gemini
+      connection: gemini_connection
 invokableClients:
   - name: search_articles-client
     type: mcp_tool

dialect/agentfabric/src/tests/resources/agentfabric-customer-support-netwrok.agent

@@ -7,13 +7,18 @@ config:
   agent_name: "customer-support-flow"
   label: "Customer Support Workflow"
   description: "Handles customer support requests with intelligent routing."
-  default_llm: @llm.openai_gpt4
+  default_llm: @llm.gemini_2_5_flash
 
 llm:
   openai_gpt4:
     target: "llm://openai_connection"
     kind: "openai"
     model: "gpt-4o"
+    max_output_tokens: 1024
+  gemini_2_5_flash:
+    target: "llm://gemini_connection"
+    kind: "gemini"
+    model: "gemini-2.5-flash"
 
 action_definitions:
   search_articles:
@@ -33,22 +38,22 @@ action_definitions:
 variables:
   requestTimestamp: mutable string = ""
   customerMessage: mutable string = ""
-  output_response: mutable string = ""
 
 trigger customerSupportTrigger:
+  kind: "a2a"
   target: "brokers://customer-support-flow/a2a"
   on_message: ->
     transition to @executor.set_context
 
 executor set_context:
   do: ->
-    set @variables.requestTimestamp = "2026-03-06T10:23:24.371729+00:00"
+    set @variables.requestTimestamp = now()
     set @variables.customerMessage = @request.payload.message
   on_exit: ->
     transition to @generator.analyze_request
 
 generator analyze_request:
-  llm: @llm.openai_gpt4
+  llm: @llm.gemini_2_5_flash
   prompt: ->
     | Analyze the following customer support request and categorize it as one of: 'billing', 'technical', 'general'. Also extract key entities and sentiment.
     |
@@ -96,7 +101,7 @@ executor billing_handler:
     run @actions.billing_agent
       with message = @variables.customerMessage
   on_exit: ->
-    transition to @executor.set_output_response_from_billing_agent
+    transition to @echo.send_response
 
 subagent technical_handler:
   description: "Handles technical support issues using knowledge-base tools."
@@ -112,7 +117,7 @@ subagent technical_handler:
       kb_search: @actions.search_articles
       kb_get_article: @actions.get_article
   on_exit: ->
-    transition to @executor.set_output_response_from_technical_agent
+    transition to @echo.send_response
 
 orchestrator general_response:
   description: "Handles general customer support responses and routing."
@@ -128,35 +133,24 @@ orchestrator general_response:
       kb_search: @actions.search_articles
       kb_get_article: @actions.get_article
       billing_delegate: @actions.billing_agent
-  on_exit: ->
-    transition to @executor.set_output_response_from_general_agent
-
-executor set_output_response_from_billing_agent:
-  do: ->
-    set @variables.output_response = @executor.billing_handler.output
-  on_exit: ->
-    transition to @echo.send_response
-
-executor set_output_response_from_technical_agent:
-  do: ->
-    set @variables.output_response = @subagent.technical_handler.output
-  on_exit: ->
-    transition to @echo.send_response
-
-executor set_output_response_from_general_agent:
-  do: ->
-    set @variables.output_response = @orchestrator.general_response.output
   on_exit: ->
     transition to @echo.send_response
 
 echo send_response:
   kind: "a2a:response"
-  message: "{!@variables.output_response}"
+  task: a2a.task({
+    state: "completed",
+    message: a2a.message({
+      parts: [
+        a2a.textPart(@variables.output_response)
+      ]
+    })
+  }
+  )
   on_exit: ->
     transition to @executor.cleanup
 
 executor cleanup:
   do: ->
     set @variables.customerMessage = ""
-    set @variables.requestTimestamp = ""
-    set @variables.output_response = ""
+    set @variables.requestTimestamp = ""