-

Author: polina-c

examples/eval/pubspec.yaml

@@ -16,6 +16,8 @@ dependencies:
   flutter:
     sdk: flutter
   genui: ^0.7.0
+  simple_chat:
+    path: ../simple_chat
 
 dev_dependencies:
   flutter_test:

examples/eval/test/infra_test.dart

@@ -0,0 +1,31 @@
+// 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.
+
+// ignore_for_file: avoid_print
+
+import 'package:flutter_test/flutter_test.dart';
+
+import 'test_infra/ai_client.dart';
+import 'test_infra/api_key.dart';
+
+void main() {
+  test('test can read api key "$geminiApiKeyName"', () {
+    final String key = apiKeyForEval();
+    expect(key, isNotEmpty);
+    print('API Key: ${key.substring(0, 1)}...${key.substring(key.length - 1)}');
+  });
+
+  test('test can talk with AI', () async {
+    final aiClient = DartanticAiClient();
+    addTearDown(aiClient.dispose);
+
+    final String result =
+        (await aiClient
+                .sendStream('Please, tell me a joke.', history: [])
+                .toList())
+            .join(' ');
+    expect(result, isNotEmpty);
+    print('Result: $result');
+  });
+}

examples/eval/test/smoke_test.dart

@@ -5,25 +5,12 @@
 // ignore_for_file: avoid_print
 
 import 'package:flutter_test/flutter_test.dart';
-
-import 'test_infra/ai_client.dart';
-import 'test_infra/api_key.dart';
+import 'package:genui/genui.dart';
+import 'package:simple_chat/chat_session.dart';
 
 void main() {
-  test('test can read api key "$geminiApiKeyName"', () {
-    final String key = apiKeyForEval();
-    expect(key, isNotEmpty);
-    print('API Key: ${key.substring(0, 1)}...${key.substring(key.length - 1)}');
-  });
-
-  test('test can talk with AI', () async {
-    final aiClient = DartanticAiClient();
-    addTearDown(aiClient.dispose);
-
-    final String result = await aiClient
-        .sendStream('Please, tell me a joke.', history: [])
-        .first;
-    expect(result, isNotEmpty);
-    print('Result: $result');
+  test('Model respects configuration of prompt builder.', () async {
+    // final Catalog catalog = simpleChatCatalog;
+    // final promptBuilder = simpleChatPromptBuilder;
   });
 }

examples/simple_chat/lib/chat_session.dart

@@ -12,31 +12,41 @@ import 'ai_client.dart';
 import 'ai_client_transport.dart';
 import 'message.dart';
 
-/// A class that manages the chat session state and logic.
-class ChatSession extends ChangeNotifier {
-  ChatSession({required AiClient aiClient}) {
-    // 1. Create Transport
-    _transport = AiClientTransport(aiClient: aiClient);
-
-    // 2. Initialize Catalog & Controller
-    final Catalog catalog = BasicCatalogItems.asCatalog(
-      systemPromptFragments: [
-        '''
+final Catalog _catalog = BasicCatalogItems.asCatalog(
+  systemPromptFragments: [
+    '''
 When you need additional information from the user, try to use the component '${BasicCatalogItems.choicePicker.name}' to ask for it.
 ''',
-        '''
+    '''
 If there is no way to itemize all the options, either use the component '${BasicCatalogItems.textField.name}' or add option 'Other' to the '${BasicCatalogItems.choicePicker.name}'.
 ''',
-      ],
-    );
-    _surfaceController = SurfaceController(catalogs: [catalog]);
+  ],
+);
+
+final PromptBuilder _promptBuilder = PromptBuilder.chat(
+  catalog: _catalog,
+  systemPromptFragments: [
+    'You are a helpful assistant who chats with a user.',
+    PromptFragments.acknowledgeUser(),
+    PromptFragments.requireAtLeastOneSubmitElement(
+      prefix: PromptBuilder.defaultImportancePrefix,
+    ),
+    PromptFragments.uiGenerationRestriction(
+      prefix: PromptBuilder.defaultImportancePrefix,
+    ),
+  ],
+);
 
-    // 3. Initialize Conversation
+/// A class that manages the chat session state and logic.
+class ChatSession extends ChangeNotifier {
+  ChatSession({required AiClient aiClient}) {
+    _transport = AiClientTransport(aiClient: aiClient);
+    _surfaceController = SurfaceController(catalogs: [_catalog]);
     _conversation = Conversation(
       controller: _surfaceController,
       transport: _transport,
     );
-    _init(catalog);
+    _init(_catalog);
   }
 
   late final AiClientTransport _transport;
@@ -75,20 +85,7 @@ If there is no way to itemize all the options, either use the component '${Basic
       }
     });
 
-    final promptBuilder = PromptBuilder.chat(
-      catalog: catalog,
-      systemPromptFragments: [
-        'You are a helpful assistant who chats with a user.',
-        PromptFragments.acknowledgeUser(),
-        PromptFragments.requireAtLeastOneSubmitElement(
-          prefix: PromptBuilder.defaultImportancePrefix,
-        ),
-        PromptFragments.uiGenerationRestriction(
-          prefix: PromptBuilder.defaultImportancePrefix,
-        ),
-      ],
-    );
-    _transport.addSystemMessage(promptBuilder.systemPromptJoined());
+    _transport.addSystemMessage(_promptBuilder.systemPromptJoined());
   }
 
   void _addSurfaceMessage(String surfaceId) {