support client side tools

aws-amplify · Aug 8, 2024 · 529f7a1 · 529f7a1
1 parent c52a7f3
commit 529f7a1
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 17 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
@@ -165,17 +165,19 @@ exports[`Conversation Route Introspection Visitor Metadata snapshot should gener
                             \\"isRequired\\": false,
                             \\"attributes\\": []
                         },
-                        \\"context\\": {
-                            \\"name\\": \\"context\\",
+                        \\"aiContext\\": {
+                            \\"name\\": \\"aiContext\\",
                             \\"isArray\\": false,
                             \\"type\\": \\"AWSJSON\\",
                             \\"isRequired\\": false,
                             \\"attributes\\": []
                         },
-                        \\"uiComponents\\": {
-                            \\"name\\": \\"uiComponents\\",
-                            \\"isArray\\": true,
-                            \\"type\\": \\"AWSJSON\\",
+                        \\"toolConfiguration\\": {
+                            \\"name\\": \\"toolConfiguration\\",
+                            \\"isArray\\": false,
+                            \\"type\\": {
+                                \\"nonModel\\": \\"ToolConfiguration\\"
+                            },
                             \\"isRequired\\": false,
                             \\"attributes\\": [],
                             \\"isArrayNullable\\": true

diff --git a/...ppsync-modelgen-plugin/src/__tests__/visitors/appsync-model-introspection-visitor.test.ts b/...ppsync-modelgen-plugin/src/__tests__/visitors/appsync-model-introspection-visitor.test.ts
@@ -39,15 +39,17 @@ describe('Conversation Route Introspection Visitor', () => {
       id: ID!
       conversationId: ID!
       role: ConversationParticipantRole
-      # TODO: Update to ContentBlock -- need the types
-      content: String
-      context: AWSJSON
-      uiComponents: [AWSJSON]
+      content: [ContentBlock]
+      aiContext: AWSJSON
+      toolConfiguration: ToolConfiguration
+      createdAt: AWSDateTime
+      updatedAt: AWSDateTime
+      owner: String
     }
 
     type Mutation {
       # TODO: Update to ContentBlock -- need the types
-      pirateChat(conversationId: ID, content: String): ConversationMessage
+      pirateChat(conversationId: ID, content: [ContentBlockInput], aiContext: AWSJSON, toolConfiguration: ToolConfigurationInput): ConversationMessage
       @conversation(aiModel: "Claude3Haiku", functionName: "conversation-handler")
     }
   `;

diff --git a/packages/appsync-modelgen-plugin/src/visitors/appsync-model-introspection-visitor.ts b/packages/appsync-modelgen-plugin/src/visitors/appsync-model-introspection-visitor.ts
@@ -403,17 +403,17 @@ export class AppSyncModelIntrospectionVisitor<
           isRequired: false,
           attributes: []
         },
-        context: {
-          name: 'context',
+        aiContext: {
+          name: 'aiContext',
           isArray: false,
           type: 'AWSJSON',
           isRequired: false,
           attributes: []
         },
-        uiComponents: {
-          name: 'uiComponents',
-          isArray: true,
-          type: 'AWSJSON',
+        toolConfiguration: {
+          name: 'toolConfiguration',
+          isArray: false,
+          type: { nonModel: 'ToolConfiguration' },
           isRequired: false,
           attributes: [],
           isArrayNullable: true