@aws-amplify/data-schema 1.6.3 → 1.8.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aws-amplify/data-schema",
-  "version": "1.6.3",
+  "version": "1.8.0",
   "license": "Apache-2.0",
   "repository": {
     "type": "git",

package/src/ai/ConversationType.ts

@@ -29,6 +29,10 @@ interface ConversationRouteGetInput {
   id: string;
 }
 
+interface ConversationRouteDeleteInput {
+  id: string;
+}
+
 interface ConversationRouteListInput {
   limit?: number;
   nextToken?: string | null;
@@ -47,6 +51,14 @@ export interface ConversationRoute {
    * Gets an existing {@link Conversation} based on ID.
    */
   get: (input: ConversationRouteGetInput) => SingularReturnValue<Conversation>;
+  /**
+   * @experimental
+   *
+   * Deletes an existing {@link Conversation} based on ID.
+   */
+  delete: (
+    input: ConversationRouteDeleteInput,
+  ) => SingularReturnValue<Conversation>;
   /**
    * @experimental
    *
@@ -56,7 +68,7 @@ export interface ConversationRoute {
 }
 
 // conversation types
-interface ConversationSendMessageInput {
+export interface ConversationSendMessageInput {
   content: ConversationSendMessageInputContent[];
   aiContext?: string | Record<string, any>;
   toolConfiguration?: ToolConfiguration;
@@ -77,7 +89,7 @@ export interface Conversation {
    * Sends a message to the current conversation.
    */
   sendMessage: (
-    input: ConversationSendMessageInput,
+    input: ConversationSendMessageInput | string,
   ) => SingularReturnValue<ConversationMessage>;
   /**
    * @experimental

package/src/runtime/internals/ai/createDeleteConversationFunction.ts

@@ -0,0 +1,54 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import type { ConversationRoute } from '../../../ai/ConversationType';
+import type { SingularReturnValue } from '../../../runtime/client';
+import {
+  BaseClient,
+  ClientInternalsGetter,
+  ModelIntrospectionSchema,
+  SchemaModel,
+} from '../../bridge-types';
+import { getFactory } from '../operations/get';
+import { convertItemToConversation } from './convertItemToConversation';
+import {
+  AiAction,
+  getCustomUserAgentDetails,
+} from './getCustomUserAgentDetails';
+
+export const createDeleteConversationFunction =
+  (
+    client: BaseClient,
+    modelIntrospection: ModelIntrospectionSchema,
+    conversationRouteName: string,
+    conversationModel: SchemaModel,
+    conversationMessageModel: SchemaModel,
+    getInternals: ClientInternalsGetter,
+  ): ConversationRoute['delete'] =>
+  async ({ id }) => {
+    const deleteOperation = getFactory(
+      client,
+      modelIntrospection,
+      conversationModel,
+      'DELETE',
+      getInternals,
+      false,
+      getCustomUserAgentDetails(AiAction.DeleteConversation),
+    ) as (
+      args?: Record<string, any>,
+    ) => SingularReturnValue<Record<string, any>>;
+    const { data, errors } = await deleteOperation({ id });
+    return {
+      data: data
+        ? convertItemToConversation(
+            client,
+            modelIntrospection,
+            data?.id,
+            conversationRouteName,
+            conversationMessageModel,
+            getInternals,
+          )
+        : data,
+      errors,
+    };
+  };

package/src/runtime/internals/ai/createSendMessageFunction.ts

@@ -5,6 +5,7 @@ import { SingularReturnValue } from '../..';
 import type {
   Conversation,
   ConversationMessage,
+  ConversationSendMessageInput,
 } from '../../../ai/ConversationType';
 import {
   BaseClient,
@@ -31,13 +32,19 @@ export const createSendMessageFunction =
     conversationRouteName: string,
     getInternals: ClientInternalsGetter,
   ): Conversation['sendMessage'] =>
-  async ({ aiContext, content, toolConfiguration }) => {
+  async (input: ConversationSendMessageInput | string) => {
     const { conversations } = modelIntrospection;
 
     // Safe guard for standalone function. When called as part of client generation, this should never be falsy.
     if (!conversations) {
       return {} as SingularReturnValue<ConversationMessage>;
     }
+
+    const processedInput: ConversationSendMessageInput =
+      typeof input === 'string' ? { content: [{ text: input }] } : input;
+
+    const { content, aiContext, toolConfiguration } = processedInput;
+
     const sendSchema = conversations[conversationRouteName].message.send;
     const sendOperation = customOpFactory(
       client,

package/src/runtime/internals/ai/getCustomUserAgentDetails.ts

@@ -21,11 +21,11 @@ export enum AiAction {
   CreateConversation = '1',
   GetConversation = '2',
   ListConversations = '3',
-  SendMessage = '4',
-  ListMessages = '5',
-  OnMessage = '6',
-  Generation = '7',
-  Delete = '8',
+  DeleteConversation = '4',
+  SendMessage = '5',
+  ListMessages = '6',
+  OnMessage = '7',
+  Generation = '8',
 }
 
 export const getCustomUserAgentDetails = (

package/src/runtime/internals/utils/clientProperties/generateConversationsProperty.ts

@@ -11,6 +11,7 @@ import {
 import { createCreateConversationFunction } from '../../ai/createCreateConversationFunction';
 import { createGetConversationFunction } from '../../ai/createGetConversationFunction';
 import { createListConversationsFunction } from '../../ai/createListConversationsFunction';
+import { createDeleteConversationFunction } from '../../ai/createDeleteConversationFunction';
 
 export function generateConversationsProperty(
   client: BaseClient,
@@ -76,6 +77,14 @@ export function generateConversationsProperty(
         conversationMessageModel,
         getInternals,
       ),
+      delete: createDeleteConversationFunction(
+        client,
+        conversationModelIntrospection,
+        name,
+        conversationModel,
+        conversationMessageModel,
+        getInternals,
+      ),
       list: createListConversationsFunction(
         client,
         conversationModelIntrospection,