@aws-amplify/data-schema 1.4.1 → 1.5.1

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

@@ -0,0 +1,91 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import { ConversationRoute } from '../../../../ai/ConversationType';
+import {
+  BaseClient,
+  ClientInternalsGetter,
+  GraphQLProviderConfig,
+  ModelIntrospectionSchema,
+} from '../../../bridge-types';
+import { createCreateConversationFunction } from '../../ai/createCreateConversationFunction';
+import { createGetConversationFunction } from '../../ai/createGetConversationFunction';
+import { createListConversationsFunction } from '../../ai/createListConversationsFunction';
+
+export function generateConversationsProperty(
+  client: BaseClient,
+  apiGraphQLConfig: GraphQLProviderConfig['GraphQL'],
+  getInternals: ClientInternalsGetter,
+): Record<string, ConversationRoute> {
+  const modelIntrospection: ModelIntrospectionSchema | undefined =
+    apiGraphQLConfig?.modelIntrospection;
+
+  // conversations will be absent from model intro schema if no conversation routes
+  // are present on the source schema.
+  if (!modelIntrospection?.conversations) {
+    return {};
+  }
+
+  const conversations: Record<string, ConversationRoute> = {};
+
+  for (const {
+    name,
+    conversation,
+    message,
+    models,
+    nonModels,
+    enums,
+  } of Object.values(modelIntrospection.conversations)) {
+    const conversationModel = models[conversation.modelName];
+    const conversationMessageModel = models[message.modelName];
+
+    if (!conversationModel || !conversationMessageModel) {
+      return {};
+    }
+
+    const conversationModelIntrospection: ModelIntrospectionSchema = {
+      ...modelIntrospection,
+      models: {
+        ...modelIntrospection.models,
+        ...models,
+      },
+      nonModels: {
+        ...modelIntrospection.nonModels,
+        ...nonModels,
+      },
+      enums: {
+        ...modelIntrospection.enums,
+        ...enums,
+      },
+    };
+
+    conversations[name] = {
+      create: createCreateConversationFunction(
+        client,
+        conversationModelIntrospection,
+        name,
+        conversationModel,
+        conversationMessageModel,
+        getInternals,
+      ),
+      get: createGetConversationFunction(
+        client,
+        conversationModelIntrospection,
+        name,
+        conversationModel,
+        conversationMessageModel,
+        getInternals,
+      ),
+      list: createListConversationsFunction(
+        client,
+        conversationModelIntrospection,
+        name,
+        conversationModel,
+        conversationMessageModel,
+        getInternals,
+      ),
+    };
+  }
+
+  return conversations;
+}

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

@@ -0,0 +1,41 @@
+// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+// SPDX-License-Identifier: Apache-2.0
+
+import {
+  BaseClient,
+  ClientInternalsGetter,
+  GraphQLProviderConfig,
+  ModelIntrospectionSchema,
+} from '../../../bridge-types';
+import { CustomQueries } from '../../../client';
+import { customOpFactory } from '../../operations/custom';
+
+export function generateGenerationsProperty<T extends Record<any, any>>(
+  client: BaseClient,
+  apiGraphQLConfig: GraphQLProviderConfig['GraphQL'],
+  getInternals: ClientInternalsGetter,
+): CustomQueries<T> {
+  const modelIntrospection: ModelIntrospectionSchema | undefined =
+    apiGraphQLConfig?.modelIntrospection;
+
+  // generations will be absent from model intro schema if no generation routes
+  // are present on the source schema.
+  if (!modelIntrospection?.generations) {
+    return {} as CustomQueries<T>;
+  }
+
+  const generations: Record<string, any> = {};
+
+  for (const generation of Object.values(modelIntrospection.generations)) {
+    generations[generation.name] = customOpFactory(
+      client,
+      modelIntrospection,
+      'query',
+      generation,
+      false,
+      getInternals,
+    );
+  }
+
+  return generations as CustomQueries<T>;
+}