@aws-amplify/data-schema 1.7.0 → 1.8.1

package/package.json

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

package/src/ai/ConversationType.ts

@@ -56,7 +56,9 @@ export interface ConversationRoute {
    *
    * Deletes an existing {@link Conversation} based on ID.
    */
-    delete: (input: ConversationRouteDeleteInput) => SingularReturnValue<Conversation>;
+  delete: (
+    input: ConversationRouteDeleteInput,
+  ) => SingularReturnValue<Conversation>;
   /**
    * @experimental
    *
@@ -66,12 +68,14 @@ export interface ConversationRoute {
 }
 
 // conversation types
-interface ConversationSendMessageInput {
+interface ConversationSendMessageInputObject {
   content: ConversationSendMessageInputContent[];
   aiContext?: string | Record<string, any>;
   toolConfiguration?: ToolConfiguration;
 }
 
+export type ConversationSendMessageInput = ConversationSendMessageInputObject | string;
+
 interface ConversationListMessagesInput {
   limit?: number;
   nextToken?: string | null;
@@ -81,13 +85,18 @@ type ConversationOnMessageHandler = (message: ConversationMessage) => void;
 
 export interface Conversation {
   id: string;
+  createdAt: string;
+  updatedAt: string;
+
+  metadata?: Record<string, any>;
+  name?: string;
   /**
    * @experimental
    *
    * Sends a message to the current conversation.
    */
   sendMessage: (
-    input: ConversationSendMessageInput,
+    input: ConversationSendMessageInput | string,
   ) => SingularReturnValue<ConversationMessage>;
   /**
    * @experimental

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

@@ -17,9 +17,13 @@ export const convertItemToConversation = (
   client: BaseClient,
   modelIntrospection: ModelIntrospectionSchema,
   conversationId: string,
+  conversationCreatedAt: string,
+  conversationUpdatedAt: string,
   conversationRouteName: string,
   conversationMessageModel: SchemaModel,
   getInternals: ClientInternalsGetter,
+  conversationMetadata?: Record<string, any>,
+  conversationName?: string,
 ): Conversation => {
   if (!conversationId) {
     throw new Error(
@@ -28,6 +32,10 @@ export const convertItemToConversation = (
   }
   return {
     id: conversationId,
+    createdAt: conversationCreatedAt,
+    updatedAt: conversationUpdatedAt,
+    metadata: conversationMetadata,
+    name: conversationName,
     onMessage: createOnMessageFunction(
       client as BaseBrowserClient,
       modelIntrospection,

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

@@ -41,9 +41,13 @@ export const createCreateConversationFunction =
         client,
         modelIntrospection,
         data?.id,
+        data?.createdAt,
+        data?.updatedAt,
         conversationRouteName,
         conversationMessageModel,
         getInternals,
+        data?.metadata,
+        data?.name,
       ),
       errors,
     };

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

@@ -44,9 +44,13 @@ export const createDeleteConversationFunction =
             client,
             modelIntrospection,
             data?.id,
+            data?.createdAt,
+            data?.updatedAt,
             conversationRouteName,
             conversationMessageModel,
             getInternals,
+            data?.metadata,
+            data?.name,
           )
         : data,
       errors,

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

@@ -44,9 +44,13 @@ export const createGetConversationFunction =
             client,
             modelIntrospection,
             data.id,
+            data.createdAt,
+            data.updatedAt,
             conversationRouteName,
             conversationMessageModel,
             getInternals,
+            data?.metadata,
+            data?.name,
           )
         : data,
       errors,

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

@@ -41,9 +41,13 @@ export const createListConversationsFunction =
           client,
           modelIntrospection,
           datum.id,
+          datum.createdAt,
+          datum.updatedAt,
           conversationRouteName,
           conversationMessageModel,
           getInternals,
+          datum?.metadata,
+          datum?.name,
         );
       }),
       nextToken,

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,