@deenruv/replicate-plugin 1.0.0

package/LICENSE

@@ -0,0 +1,23 @@
+# License 1
+
+The MIT License
+
+Copyright (c) 2025-present Aexol
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# License 2
+
+The MIT License
+
+Copyright (c) 2018-2025 Michael Bromley
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

package/README.md

@@ -0,0 +1,43 @@
+# @deenruv/replicate-plugin
+
+An AI-powered plugin that integrates [Replicate](https://replicate.com/) machine learning models with your Deenruv server for model training and order prediction tasks.
+
+## Installation
+
+```bash
+pnpm add @deenruv/replicate-plugin replicate
+```
+
+## Configuration
+
+```typescript
+import { ReplicatePlugin } from '@deenruv/replicate-plugin';
+
+// In your Deenruv server config:
+plugins: [
+  ReplicatePlugin.init({
+    deploymentName: 'your-replicate-deployment',
+    apiToken: process.env.REPLICATE_API_TOKEN,
+  }),
+]
+```
+
+## Features
+
+- Integration with Replicate AI deployment API
+- Model training based on order history data
+- Order export and prediction capabilities
+- Custom entity for tracking Replicate job status
+- Admin API for managing AI model operations
+- Webhook controller for receiving Replicate prediction results
+
+## Admin UI
+
+This plugin extends the admin UI with a Replicate management dashboard for configuring deployments, initiating model training, viewing prediction results, and monitoring job statuses.
+
+## API Extensions
+
+The plugin extends the **Admin API** with:
+
+- Mutations for triggering model training and order predictions
+- Queries for retrieving Replicate job status and prediction results

package/dist/plugin-server/constants.d.ts

@@ -0,0 +1,2 @@
+export declare const REPLICATE_PLUGIN_OPTIONS: unique symbol;
+export declare const LOGGER_CTX = "ReplicatePlugin";

package/dist/plugin-server/constants.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LOGGER_CTX = exports.REPLICATE_PLUGIN_OPTIONS = void 0;
+exports.REPLICATE_PLUGIN_OPTIONS = Symbol("options");
+exports.LOGGER_CTX = "ReplicatePlugin";

package/dist/plugin-server/controllers/replicate.d.ts

@@ -0,0 +1,22 @@
+import { Response } from "express";
+import { ReplicateService } from "../services/replicate.service";
+import { RequestContext } from "@deenruv/core";
+import { PredictionType } from "../graphql/generated-admin-types.js";
+export declare class ReplicateController {
+    private replicateService;
+    constructor(replicateService: ReplicateService);
+    synchronize(res: Response, ctx: RequestContext, { input, }: {
+        input: {
+            numLastOrder: number;
+            startDate: string;
+            endDate: string;
+        };
+    }): Promise<Response<any, Record<string, any>>>;
+    exportOrder(res: Response, ctx: RequestContext, input: {
+        numLastOrder: number;
+        startDate: string;
+        endDate: string;
+        predictType: PredictionType;
+        showMetrics: boolean;
+    }): Promise<Response<any, Record<string, any>>>;
+}

package/dist/plugin-server/controllers/replicate.js

@@ -0,0 +1,56 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ReplicateController = void 0;
+const common_1 = require("@nestjs/common");
+const replicate_service_1 = require("../services/replicate.service");
+const core_1 = require("@deenruv/core");
+const graphql_1 = require("@nestjs/graphql");
+let ReplicateController = class ReplicateController {
+    constructor(replicateService) {
+        this.replicateService = replicateService;
+    }
+    async synchronize(res, ctx, { input, }) {
+        await this.replicateService.modelTrainingJob(ctx, input);
+        return res.send("Job started  - model training started");
+    }
+    async exportOrder(res, ctx, input) {
+        const prediction_id = await this.replicateService.orderExportJob(ctx, input);
+        return res.send(prediction_id);
+    }
+};
+exports.ReplicateController = ReplicateController;
+__decorate([
+    (0, common_1.Get)("training/start"),
+    __param(0, (0, common_1.Res)()),
+    __param(1, (0, core_1.Ctx)()),
+    __param(2, (0, graphql_1.Args)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, core_1.RequestContext, Object]),
+    __metadata("design:returntype", Promise)
+], ReplicateController.prototype, "synchronize", null);
+__decorate([
+    (0, common_1.Get)("order/export"),
+    __param(0, (0, common_1.Res)()),
+    __param(1, (0, core_1.Ctx)()),
+    __param(2, (0, common_1.Body)()),
+    __metadata("design:type", Function),
+    __metadata("design:paramtypes", [Object, core_1.RequestContext, Object]),
+    __metadata("design:returntype", Promise)
+], ReplicateController.prototype, "exportOrder", null);
+exports.ReplicateController = ReplicateController = __decorate([
+    (0, common_1.Controller)("api/replicate"),
+    __param(0, (0, common_1.Inject)(replicate_service_1.ReplicateService)),
+    __metadata("design:paramtypes", [replicate_service_1.ReplicateService])
+], ReplicateController);

package/dist/plugin-server/e2e/order-export.e2e.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/plugin-server/e2e/order-export.e2e.test.js

@@ -0,0 +1,163 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const core_1 = require("@deenruv/core");
+const testing_1 = require("@deenruv/testing");
+const node_test_1 = __importStar(require("node:test"));
+const index_js_1 = require("../index.js");
+const replicate_service_js_1 = require("../services/replicate.service.js");
+(0, testing_1.registerInitializer)("postgres", new testing_1.PostgresInitializer());
+(0, node_test_1.describe)("integration tests with replicate token", async () => {
+    const deploymentName = "test";
+    const apiToken = "test";
+    index_js_1.ReplicatePlugin.init({ deploymentName: deploymentName, apiToken: apiToken });
+    if (!apiToken || !deploymentName)
+        throw new Error("impossible");
+    const config = Object.assign(Object.assign({}, testing_1.testConfig), { dbConnectionOptions: {
+            type: "postgres",
+            host: "localhost",
+            port: 5432,
+            username: "deenruv",
+            password: "deenruv",
+            database: "deenruv",
+            schema: "public",
+            logging: ["error", "warn", "schema", "query", "info", "log"],
+        }, plugins: [
+            core_1.DefaultJobQueuePlugin.init({}),
+            index_js_1.ReplicatePlugin.init({ deploymentName, apiToken }),
+        ] });
+    const { server, adminClient } = (0, testing_1.createTestEnvironment)(config);
+    let app;
+    let ctx;
+    (0, node_test_1.before)(async () => {
+        try {
+            await server.init({
+                initialData: {
+                    defaultLanguage: core_1.LanguageCode.en,
+                    defaultZone: "Europe",
+                    countries: [{ name: "Poland", code: "PL", zone: "Europe" }],
+                    taxRates: [],
+                    shippingMethods: [],
+                    paymentMethods: [],
+                    collections: [],
+                },
+                customerCount: 1,
+            });
+            app = server.app;
+            await app.get(core_1.JobQueueService).start();
+            const { superadminCredentials } = app.get(core_1.ConfigService).authOptions;
+            const superAdminUser = await app
+                .get(core_1.TransactionalConnection)
+                .getRepository(core_1.User)
+                .findOneOrFail({
+                where: { identifier: superadminCredentials.identifier },
+            });
+            ctx = await app.get(core_1.RequestContextService).create({
+                apiType: "admin",
+                user: superAdminUser,
+            });
+        }
+        catch (error) {
+            console.error("Error:", error);
+        }
+    });
+    (0, node_test_1.after)(async () => {
+        await server.destroy();
+    });
+    (0, node_test_1.default)("test order export job - model training", async () => {
+        const productVariantService = app.get(core_1.ProductVariantService);
+        const replicateService = app.get(replicate_service_js_1.ReplicateService);
+        const orderService = app.get(core_1.OrderService);
+        const productService = app.get(core_1.ProductService);
+        const assetService = app.get(core_1.AssetService);
+        // await productService.create(ctx, {
+        //     translations: [
+        //         {
+        //             languageCode: LanguageCode.en,
+        //             name: 'Test Product',
+        //             slug: 'test-product',
+        //             description: 'A product for testing purposes',
+        //         },
+        //     ],
+        //     enabled: true,
+        // });
+        // const assetInput: CreateAssetInput = {
+        //     file: {
+        //         createReadStream: () => fs.createReadStream(__dirname + '/fixtures/test_product_asset.png'),
+        //         filename: __dirname + '/fixtures/test_product_asset.png',
+        //         mimetype: 'image/png',
+        //     },
+        //     tags: ['test', 'product'],
+        //     customFields: {
+        //         field1: 'value1',
+        //         field2: 'value2',
+        //     },
+        // };
+        // await assetService.create(ctx, assetInput);
+        // await productVariantService.create(ctx, [
+        //     {
+        //         productId: 1,
+        //         translations: [
+        //             {
+        //                 languageCode: LanguageCode.en,
+        //                 name: 'Test Product Variant',
+        //             },
+        //         ],
+        //         facetValueIds: [],
+        //         sku: 'TEST-SKU',
+        //         price: 1000,
+        //         optionIds: [],
+        //         featuredAssetId: 1,
+        //         assetIds: [],
+        //     },
+        // ]);
+        await productVariantService.create(ctx, [
+            {
+                productId: 1,
+                translations: [
+                    {
+                        languageCode: core_1.LanguageCode.en,
+                        name: "Test Product Variant",
+                    },
+                ],
+                facetValueIds: [],
+                sku: "TEST-SKU",
+                price: 1000,
+                optionIds: [],
+                featuredAssetId: 1,
+                assetIds: [],
+            },
+        ]);
+        // const numLastOrder = 100;
+        // await replicateService.startModelTraining(ctx, {
+        //     numLastOrder,
+        // });
+        // const ordersCount = (await orderService.findAll(ctx, {})).totalItems;
+        // ok(
+        //     ordersCount >= numLastOrder,
+        //     'Total number of orders should be equal or greater than numLastOrder',
+        // );
+    });
+});

package/dist/plugin-server/e2e/with-token.e2e.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/plugin-server/e2e/with-token.e2e.test.js

@@ -0,0 +1,72 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const core_1 = require("@deenruv/core");
+const testing_1 = require("@deenruv/testing");
+const node_test_1 = __importStar(require("node:test"));
+const index_js_1 = require("../index.js");
+(0, testing_1.registerInitializer)("postgres", new testing_1.PostgresInitializer());
+(0, node_test_1.describe)("integration tests with replicate token", async () => {
+    const deploymentName = "test";
+    const apiToken = "test";
+    index_js_1.ReplicatePlugin.init({ deploymentName: deploymentName, apiToken: apiToken });
+    if (!apiToken || !deploymentName)
+        throw new Error("impossible");
+    const config = Object.assign(Object.assign({}, testing_1.testConfig), { dbConnectionOptions: {
+            type: "postgres",
+            host: "localhost",
+            port: 5432,
+            username: "deenruv",
+            password: "deenruv",
+            database: "deenruv",
+            schema: "public",
+        }, plugins: [
+            core_1.DefaultJobQueuePlugin.init({}),
+            index_js_1.ReplicatePlugin.init({ deploymentName, apiToken }),
+        ] });
+    const { server, adminClient } = (0, testing_1.createTestEnvironment)(config);
+    let app;
+    (0, node_test_1.before)(async () => {
+        await server.init({
+            initialData: {
+                defaultLanguage: core_1.LanguageCode.en,
+                defaultZone: "Europe",
+                countries: [{ name: "Poland", code: "PL", zone: "Europe" }],
+                taxRates: [],
+                shippingMethods: [],
+                paymentMethods: [],
+                collections: [],
+            },
+        });
+        app = server.app;
+        await app.get(core_1.JobQueueService).start();
+    });
+    (0, node_test_1.after)(async () => {
+        await server.destroy();
+    });
+    (0, node_test_1.default)("test model training integration controller api", async () => {
+        await adminClient.asSuperAdmin();
+    });
+});

package/dist/plugin-server/e2e/without-token.e2e.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/plugin-server/e2e/without-token.e2e.test.js

@@ -0,0 +1,75 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+const core_1 = require("@deenruv/core");
+const testing_1 = require("@deenruv/testing");
+const node_test_1 = __importStar(require("node:test"));
+const index_js_1 = require("../index.js");
+const constants_js_1 = require("../constants.js");
+const assert_1 = require("assert");
+(0, testing_1.registerInitializer)("postgres", new testing_1.PostgresInitializer());
+(0, node_test_1.describe)("integration tests without replicate token", async () => {
+    const deploymentName = "test";
+    const apiToken = "test";
+    if (!apiToken || !deploymentName)
+        throw new Error("impossible");
+    const config = Object.assign(Object.assign({}, testing_1.testConfig), { dbConnectionOptions: {
+            type: "postgres",
+            host: "localhost",
+            port: 5432,
+            username: "deenruv",
+            password: "deenruv",
+            database: "deenruv",
+            schema: "public",
+        }, plugins: [
+            core_1.DefaultJobQueuePlugin.init({}),
+            index_js_1.ReplicatePlugin.init({
+                deploymentName,
+                apiToken,
+            }),
+        ] });
+    const { server } = (0, testing_1.createTestEnvironment)(config);
+    let app;
+    (0, node_test_1.before)(async () => {
+        await server.init({
+            initialData: {
+                defaultLanguage: core_1.LanguageCode.en,
+                defaultZone: "Europe",
+                countries: [{ name: "Poland", code: "PL", zone: "Europe" }],
+                taxRates: [],
+                shippingMethods: [],
+                paymentMethods: [],
+                collections: [],
+            },
+        });
+        app = server.app;
+    });
+    (0, node_test_1.after)(async () => {
+        await server.destroy();
+    });
+    (0, node_test_1.default)("plugin loads api key", () => {
+        (0, assert_1.equal)(app.get(constants_js_1.REPLICATE_PLUGIN_OPTIONS).apiToken, "test");
+    });
+});

package/dist/plugin-server/entites/replicate.entity.d.ts

@@ -0,0 +1,13 @@
+import { DeenruvEntity } from "@deenruv/core";
+import type { DeepPartial, ID } from "@deenruv/core";
+import "reflect-metadata";
+export declare class ReplicateEntity extends DeenruvEntity {
+    constructor(input?: DeepPartial<ReplicateEntity>);
+    prediction_id: string;
+    output: [{
+        customerId: ID;
+        score: number;
+    }];
+    status: string;
+    finishedAt: Date | null;
+}

package/dist/plugin-server/entites/replicate.entity.js

@@ -0,0 +1,41 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ReplicateEntity = void 0;
+const core_1 = require("@deenruv/core");
+const typeorm_1 = require("typeorm");
+require("reflect-metadata");
+let ReplicateEntity = class ReplicateEntity extends core_1.DeenruvEntity {
+    constructor(input) {
+        super(input);
+    }
+};
+exports.ReplicateEntity = ReplicateEntity;
+__decorate([
+    (0, typeorm_1.Column)({ nullable: true, type: "varchar", length: 255 }),
+    __metadata("design:type", String)
+], ReplicateEntity.prototype, "prediction_id", void 0);
+__decorate([
+    (0, typeorm_1.Column)("simple-json", { nullable: true }),
+    __metadata("design:type", Array)
+], ReplicateEntity.prototype, "output", void 0);
+__decorate([
+    (0, typeorm_1.Column)({ default: "starting" }),
+    __metadata("design:type", String)
+], ReplicateEntity.prototype, "status", void 0);
+__decorate([
+    (0, typeorm_1.Column)({ type: Date, nullable: true }),
+    __metadata("design:type", Object)
+], ReplicateEntity.prototype, "finishedAt", void 0);
+exports.ReplicateEntity = ReplicateEntity = __decorate([
+    (0, typeorm_1.Entity)(),
+    __metadata("design:paramtypes", [Object])
+], ReplicateEntity);

package/dist/plugin-server/extensions/replicate.extension.d.ts

@@ -0,0 +1 @@
+export declare const AdminExtension: import("graphql").DocumentNode;

package/dist/plugin-server/extensions/replicate.extension.js

@@ -0,0 +1,78 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AdminExtension = void 0;
+const graphql_tag_1 = require("graphql-tag");
+exports.AdminExtension = (0, graphql_tag_1.gql) `
+  enum PredictionType {
+    RFM_SCORE
+    SEGMENTATION
+  }
+  input StartOrderExportToReplicateInput {
+    numLastOrder: Int
+    startDate: DateTime
+    endDate: DateTime
+    predictType: PredictionType
+    showMetrics: Boolean
+  }
+  input startModelTrainingInput {
+    numLastOrder: Int
+    startDate: DateTime
+    endDate: DateTime
+  }
+
+  extend type Mutation {
+    startOrderExportToReplicate(
+      input: StartOrderExportToReplicateInput!
+    ): String!
+  }
+
+  extend type Mutation {
+    startModelTraining(input: startModelTrainingInput!): String!
+  }
+
+  input GetPredictionInput {
+    prediction_id: String!
+  }
+
+  input GetPredictionEntityInput {
+    prediction_entity_id: String!
+  }
+
+  enum PredictionStatus {
+    starting
+    succeeded
+    failed
+  }
+
+  type Prediction {
+    id: String!
+    score: Float!
+    customer: Customer
+  }
+
+  type PredictionResult {
+    status: PredictionStatus!
+    predictions: [Prediction!]
+  }
+
+  input ReplicateEntityListOptions
+
+  extend type Query {
+    getPredictionID(input: GetPredictionEntityInput!): String
+    getReplicatePredictions(
+      options: ReplicateEntityListOptions
+    ): ReplicateEntityList!
+    getPredictionItem(id: String!): PredictionResult!
+  }
+
+  type ReplicateEntity implements Node {
+    id: ID!
+    status: PredictionStatus!
+    finishedAt: DateTime
+  }
+
+  type ReplicateEntityList implements PaginatedList {
+    items: [ReplicateEntity!]!
+    totalItems: Int!
+  }
+`;