fix tests

firebase · Oct 9, 2024 · 51fc580 · 51fc580
1 parent 0f543e4
commit 51fc580
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/js/genkit/tests/helpers.ts b/js/genkit/tests/helpers.ts
@@ -63,8 +63,9 @@ export function defineEchoModel(ai: Genkit) {
                 request.messages
                   .map(
                     (m) =>
-                      (m.role === 'user' ? '' : `${m.role}: `) +
-                      m.content.map((c) => c.text).join()
+                      (m.role === 'user' || m.role === 'model'
+                        ? ''
+                        : `${m.role}: `) + m.content.map((c) => c.text).join()
                   )
                   .join(),
             },