feat(genui): support functions in prompts and verify rendering

.github/workflows/pub_publish.yaml

@@ -23,7 +23,7 @@ jobs:
       sdk: beta # version of dart sdk to use for publishing
       use-flutter: true
       write-comments: false
-      checkout_submodules: false
+      checkout_submodules: true
     permissions:
       id-token: write
       pull-requests: write

dev_tools/catalog_gallery/integration_test/app_test.dart

@@ -58,9 +58,7 @@ void main() {
   testWidgets('catalog_gallery smoke test - verify initial state', (
     tester,
   ) async {
-    await tester.pumpWidget(
-      CatalogGalleryApp(sampleSource: sampleSource),
-    );
+    await tester.pumpWidget(CatalogGalleryApp(sampleSource: sampleSource));
     await tester.pumpAndSettle();
 
     expect(find.text('Catalog Gallery'), findsOneWidget);

dev_tools/catalog_gallery/lib/sample_source.dart

@@ -86,9 +86,7 @@ class DirectorySampleSource implements SampleSource {
     final List<SampleRef> refs = files
         .map(
           (file) => SampleRef(
-            name: directory.fileSystem.path.basenameWithoutExtension(
-              file.path,
-            ),
+            name: directory.fileSystem.path.basenameWithoutExtension(file.path),
             load: file.readAsString,
           ),
         )

dev_tools/catalog_gallery/pubspec.yaml

@@ -18,7 +18,7 @@ dependencies:
   file: ^7.0.1
   flutter:
     sdk: flutter
-  genui: ^0.9.0
+  genui: ^0.10.0
   json_schema_builder: ^0.1.3
   yaml: ^3.1.3
 

dev_tools/catalog_gallery/test/sample_source_test.dart

@@ -0,0 +1,66 @@
+// Copyright 2025 The Flutter Authors.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+import 'dart:convert';
+import 'dart:typed_data';
+
+import 'package:catalog_gallery/sample_source.dart';
+import 'package:flutter/services.dart';
+import 'package:flutter_test/flutter_test.dart';
+
+void main() {
+  TestWidgetsFlutterBinding.ensureInitialized();
+
+  group('AssetSampleSource', () {
+    test('listSamples loads and parses samples from asset bundle', () async {
+      final binaryMessenger =
+          TestDefaultBinaryMessengerBinding.instance.defaultBinaryMessenger;
+
+      // Mock AssetManifest.bin
+      final manifestData = {
+        'samples/hello.sample': [
+          {'asset': 'samples/hello.sample'},
+        ],
+        'samples/world.sample': [
+          {'asset': 'samples/world.sample'},
+        ],
+        'other/file.txt': [
+          {'asset': 'other/file.txt'},
+        ],
+      };
+
+      final ByteData manifestByteData = const StandardMessageCodec()
+          .encodeMessage(manifestData)!;
+
+      binaryMessenger.setMockMessageHandler('flutter/assets', (
+        ByteData? message,
+      ) async {
+        final String key = utf8.decode(message!.buffer.asUint8List());
+        if (key == 'AssetManifest.bin' || key == 'AssetManifest.json') {
+          return manifestByteData;
+        }
+        if (key == 'samples/hello.sample') {
+          return ByteData.view(utf8.encode('hello content').buffer);
+        }
+        if (key == 'samples/world.sample') {
+          return ByteData.view(utf8.encode('world content').buffer);
+        }
+        return null;
+      });
+
+      addTearDown(() {
+        binaryMessenger.setMockMessageHandler('flutter/assets', null);
+      });
+
+      const source = AssetSampleSource();
+      final samples = await source.listSamples();
+
+      expect(samples, hasLength(2));
+      expect(samples[0].name, 'hello');
+      expect(await samples[0].load(), 'hello content');
+      expect(samples[1].name, 'world');
+      expect(await samples[1].load(), 'world content');
+    });
+  });
+}

dev_tools/composer/lib/create_tab.dart

@@ -78,7 +78,7 @@ class _CreateTabState extends State<CreateTab> {
         transport: transport,
       );
 
-      final promptBuilder = PromptBuilder.chat(
+      final promptBuilder = await PromptBuilder.createChat(
         catalog: catalog,
         systemPromptFragments: [
           'You are a UI generator. The user will describe a UI they want. '

dev_tools/composer/pubspec.yaml

@@ -18,7 +18,7 @@ dependencies:
     sdk: flutter
   flutter_code_editor: ^0.3.5
   flutter_highlight: ^0.7.0
-  genui: ^0.9.0
+  genui: ^0.10.0
   highlight: ^0.7.0
   logging: ^1.3.0
   window_manager: ^0.5.1

examples/simple_chat/lib/chat_session.dart

@@ -64,19 +64,20 @@ final Catalog _customCatalog = _basicCatalog.copyWith(
   newItems: [climbingLocationItem],
 );
 
-PromptBuilder _promptBuilderFor(Catalog catalog) => PromptBuilder.chat(
-  catalog: catalog,
-  systemPromptFragments: [
-    Prompts.summary,
-    PromptFragments.acknowledgeUser(),
-    PromptFragments.requireAtLeastOneSubmitElement(
-      prefix: PromptBuilder.defaultImportancePrefix,
-    ),
-    PromptFragments.uiGenerationRestriction(
-      prefix: PromptBuilder.defaultImportancePrefix,
-    ),
-  ],
-);
+Future<PromptBuilder> _promptBuilderFor(Catalog catalog) async =>
+    await PromptBuilder.createChat(
+      catalog: catalog,
+      systemPromptFragments: [
+        Prompts.summary,
+        PromptFragments.acknowledgeUser(),
+        PromptFragments.requireAtLeastOneSubmitElement(
+          prefix: PromptBuilder.defaultImportancePrefix,
+        ),
+        PromptFragments.uiGenerationRestriction(
+          prefix: PromptBuilder.defaultImportancePrefix,
+        ),
+      ],
+    );
 
 sealed class ChatSession extends ChangeNotifier {
   ChatSession._();
@@ -196,7 +197,7 @@ class A2uiChatSession extends ChatSession {
   late final StreamSubscription<ChatMessage> _submitSub;
   late final StreamSubscription<SurfaceUpdate> _surfaceSub;
 
-  void _init() {
+  Future<void> _init() async {
     _messageSub = _transport.incomingMessages.listen(
       _surfaceController.handleMessage,
     );
@@ -206,9 +207,8 @@ class A2uiChatSession extends ChatSession {
     );
     _surfaceSub = _surfaceController.surfaceUpdates.listen(_onSurfaceUpdate);
 
-    _transport.addSystemMessage(
-      _promptBuilderFor(_catalog).systemPromptJoined(),
-    );
+    final PromptBuilder pb = await _promptBuilderFor(_catalog);
+    _transport.addSystemMessage(pb.systemPromptJoined());
   }
 
   void _onSurfaceUpdate(SurfaceUpdate update) {

examples/simple_chat/pubspec.yaml

@@ -17,7 +17,7 @@ dependencies:
   flutter:
     sdk: flutter
   flutter_markdown_plus: ^1.0.7
-  genui: ^0.9.0
+  genui: ^0.10.0
   json_schema_builder: ^0.1.3
   logging: ^1.3.0
 

examples/verdure/client/pubspec.yaml

@@ -20,7 +20,7 @@ dependencies:
     sdk: flutter
   flutter_riverpod: ^3.1.0
   flutter_svg: ^2.2.2
-  genui: ^0.9.0
+  genui: ^0.10.0
   genui_a2a: ^0.9.0
   go_router: ^17.0.0
   image_picker: ^1.2.0

packages/genui/CHANGELOG.md

@@ -1,8 +1,8 @@
 # `genui` Changelog
 
-## 0.10.0 (in progress)
+## 0.10.0
 
-- **Refactor**: genui now runs on `package:a2ui_core`. See
+- **BREAKING**: genui now runs on `package:a2ui_core`. See
   [the migration guide](../../docs/usage/migration/migration_0.9.1_to_0.10.0.md).
 - **BREAKING**: A2UI message types are now `package:a2ui_core` types. The GenUI
   message classes (`A2uiMessage`, `CreateSurface`, `UpdateComponents`,
@@ -14,13 +14,21 @@
   surface's data model via `SurfaceController.contextFor(id).dataModel`.
 - **BREAKING**: `SurfaceRegistry.updateSurface(...)` is removed; drive surfaces
   through `SurfaceController.handleMessage`.
+- **BREAKING**: Changed `PromptBuilder.chat` and `PromptBuilder.custom` from synchronous factory constructors to asynchronous static methods (`createChat` and `createCustom`) to support asynchronous asset loading.
+- **BREAKING**: Changed `_loadSchemas` return type to a named record structure.
+- **BREAKING**: Restricted public API surface of low-level `primitives` exports. Only `CancellationException`, `CancellationSignal`, `JsonMap`, `basicCatalogId`, `configureLogging`, `genUiLogger`, and `generateId` are now exported from `package:genui/genui.dart`.
 - **Behavior**: `DataModel` writes are stricter; some writes that previously did
   nothing now throw, and sparse list writes fill skipped entries with `null`.
 - **Behavior**: A duplicate `createSurface` for an already-active surface id is now
   an error.
 - The catalog-widget authoring API is unchanged; `SurfaceDefinition` and
   `Component` remain genui types.
+- **Refactor**: Extracted exception mapping logic to a private helper `_errorToMap` in `SurfaceController`.
+- **Refactor**: Centralized and shared common schema registry initialization helper.
+- **Refactor**: Extracted mock binary messenger asset setup to a shared helper for test reuse.
+- **Fix**: Sanitized raw error messages exposed from `ArgumentError` in `Button` press handlers.
 
+## 0.9.2
 ## 0.9.1
 
 - **Feature**: Updated example/README.md.

packages/genui/assets/schemas/common_types.json

@@ -0,0 +1 @@
+../../../../submodules/a2ui/specification/v0_9/json/common_types.json

packages/genui/assets/schemas/server_to_client.json

@@ -0,0 +1 @@
+../../../../submodules/a2ui/specification/v0_9/json/server_to_client.json

packages/genui/lib/src/catalog/basic_catalog_widgets/button.dart

@@ -2,9 +2,12 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+import 'dart:async';
+
 import 'package:flutter/material.dart';
 import 'package:json_schema_builder/json_schema_builder.dart';
 
+import '../../model/a2ui_exceptions.dart';
 import '../../model/a2ui_schemas.dart';
 import '../../model/catalog_item.dart';
 import '../../model/data_model.dart';
@@ -231,9 +234,40 @@ Future<void> _handlePress(
       funcMap,
     );
     try {
-      await resultStream.first;
+      await resultStream.first.timeout(
+        const Duration(seconds: 10),
+        onTimeout: () => throw TimeoutException(
+          'Function execution for $callName timed out',
+        ),
+      );
     } catch (exception, stackTrace) {
-      itemContext.reportError(exception, stackTrace);
+      genUiLogger.severe(
+        'Error executing function call "$callName" on button press',
+        exception,
+        stackTrace,
+      );
+
+      if (exception is A2uiFunctionException) {
+        itemContext.reportError(exception, stackTrace);
+      } else if (exception is TimeoutException) {
+        itemContext.reportError(
+          A2uiFunctionException(
+            'Function execution timed out.',
+            functionName: callName,
+            cause: exception,
+          ),
+          stackTrace,
+        );
+      } else {
+        itemContext.reportError(
+          A2uiFunctionException(
+            'Function execution failed. Please check arguments and try again.',
+            functionName: callName,
+            cause: exception,
+          ),
+          stackTrace,
+        );
+      }
     }
   } else {
     genUiLogger.warning(

packages/genui/lib/src/catalog/basic_catalog_widgets/date_time_input.dart

@@ -340,9 +340,7 @@ final dateTimeInput = CatalogItem(
         {
           "id": "root",
           "component": "DateTimeInput",
-          "value": {
-            "path": "/myDateTime"
-          }
+          "value": "2026-05-15"
         }
       ]
     ''',
@@ -354,7 +352,7 @@ final dateTimeInput = CatalogItem(
           "value": {
             "path": "/myDate"
           },
-          "enableTime": false
+          "variant": "date"
         }
       ]
     ''',
@@ -366,7 +364,7 @@ final dateTimeInput = CatalogItem(
           "value": {
             "path": "/myTime"
           },
-          "enableDate": false
+          "variant": "time"
         }
       ]
     ''',

packages/genui/lib/src/catalog/basic_catalog_widgets/image.dart

@@ -78,7 +78,9 @@ final CatalogItem image = CatalogItem(
       {
         "id": "root",
         "component": "Image",
-        "url": "https://storage.googleapis.com/cms-storage-bucket/lockup_flutter_horizontal.c823e53b3a1a7b0d36a9.png",
+        "url": {
+          "path": "/imageUrl"
+        },
         "variant": "mediumFeature"
       }
     ]

packages/genui/lib/src/catalog/basic_catalog_widgets/text.dart

@@ -52,8 +52,7 @@ final text = CatalogItem(
         {
           "id": "root",
           "component": "Text",
-          "text": "Hello World",
-          "variant": "h1"
+          "text": "Hello World"
         }
       ]
     ''',