Skip to content

Commit ff2d83b

Browse files
committed
feat(appsync-modelgen-plugin): add metrics and usage fields to conversation message model
1 parent 63c7a52 commit ff2d83b

2 files changed

Lines changed: 20 additions & 0 deletions

File tree

packages/appsync-modelgen-plugin/src/__tests__/visitors/__snapshots__/appsync-model-introspection-visitor.test.ts.snap

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -214,6 +214,24 @@ exports[`Conversation Route Introspection Visitor Metadata snapshot should gener
214214
\\"isRequired\\": false,
215215
\\"isArrayNullable\\": true
216216
},
217+
\\"metrics\\": {
218+
\\"name\\": \\"metrics\\",
219+
\\"type\\": {
220+
\\"nonModel\\": \\"AmplifyAIMetrics\\"
221+
},
222+
\\"attributes\\": [],
223+
\\"isArray\\": false,
224+
\\"isRequired\\": false
225+
},
226+
\\"usage\\": {
227+
\\"name\\": \\"usage\\",
228+
\\"type\\": {
229+
\\"nonModel\\": \\"AmplifyAIUsage\\"
230+
},
231+
\\"attributes\\": [],
232+
\\"isArray\\": false,
233+
\\"isRequired\\": false
234+
},
217235
\\"createdAt\\": {
218236
\\"name\\": \\"createdAt\\",
219237
\\"type\\": \\"AWSDateTime\\",

packages/appsync-modelgen-plugin/src/utils/process-conversation.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,8 @@ function generateConversationMessageModel(conversationModelName: string, modelNa
9595
content: generateField('content', { nonModel: 'AmplifyAIContentBlock' }, { isArray: true }),
9696
aiContext: generateField('aiContext', 'AWSJSON'),
9797
toolConfiguration: generateField('toolConfiguration', { nonModel: 'AmplifyAIToolConfiguration' }, { isArray: true, isArrayNullable: true }),
98+
metrics: generateField('metrics', { nonModel: 'AmplifyAIMetrics' }),
99+
usage: generateField('usage', { nonModel: 'AmplifyAIUsage' }),
98100
createdAt: generateTimestampField('createdAt'),
99101
updatedAt: generateTimestampField('updatedAt'),
100102
},

0 commit comments

Comments
 (0)