dmitry.galkin/chatbot-template
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add artifact tests (#859)

Browse source
This commit is contained in:
Jeremy 2025-03-11 14:39:36 -07:00 • committed by GitHub
parent 9628c54755
commit 8e561dced4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 669 additions and 255 deletions
Download patch file Download diff file Expand all files Collapse all files

141
tests/prompts/basic.ts Normal file
View file

@ -0,0 +1,141 @@
import { CoreMessage } from 'ai';
export const TEST_PROMPTS: Record<string, CoreMessage> = {
USER_SKY: {
role: 'user',
content: [{ type: 'text', text: 'Why is the sky blue?' }],
},
USER_GRASS: {
role: 'user',
content: [{ type: 'text', text: 'Why is grass green?' }],
},
USER_THANKS: {
role: 'user',
content: [{ type: 'text', text: 'Thanks!' }],
},
USER_NEXTJS: {
role: 'user',
content: [
{ type: 'text', text: 'What are the advantages of using Next.js?' },
],
},
USER_IMAGE_ATTACHMENT: {
role: 'user',
content: [
{
type: 'text',
text: 'Who painted this?',
},
{
type: 'image',
image: '...',
},
],
},
USER_TEXT_ARTIFACT: {
role: 'user',
content: [
{
type: 'text',
text: 'Help me write an essay about Silicon Valley',
},
],
},
CREATE_DOCUMENT_TEXT_CALL: {
role: 'user',
content: [
{
type: 'text',
text: 'Essay about Silicon Valley',
},
],
},
CREATE_DOCUMENT_TEXT_RESULT: {
role: 'tool',
content: [
{
type: 'tool-result',
toolCallId: 'call_123',
toolName: 'createDocument',
result: {
id: '3ca386a4-40c6-4630-8ed1-84cbd46cc7eb',
title: 'Essay about Silicon Valley',
kind: 'text',
content: 'A document was created and is now visible to the user.',
},
},
],
},
GET_WEATHER_CALL: {
role: 'user',
content: [
{
type: 'text',
text: "What's the weather in sf?",
},
],
},
GET_WEATHER_RESULT: {
role: 'tool',
content: [
{
type: 'tool-result',
toolCallId: 'call_456',
toolName: 'getWeather',
result: {
latitude: 37.763283,
longitude: -122.41286,
generationtime_ms: 0.06449222564697266,
utc_offset_seconds: -25200,
timezone: 'America/Los_Angeles',
timezone_abbreviation: 'GMT-7',
elevation: 18,
current_units: {
time: 'iso8601',
interval: 'seconds',
temperature_2m: '°C',
},
current: {
time: '2025-03-10T14:00',
interval: 900,
temperature_2m: 17,
},
daily_units: {
time: 'iso8601',
sunrise: 'iso8601',
sunset: 'iso8601',
},
daily: {
time: [
'2025-03-10',
'2025-03-11',
'2025-03-12',
'2025-03-13',
'2025-03-14',
'2025-03-15',
'2025-03-16',
],
sunrise: [
'2025-03-10T07:27',
'2025-03-11T07:25',
'2025-03-12T07:24',
'2025-03-13T07:22',
'2025-03-14T07:21',
'2025-03-15T07:19',
'2025-03-16T07:18',
],
sunset: [
'2025-03-10T19:12',
'2025-03-11T19:13',
'2025-03-12T19:14',
'2025-03-13T19:15',
'2025-03-14T19:16',
'2025-03-15T19:17',
'2025-03-16T19:17',
],
},
},
},
],
},
};