feat(ai): support streaming (aws-amplify#899)

sundersc · Nov 6, 2024 · 7b8ee57 · 7b8ee57
1 parent cbe1775
commit 7b8ee57
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/...gin/src/__tests__/visitors/__snapshots__/appsync-model-introspection-visitor.test.ts.snap b/...gin/src/__tests__/visitors/__snapshots__/appsync-model-introspection-visitor.test.ts.snap
@@ -326,7 +326,7 @@ exports[`Conversation Route Introspection Visitor Metadata snapshot should gener
                     \\"isRequired\\": false,
                     \\"name\\": \\"onCreateAssistantResponsePirateChat\\",
                     \\"type\\": {
-                        \\"model\\": \\"ConversationMessagePirateChat\\"
+                        \\"nonModel\\": \\"ConversationMessageStreamPart\\"
                     },
                     \\"arguments\\": {
                         \\"conversationId\\": {

diff --git a/packages/appsync-modelgen-plugin/src/utils/process-conversation.ts b/packages/appsync-modelgen-plugin/src/utils/process-conversation.ts
@@ -185,7 +185,7 @@ function generateSubscriptionMetadata(routeName: string, modelName: string): Sch
     isArray: false,
     isRequired: false,
     name: `onCreateAssistantResponse${routeName}`,
-    type: { model: modelName },
+    type: { nonModel: 'ConversationMessageStreamPart' },
     arguments: {
       'conversationId': {
         name: 'conversationId',