fix(ai): make stream part - contentBlockIndex optional

atierian · atierian · commit a445370d4d1b · 2024-11-11T17:54:49.000-05:00
diff --git a/.changeset/many-dots-invite.md b/.changeset/many-dots-invite.md
@@ -0,0 +1,5 @@
+---
+"@aws-amplify/data-schema": patch
+---
+
+make ConversationMessageStreamPart.contentBlockIndex optional for error events
diff --git a/packages/data-schema/__tests__/__snapshots__/ClientSchema.test.ts.snap b/packages/data-schema/__tests__/__snapshots__/ClientSchema.test.ts.snap
@@ -214,7 +214,7 @@ type ConversationMessageStreamPart @aws_cognito_user_pools {
   owner: String
   conversationId: ID!
   associatedUserMessageId: ID!
-  contentBlockIndex: Int!
+  contentBlockIndex: Int
   contentBlockText: String
   contentBlockDeltaIndex: Int
   contentBlockToolUse: ToolUseBlock
diff --git a/packages/data-schema/src/ai/ConversationSchemaTypes.ts b/packages/data-schema/src/ai/ConversationSchemaTypes.ts
@@ -235,7 +235,7 @@ const ConversationMessageStreamEvent = `type ConversationMessageStreamPart @aws_
   owner: String
   conversationId: ID!
   associatedUserMessageId: ID!
-  contentBlockIndex: Int!
+  contentBlockIndex: Int
   contentBlockText: String
   contentBlockDeltaIndex: Int
   contentBlockToolUse: ToolUseBlock
diff --git a/packages/integration-tests/__tests__/defined-behavior/2-expected-use/__snapshots__/ai-conversation.ts.snap b/packages/integration-tests/__tests__/defined-behavior/2-expected-use/__snapshots__/ai-conversation.ts.snap
@@ -562,7 +562,7 @@ type ConversationMessageStreamPart @aws_cognito_user_pools {
   owner: String
   conversationId: ID!
   associatedUserMessageId: ID!
-  contentBlockIndex: Int!
+  contentBlockIndex: Int
   contentBlockText: String
   contentBlockDeltaIndex: Int
   contentBlockToolUse: ToolUseBlock

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +---
 +"@aws-amplify/data-schema": patch
 +---
++
 +make ConversationMessageStreamPart.contentBlockIndex optional for error events