@ampsec/platform-client 57.0.0 → 57.1.0

package/build/src/dto/enums/aiServiceModelName.d.ts

@@ -0,0 +1,6 @@
+export declare enum AIServiceModelName {
+    GPT3 = "gpt-3.5-turbo-16k",
+    GPT4 = "gpt-4",
+    TEXT_BISON = "text-bison-32k",
+    CODE_BISON = "code-bison-32k"
+}

package/build/src/dto/enums/aiServiceModelName.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIServiceModelName = void 0;
+/* eslint-disable no-unused-vars */
+var AIServiceModelName;
+(function (AIServiceModelName) {
+    AIServiceModelName["GPT3"] = "gpt-3.5-turbo-16k";
+    AIServiceModelName["GPT4"] = "gpt-4";
+    AIServiceModelName["TEXT_BISON"] = "text-bison-32k";
+    AIServiceModelName["CODE_BISON"] = "code-bison-32k";
+})(AIServiceModelName || (exports.AIServiceModelName = AIServiceModelName = {}));
+//# sourceMappingURL=aiServiceModelName.js.map

package/build/src/dto/enums/aiServiceModelName.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"aiServiceModelName.js","sourceRoot":"","sources":["../../../../src/dto/enums/aiServiceModelName.ts"],"names":[],"mappings":";;;AAAA,mCAAmC;AACnC,IAAY,kBAKX;AALD,WAAY,kBAAkB;IAC5B,gDAA0B,CAAA;IAC1B,oCAAc,CAAA;IACd,mDAA6B,CAAA;IAC7B,mDAA6B,CAAA;AAC/B,CAAC,EALW,kBAAkB,kCAAlB,kBAAkB,QAK7B"}

package/build/src/dto/enums/index.d.ts

@@ -23,3 +23,4 @@ export * from './saasComponentKind';
 export * from './supportedMessageChannelKind.enum';
 export * from './workflowStep.kind';
 export * from './workflowTrigger.kind';
+export * from './aiServiceModelName';

package/build/src/dto/enums/index.js

@@ -39,4 +39,5 @@ __exportStar(require("./saasComponentKind"), exports);
 __exportStar(require("./supportedMessageChannelKind.enum"), exports);
 __exportStar(require("./workflowStep.kind"), exports);
 __exportStar(require("./workflowTrigger.kind"), exports);
+__exportStar(require("./aiServiceModelName"), exports);
 //# sourceMappingURL=index.js.map

package/build/src/dto/enums/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/dto/enums/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8CAA4B;AAC5B,iDAA+B;AAC/B,wDAAsC;AACtC,sDAAoC;AACpC,6CAA2B;AAC3B,qDAAmC;AACnC,qDAAmC;AACnC,iDAA+B;AAC/B,yDAAuC;AACvC,0DAAwC;AACxC,gDAA8B;AAC9B,qDAAmC;AACnC,mDAAiC;AACjC,oDAAkC;AAClC,qDAAmC;AACnC,oDAAkC;AAClC,wDAAsC;AACtC,mDAAiC;AACjC,wDAAsC;AACtC,8DAA4C;AAC5C,yDAAuC;AACvC,sDAAoC;AACpC,qEAAmD;AACnD,sDAAoC;AACpC,yDAAuC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/dto/enums/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8CAA4B;AAC5B,iDAA+B;AAC/B,wDAAsC;AACtC,sDAAoC;AACpC,6CAA2B;AAC3B,qDAAmC;AACnC,qDAAmC;AACnC,iDAA+B;AAC/B,yDAAuC;AACvC,0DAAwC;AACxC,gDAA8B;AAC9B,qDAAmC;AACnC,mDAAiC;AACjC,oDAAkC;AAClC,qDAAmC;AACnC,oDAAkC;AAClC,wDAAsC;AACtC,mDAAiC;AACjC,wDAAsC;AACtC,8DAA4C;AAC5C,yDAAuC;AACvC,sDAAoC;AACpC,qEAAmD;AACnD,sDAAoC;AACpC,yDAAuC;AACvC,uDAAqC"}

package/build/src/services/AmpApi.d.ts

@@ -1,5 +1,5 @@
-import { AgentDto, AgentUpsertDto, AssetDto, FindingsInsightsDto, ProviderDto, ReportResultDto, SaasAssetDto, SaasComponentDto, SaasUserDto, TenantDto, TenantUpsertDto, UserDto } from '../dto';
-import { AmpEntityService, AmpDataService, AmpReportService, AmpSettingsService, ConnectorsService, FindingsService, NotificationService } from '.';
+import { AgentDto, AgentUpsertDto, AssetDto, ProviderDto, ReportResultDto, SaasAssetDto, SaasComponentDto, SaasUserDto, TenantDto, TenantUpsertDto, UserDto } from '../dto';
+import { AmpEntityService, AmpDataService, AmpReportService, AmpSettingsService, ConnectorsService, FindingsInsightsService, FindingsService, NotificationService, PredictionService } from '.';
 import { AmpRestClientOptions, AgentIdentityService, ConnectorInstallService, EnumService, RestClient } from './rest';
 export type AmpApiOptions = AmpRestClientOptions;
 /**
@@ -20,7 +20,7 @@ export declare class AmpApi {
     readonly connectors: ConnectorsService;
     readonly enums: EnumService;
     readonly findings: FindingsService;
-    readonly findingsInsights: AmpDataService<FindingsInsightsDto>;
+    readonly findingsInsights: FindingsInsightsService;
     readonly identity: AgentIdentityService;
     readonly install: ConnectorInstallService;
     readonly notifications: NotificationService;
@@ -33,6 +33,7 @@ export declare class AmpApi {
     readonly settings: AmpSettingsService;
     readonly tenants: AmpEntityService<TenantUpsertDto, TenantDto>;
     readonly users: AmpDataService<UserDto>;
+    readonly prediction: PredictionService;
     constructor(rest: RestClient);
     static instance(options: AmpApiOptions): AmpApi;
 }

package/build/src/services/AmpApi.js

@@ -23,7 +23,7 @@ class AmpApi {
         this.connectors = new _1.ConnectorsService(rest);
         this.enums = new rest_1.DefaultEnumService(rest);
         this.findings = new _1.FindingsService(rest);
-        this.findingsInsights = new _1.AmpDataServiceImpl(rest, constants_1.KIND.FINDINGS_INSIGHTS);
+        this.findingsInsights = new _1.FindingsInsightsService(rest, constants_1.KIND.FINDINGS_INSIGHTS);
         this.identity = new rest_1.AgentIdentityService(rest);
         this.install = new rest_1.ConnectorInstallService(rest, constants_1.TARGET_API_AGENT);
         this.notifications = new _1.NotificationService(rest);
@@ -36,6 +36,7 @@ class AmpApi {
         this.settings = new _1.AmpSettingsService(rest);
         this.tenants = new _1.AmpEntityServiceImpl(rest, constants_1.KIND.TENANTS);
         this.users = new _1.AmpDataServiceImpl(rest, constants_1.KIND.USERS);
+        this.prediction = new _1.PredictionService(rest, constants_1.KIND.FINDINGS);
     }
     static instance(options) {
         const rest = (0, rest_1.getAmpRestClient)(options);

package/build/src/services/AmpApi.js.map

@@ -1 +1 @@
-{"version":3,"file":"AmpApi.js","sourceRoot":"","sources":["../../../src/services/AmpApi.ts"],"names":[],"mappings":";;;AAcA,wBAWW;AACX,2CAAmD;AACnD,iCAA0J;AAI1J;;;;;;;;;;;GAWG;AACH,MAAa,MAAM;IAoBjB,YAAY,IAAgB;QAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,uBAAoB,CAA2B,IAAI,EAAE,gBAAI,CAAC,MAAM,CAAC,CAAC;QACpF,IAAI,CAAC,KAAK,GAAG,IAAI,qBAAkB,CAAW,IAAI,EAAE,gBAAI,CAAC,MAAM,CAAC,CAAC;QACjE,IAAI,CAAC,UAAU,GAAG,IAAI,oBAAiB,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK,GAAG,IAAI,yBAAkB,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,QAAQ,GAAG,IAAI,kBAAe,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,gBAAgB,GAAG,IAAI,qBAAkB,CAAsB,IAAI,EAAE,gBAAI,CAAC,iBAAiB,CAAC,CAAC;QAClG,IAAI,CAAC,QAAQ,GAAG,IAAI,2BAAoB,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,GAAG,IAAI,8BAAuB,CAAC,IAAI,EAAE,4BAAgB,CAAC,CAAC;QACnE,IAAI,CAAC,aAAa,GAAG,IAAI,sBAAmB,CAAC,IAAI,CAAC,CAAC;QACnD,IAAI,CAAC,SAAS,GAAG,IAAI,qBAAkB,CAAc,IAAI,EAAE,gBAAI,CAAC,SAAS,CAAC,CAAC;QAC3E,IAAI,CAAC,aAAa,GAAG,IAAI,qBAAkB,CAAkB,IAAI,EAAE,gBAAI,CAAC,cAAc,CAAC,CAAC;QACxF,IAAI,CAAC,OAAO,GAAG,IAAI,uBAAoB,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,UAAU,GAAG,IAAI,qBAAkB,CAAe,IAAI,EAAE,gBAAI,CAAC,WAAW,CAAC,CAAC;QAC/E,IAAI,CAAC,cAAc,GAAG,IAAI,qBAAkB,CAAmB,IAAI,EAAE,gBAAI,CAAC,eAAe,CAAC,CAAC;QAC3F,IAAI,CAAC,SAAS,GAAG,IAAI,qBAAkB,CAAc,IAAI,EAAE,gBAAI,CAAC,UAAU,CAAC,CAAC;QAC5E,IAAI,CAAC,QAAQ,GAAG,IAAI,qBAAkB,CAAC,IAAI,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,GAAG,IAAI,uBAAoB,CAA6B,IAAI,EAAE,gBAAI,CAAC,OAAO,CAAC,CAAC;QACxF,IAAI,CAAC,KAAK,GAAG,IAAI,qBAAkB,CAAU,IAAI,EAAE,gBAAI,CAAC,KAAK,CAAC,CAAC;IACjE,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,OAAsB;QACpC,MAAM,IAAI,GAAG,IAAA,uBAAgB,EAAC,OAAO,CAAC,CAAC;QACvC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AA7CD,wBA6CC"}
+{"version":3,"file":"AmpApi.js","sourceRoot":"","sources":["../../../src/services/AmpApi.ts"],"names":[],"mappings":";;;AACA,wBAaW;AACX,2CAAmD;AACnD,iCAA0J;AAI1J;;;;;;;;;;;GAWG;AACH,MAAa,MAAM;IAqBjB,YAAY,IAAgB;QAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,uBAAoB,CAA2B,IAAI,EAAE,gBAAI,CAAC,MAAM,CAAC,CAAC;QACpF,IAAI,CAAC,KAAK,GAAG,IAAI,qBAAkB,CAAW,IAAI,EAAE,gBAAI,CAAC,MAAM,CAAC,CAAC;QACjE,IAAI,CAAC,UAAU,GAAG,IAAI,oBAAiB,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK,GAAG,IAAI,yBAAkB,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,QAAQ,GAAG,IAAI,kBAAe,CAAC,IAAI,CAAC,CAAC;QAC1C,IAAI,CAAC,gBAAgB,GAAG,IAAI,0BAAuB,CAAC,IAAI,EAAE,gBAAI,CAAC,iBAAiB,CAAC,CAAC;QAClF,IAAI,CAAC,QAAQ,GAAG,IAAI,2BAAoB,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,OAAO,GAAG,IAAI,8BAAuB,CAAC,IAAI,EAAE,4BAAgB,CAAC,CAAC;QACnE,IAAI,CAAC,aAAa,GAAG,IAAI,sBAAmB,CAAC,IAAI,CAAC,CAAC;QACnD,IAAI,CAAC,SAAS,GAAG,IAAI,qBAAkB,CAAc,IAAI,EAAE,gBAAI,CAAC,SAAS,CAAC,CAAC;QAC3E,IAAI,CAAC,aAAa,GAAG,IAAI,qBAAkB,CAAkB,IAAI,EAAE,gBAAI,CAAC,cAAc,CAAC,CAAC;QACxF,IAAI,CAAC,OAAO,GAAG,IAAI,uBAAoB,CAAC,IAAI,CAAC,CAAC;QAC9C,IAAI,CAAC,UAAU,GAAG,IAAI,qBAAkB,CAAe,IAAI,EAAE,gBAAI,CAAC,WAAW,CAAC,CAAC;QAC/E,IAAI,CAAC,cAAc,GAAG,IAAI,qBAAkB,CAAmB,IAAI,EAAE,gBAAI,CAAC,eAAe,CAAC,CAAC;QAC3F,IAAI,CAAC,SAAS,GAAG,IAAI,qBAAkB,CAAc,IAAI,EAAE,gBAAI,CAAC,UAAU,CAAC,CAAC;QAC5E,IAAI,CAAC,QAAQ,GAAG,IAAI,qBAAkB,CAAC,IAAI,CAAC,CAAC;QAC7C,IAAI,CAAC,OAAO,GAAG,IAAI,uBAAoB,CAA6B,IAAI,EAAE,gBAAI,CAAC,OAAO,CAAC,CAAC;QACxF,IAAI,CAAC,KAAK,GAAG,IAAI,qBAAkB,CAAU,IAAI,EAAE,gBAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,IAAI,CAAC,UAAU,GAAG,IAAI,oBAAiB,CAAC,IAAI,EAAE,gBAAI,CAAC,QAAQ,CAAC,CAAC;IAC/D,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,OAAsB;QACpC,MAAM,IAAI,GAAG,IAAA,uBAAgB,EAAC,OAAO,CAAC,CAAC;QACvC,OAAO,IAAI,MAAM,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AA/CD,wBA+CC"}

package/build/src/services/AmpSdk.d.ts

@@ -7,7 +7,7 @@ import { EnumService } from './rest/EnumService';
 import { AmpSdkSettingsService } from './settings.service';
 import { AmpSaaSEntityService, AmpSdkSaasAssetService, AmpSdkSaasComponentService, AmpSdkSaasUserService } from './saasEntity.service';
 import { ContentfulService } from './contentful.service';
-import { PlatformFindingsInsightsDto, PlatformFindingsInsightsUpsertDto } from '../dto/platform/platform.findingsInsights.dto';
+import { FindingsInsightsService } from './findingsInsights.service';
 export type AmpSdkOptions = AmpRestClientOptions;
 /**
  * AMP API
@@ -25,7 +25,7 @@ export declare class AmpSdkServices {
     readonly contentful: ContentfulService;
     readonly enums: EnumService;
     readonly findings: AmpSaaSEntityService<PlatformFindingUpsertDto, PlatformFindingDto>;
-    readonly findingsInsights: AmpSaaSEntityService<PlatformFindingsInsightsUpsertDto, PlatformFindingsInsightsDto>;
+    readonly findingsInsights: FindingsInsightsService;
     readonly jobExecutions: AmpEntityService<PlatformJobExecutionStateUpsertDto, PlatformJobExecutionStateDto>;
     readonly jobs: AmpEntityService<PlatformJobSpecUpsertDto, PlatformJobSpecDto>;
     readonly notifications: AmpEntityService<PlatformNotificationUpsertDto, PlatformNotificationDto>;

package/build/src/services/AmpSdk.js

@@ -8,6 +8,7 @@ const EnumService_1 = require("./rest/EnumService");
 const settings_service_1 = require("./settings.service");
 const saasEntity_service_1 = require("./saasEntity.service");
 const contentful_service_1 = require("./contentful.service");
+const findingsInsights_service_1 = require("./findingsInsights.service");
 /**
  * AMP API
  * This client is a wrapper around the AMP REST API meant to be used by
@@ -25,7 +26,7 @@ class AmpSdkServices {
         this.contentful = new contentful_service_1.ContentfulService();
         this.enums = new EnumService_1.DefaultEnumService(rest, constants_1.TARGET_API_PLATFORM);
         this.findings = new saasEntity_service_1.AmpSaaSEntityServiceImpl(rest, constants_1.KIND.FINDINGS, constants_1.TARGET_API_PLATFORM);
-        this.findingsInsights = new saasEntity_service_1.AmpSaaSEntityServiceImpl(rest, constants_1.KIND.FINDINGS_INSIGHTS, constants_1.TARGET_API_PLATFORM);
+        this.findingsInsights = new findingsInsights_service_1.FindingsInsightsService(rest, constants_1.KIND.FINDINGS_INSIGHTS, constants_1.TARGET_API_PLATFORM);
         this.jobExecutions = new entity_service_1.AmpEntityServiceImpl(rest, constants_1.KIND.JOB_EXECUTIONS, constants_1.TARGET_API_PLATFORM);
         this.jobs = new entity_service_1.AmpEntityServiceImpl(rest, constants_1.KIND.JOB_SPEC, constants_1.TARGET_API_PLATFORM);
         this.notifications = new entity_service_1.AmpEntityServiceImpl(rest, constants_1.KIND.NOTIFICATIONS, constants_1.TARGET_API_PLATFORM);

package/build/src/services/AmpSdk.js.map

@@ -1 +1 @@
-{"version":3,"file":"AmpSdk.js","sourceRoot":"","sources":["../../../src/services/AmpSdk.ts"],"names":[],"mappings":";;;AAoBA,qDAAoI;AACpI,iCAA0E;AAC1E,2CAAsD;AAGtD,oDAAmE;AACnE,yDAAyD;AACzD,6DAA+J;AAC/J,6DAAuD;AAKvD;;;;;;;;GAQG;AACH,MAAa,cAAc;IAsBzB,YAAY,IAAgB;QAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,qCAAoB,CAA2C,IAAI,EAAE,gBAAI,CAAC,MAAM,EAAE,+BAAmB,CAAC,CAAC;QACzH,IAAI,CAAC,KAAK,GAAG,IAAI,mCAAkB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC/D,IAAI,CAAC,UAAU,GAAG,IAAI,qCAAoB,CAAmD,IAAI,EAAE,gBAAI,CAAC,UAAU,EAAE,+BAAmB,CAAC,CAAC;QACzI,IAAI,CAAC,UAAU,GAAG,IAAI,sCAAiB,EAAE,CAAC;QAC1C,IAAI,CAAC,KAAK,GAAG,IAAI,gCAAkB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,GAAG,IAAI,6CAAwB,CAA+C,IAAI,EAAE,gBAAI,CAAC,QAAQ,EAAE,+BAAmB,CAAC,CAAC;QACrI,IAAI,CAAC,gBAAgB,GAAG,IAAI,6CAAwB,CAAiE,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QACxK,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAmE,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QAChK,IAAI,CAAC,IAAI,GAAG,IAAI,qCAAoB,CAA+C,IAAI,EAAE,gBAAI,CAAC,QAAQ,EAAE,+BAAmB,CAAC,CAAC;QAC7H,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAyD,IAAI,EAAE,gBAAI,CAAC,aAAa,EAAE,+BAAmB,CAAC,CAAC;QACrJ,IAAI,CAAC,SAAS,GAAG,IAAI,qCAAoB,CAAyC,IAAI,EAAE,gBAAI,CAAC,SAAS,EAAE,+BAAmB,CAAC,CAAC;QAC7H,IAAI,CAAC,aAAa,GAAG,IAAI,6CAAwB,CAAyD,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QAC1J,IAAI,CAAC,gBAAgB,GAAG,IAAI,6CAAwB,CAA+D,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QACtK,IAAI,CAAC,UAAU,GAAG,IAAI,2CAAsB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QACxE,IAAI,CAAC,cAAc,GAAG,IAAI,+CAA0B,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,GAAG,IAAI,0CAAqB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QACtE,IAAI,CAAC,QAAQ,GAAG,IAAI,wCAAqB,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,GAAG,IAAI,qCAAoB,CAA6B,IAAI,EAAE,gBAAI,CAAC,OAAO,EAAE,+BAAmB,CAAC,CAAC;QAC7G,IAAI,CAAC,KAAK,GAAG,IAAI,kCAAiB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC9D,IAAI,CAAC,iBAAiB,GAAG,IAAI,qCAAoB,CAAiE,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;IACvK,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,OAAsB;QACpC,MAAM,IAAI,GAAG,IAAA,uBAAgB,EAAC,OAAO,CAAC,CAAC;QACvC,OAAO,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC;CACF;AAjDD,wCAiDC"}
+{"version":3,"file":"AmpSdk.js","sourceRoot":"","sources":["../../../src/services/AmpSdk.ts"],"names":[],"mappings":";;;AAoBA,qDAAoI;AACpI,iCAA0E;AAC1E,2CAAsD;AAGtD,oDAAmE;AACnE,yDAAyD;AACzD,6DAA+J;AAC/J,6DAAuD;AACvD,yEAAmE;AAInE;;;;;;;;GAQG;AACH,MAAa,cAAc;IAsBzB,YAAY,IAAgB;QAC1B,IAAI,CAAC,MAAM,GAAG,IAAI,qCAAoB,CAA2C,IAAI,EAAE,gBAAI,CAAC,MAAM,EAAE,+BAAmB,CAAC,CAAC;QACzH,IAAI,CAAC,KAAK,GAAG,IAAI,mCAAkB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC/D,IAAI,CAAC,UAAU,GAAG,IAAI,qCAAoB,CAAmD,IAAI,EAAE,gBAAI,CAAC,UAAU,EAAE,+BAAmB,CAAC,CAAC;QACzI,IAAI,CAAC,UAAU,GAAG,IAAI,sCAAiB,EAAE,CAAC;QAC1C,IAAI,CAAC,KAAK,GAAG,IAAI,gCAAkB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,GAAG,IAAI,6CAAwB,CAA+C,IAAI,EAAE,gBAAI,CAAC,QAAQ,EAAE,+BAAmB,CAAC,CAAC;QACrI,IAAI,CAAC,gBAAgB,GAAG,IAAI,kDAAuB,CAAC,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QACvG,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAmE,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QAChK,IAAI,CAAC,IAAI,GAAG,IAAI,qCAAoB,CAA+C,IAAI,EAAE,gBAAI,CAAC,QAAQ,EAAE,+BAAmB,CAAC,CAAC;QAC7H,IAAI,CAAC,aAAa,GAAG,IAAI,qCAAoB,CAAyD,IAAI,EAAE,gBAAI,CAAC,aAAa,EAAE,+BAAmB,CAAC,CAAC;QACrJ,IAAI,CAAC,SAAS,GAAG,IAAI,qCAAoB,CAAyC,IAAI,EAAE,gBAAI,CAAC,SAAS,EAAE,+BAAmB,CAAC,CAAC;QAC7H,IAAI,CAAC,aAAa,GAAG,IAAI,6CAAwB,CAAyD,IAAI,EAAE,gBAAI,CAAC,cAAc,EAAE,+BAAmB,CAAC,CAAC;QAC1J,IAAI,CAAC,gBAAgB,GAAG,IAAI,6CAAwB,CAA+D,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;QACtK,IAAI,CAAC,UAAU,GAAG,IAAI,2CAAsB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QACxE,IAAI,CAAC,cAAc,GAAG,IAAI,+CAA0B,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,GAAG,IAAI,0CAAqB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QACtE,IAAI,CAAC,QAAQ,GAAG,IAAI,wCAAqB,CAAC,IAAI,CAAC,CAAC;QAChD,IAAI,CAAC,OAAO,GAAG,IAAI,qCAAoB,CAA6B,IAAI,EAAE,gBAAI,CAAC,OAAO,EAAE,+BAAmB,CAAC,CAAC;QAC7G,IAAI,CAAC,KAAK,GAAG,IAAI,kCAAiB,CAAC,IAAI,EAAE,+BAAmB,CAAC,CAAC;QAC9D,IAAI,CAAC,iBAAiB,GAAG,IAAI,qCAAoB,CAAiE,IAAI,EAAE,gBAAI,CAAC,iBAAiB,EAAE,+BAAmB,CAAC,CAAC;IACvK,CAAC;IAED,MAAM,CAAC,QAAQ,CAAC,OAAsB;QACpC,MAAM,IAAI,GAAG,IAAA,uBAAgB,EAAC,OAAO,CAAC,CAAC;QACvC,OAAO,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;IAClC,CAAC;CACF;AAjDD,wCAiDC"}

package/build/src/services/findingsInsights.service.d.ts

@@ -0,0 +1,25 @@
+import { RestClient, RestRequest } from './rest';
+import { ErrorHandler, TargetApi } from './constants';
+import { z } from 'zod';
+import { FilterCriteria } from '../FilterCriteria';
+export declare const AmpFindingsFilterCriteriaSchema: any;
+export type AmpFindingsFilterCriteria = z.infer<typeof AmpFindingsFilterCriteriaSchema>;
+export declare class FindingsInsightsService {
+    protected call: <T>(req: RestRequest, errorHandler: ErrorHandler<T>) => Promise<T>;
+    protected getPage: (req: RestRequest) => Promise<{
+        data: never[];
+        kind: string;
+        error: string;
+        hints: {};
+    }>;
+    protected readonly rest: RestClient;
+    protected readonly kind: string;
+    protected readonly targetApi: string;
+    constructor(rest: RestClient, kind: string, targetApi?: TargetApi);
+    create: (filterCriteria: AmpFindingsFilterCriteria, queryParamsFilter?: FilterCriteria) => Promise<{
+        data: never[];
+        kind: string;
+        error: string;
+        hints: {};
+    }>;
+}

package/build/src/services/findingsInsights.service.js

@@ -0,0 +1,57 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FindingsInsightsService = exports.AmpFindingsFilterCriteriaSchema = void 0;
+const zod_1 = require("zod");
+exports.AmpFindingsFilterCriteriaSchema = zod_1.z.lazy(() => zod_1.z.union([
+    zod_1.z.record(zod_1.z.string()).refine(obj => Object.keys(obj).every(key => typeof key === 'string')),
+    zod_1.z.object({
+        $and: zod_1.z.array(exports.AmpFindingsFilterCriteriaSchema),
+    }),
+    zod_1.z.object({
+        $or: zod_1.z.array(exports.AmpFindingsFilterCriteriaSchema),
+    }),
+]));
+class FindingsInsightsService {
+    constructor(rest, kind, targetApi = 'api') {
+        this.call = async (req, errorHandler) => {
+            try {
+                const res = await this.rest.call(req);
+                return res.data;
+            }
+            catch (error) {
+                if (error instanceof Error) {
+                    console.error(error.message);
+                }
+                if (this.rest.strict) {
+                    throw error;
+                }
+                return errorHandler(error);
+            }
+        };
+        this.getPage = async (req) => {
+            return this.call(req, (error) => {
+                const message = error instanceof Error ? error.message : 'Unknown error';
+                return {
+                    data: [],
+                    kind: this.kind.toUpperCase(),
+                    error: message,
+                    hints: {},
+                };
+            });
+        };
+        this.create = (filterCriteria, queryParamsFilter) => {
+            const filterOptions = exports.AmpFindingsFilterCriteriaSchema.parse(filterCriteria);
+            return this.getPage({
+                url: `/${this.targetApi}/v1/insights/findings`,
+                method: 'POST',
+                data: filterOptions,
+                params: queryParamsFilter,
+            });
+        };
+        this.rest = rest;
+        this.kind = kind;
+        this.targetApi = targetApi;
+    }
+}
+exports.FindingsInsightsService = FindingsInsightsService;
+//# sourceMappingURL=findingsInsights.service.js.map

package/build/src/services/findingsInsights.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"findingsInsights.service.js","sourceRoot":"","sources":["../../../src/services/findingsInsights.service.ts"],"names":[],"mappings":";;;AAEA,6BAAsB;AAGT,QAAA,+BAA+B,GAAQ,OAAC,CAAC,IAAI,CAAC,GAAG,EAAE,CAC9D,OAAC,CAAC,KAAK,CAAC;IACN,OAAC,CAAC,MAAM,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC,OAAO,GAAG,KAAK,QAAQ,CAAC,CAAC;IAC1F,OAAC,CAAC,MAAM,CAAC;QACP,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,uCAA+B,CAAC;KAC/C,CAAC;IACF,OAAC,CAAC,MAAM,CAAC;QACP,GAAG,EAAE,OAAC,CAAC,KAAK,CAAC,uCAA+B,CAAC;KAC9C,CAAC;CACH,CAAC,CACH,CAAC;AAGF,MAAa,uBAAuB;IAgClC,YAAY,IAAgB,EAAE,IAAY,EAAE,YAAuB,KAAK;QA/B9D,SAAI,GAAG,KAAK,EAAK,GAAgB,EAAE,YAA6B,EAAc,EAAE;YACxF,IAAI;gBACF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACtC,OAAO,GAAG,CAAC,IAAS,CAAC;aACtB;YAAC,OAAO,KAAc,EAAE;gBACvB,IAAI,KAAK,YAAY,KAAK,EAAE;oBAC1B,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;iBAC9B;gBACD,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;oBACpB,MAAM,KAAK,CAAC;iBACb;gBACD,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC;aAC5B;QACH,CAAC,CAAC;QAEQ,YAAO,GAAG,KAAK,EAAE,GAAgB,EAAE,EAAE;YAC7C,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,KAAc,EAAE,EAAE;gBACvC,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;gBACzE,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;oBAC7B,KAAK,EAAE,OAAO;oBACd,KAAK,EAAE,EAAE;iBACV,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC;QAYF,WAAM,GAAG,CAAC,cAAyC,EAAE,iBAAkC,EAAE,EAAE;YACzF,MAAM,aAAa,GAAG,uCAA+B,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;YAC5E,OAAO,IAAI,CAAC,OAAO,CAAC;gBAClB,GAAG,EAAE,IAAI,IAAI,CAAC,SAAS,uBAAuB;gBAC9C,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,aAAa;gBACnB,MAAM,EAAE,iBAAiB;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC;QAbA,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;CAWF;AA/CD,0DA+CC"}

package/build/src/services/index.d.ts

@@ -4,6 +4,7 @@ export * from './AmpSdk';
 export * from './constants';
 export * from './data.service';
 export * from './entity.service';
+export * from './findingsInsights.service';
 export * from './findings.service';
 export * from './lookup.service';
 export * from './rest';
@@ -14,3 +15,4 @@ export * from './connectors.service';
 export * from './notification.service';
 export * from './reports.service';
 export * from './contentful.service';
+export * from './prediction.service';

package/build/src/services/index.js

@@ -20,6 +20,7 @@ __exportStar(require("./AmpSdk"), exports);
 __exportStar(require("./constants"), exports);
 __exportStar(require("./data.service"), exports);
 __exportStar(require("./entity.service"), exports);
+__exportStar(require("./findingsInsights.service"), exports);
 __exportStar(require("./findings.service"), exports);
 __exportStar(require("./lookup.service"), exports);
 __exportStar(require("./rest"), exports);
@@ -30,4 +31,5 @@ __exportStar(require("./connectors.service"), exports);
 __exportStar(require("./notification.service"), exports);
 __exportStar(require("./reports.service"), exports);
 __exportStar(require("./contentful.service"), exports);
+__exportStar(require("./prediction.service"), exports);
 //# sourceMappingURL=index.js.map

package/build/src/services/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/services/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAyB;AACzB,8CAA4B;AAC5B,2CAAyB;AACzB,8CAA4B;AAC5B,iDAA+B;AAC/B,mDAAiC;AACjC,qDAAmC;AACnC,mDAAiC;AACjC,yCAAuB;AACvB,uDAAqC;AACrC,qDAAmC;AACnC,0CAAwB;AACxB,uDAAqC;AACrC,yDAAuC;AACvC,oDAAkC;AAClC,uDAAqC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/services/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAyB;AACzB,8CAA4B;AAC5B,2CAAyB;AACzB,8CAA4B;AAC5B,iDAA+B;AAC/B,mDAAiC;AACjC,6DAA2C;AAC3C,qDAAmC;AACnC,mDAAiC;AACjC,yCAAuB;AACvB,uDAAqC;AACrC,qDAAmC;AACnC,0CAAwB;AACxB,uDAAqC;AACrC,yDAAuC;AACvC,oDAAkC;AAClC,uDAAqC;AACrC,uDAAqC"}

package/build/src/services/prediction.service.d.ts

@@ -0,0 +1,35 @@
+import { RestClient, RestRequest } from './rest';
+import { ErrorHandler, TargetApi } from './constants';
+import { AIServiceModelName } from '../dto/enums/aiServiceModelName';
+import { z } from 'zod';
+declare const PREDICTION_SCHEMA: z.ZodObject<{
+    question: z.ZodArray<z.ZodString, "many">;
+    model: z.ZodOptional<z.ZodNativeEnum<typeof AIServiceModelName>>;
+}, "strip", z.ZodTypeAny, {
+    question: string[];
+    model?: AIServiceModelName | undefined;
+}, {
+    question: string[];
+    model?: AIServiceModelName | undefined;
+}>;
+export type PREDICTION = z.infer<typeof PREDICTION_SCHEMA>;
+export declare class PredictionService {
+    protected call: <T>(req: RestRequest, errorHandler: ErrorHandler<T>) => Promise<T>;
+    protected getPage: (req: RestRequest) => Promise<{
+        data: never[];
+        kind: string;
+        error: string;
+        hints: {};
+    }>;
+    protected readonly rest: RestClient;
+    protected readonly kind: string;
+    protected readonly targetApi: string;
+    constructor(rest: RestClient, kind: string, targetApi?: TargetApi);
+    create: ({ model, question }: PREDICTION) => Promise<{
+        data: never[];
+        kind: string;
+        error: string;
+        hints: {};
+    }>;
+}
+export {};

package/build/src/services/prediction.service.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.PredictionService = void 0;
+const aiServiceModelName_1 = require("../dto/enums/aiServiceModelName");
+const zod_1 = require("zod");
+const PREDICTION_SCHEMA = zod_1.z.object({
+    question: zod_1.z.array(zod_1.z.string()),
+    model: zod_1.z.nativeEnum(aiServiceModelName_1.AIServiceModelName).optional(),
+});
+class PredictionService {
+    constructor(rest, kind, targetApi = 'api') {
+        this.call = async (req, errorHandler) => {
+            try {
+                const res = await this.rest.call(req);
+                return res.data;
+            }
+            catch (error) {
+                if (error instanceof Error) {
+                    console.error(error.message);
+                }
+                if (this.rest.strict) {
+                    throw error;
+                }
+                return errorHandler(error);
+            }
+        };
+        this.getPage = async (req) => {
+            return this.call(req, (error) => {
+                const message = error instanceof Error ? error.message : 'Unknown error';
+                return {
+                    data: [],
+                    kind: this.kind.toUpperCase(),
+                    error: message,
+                    hints: {},
+                };
+            });
+        };
+        this.create = ({ model, question }) => {
+            return this.getPage({
+                url: `/${this.targetApi}/v1/ask/${this.kind}`,
+                method: 'POST',
+                data: { model, question },
+            });
+        };
+        this.rest = rest;
+        this.kind = kind;
+        this.targetApi = targetApi;
+    }
+}
+exports.PredictionService = PredictionService;
+//# sourceMappingURL=prediction.service.js.map

package/build/src/services/prediction.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"prediction.service.js","sourceRoot":"","sources":["../../../src/services/prediction.service.ts"],"names":[],"mappings":";;;AAEA,wEAAmE;AACnE,6BAAsB;AAEtB,MAAM,iBAAiB,GAAG,OAAC,CAAC,MAAM,CAAC;IACjC,QAAQ,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC;IAC7B,KAAK,EAAE,OAAC,CAAC,UAAU,CAAC,uCAAkB,CAAC,CAAC,QAAQ,EAAE;CACnD,CAAC,CAAC;AAIH,MAAa,iBAAiB;IAgC5B,YAAY,IAAgB,EAAE,IAAY,EAAE,YAAuB,KAAK;QA/B9D,SAAI,GAAG,KAAK,EAAK,GAAgB,EAAE,YAA6B,EAAc,EAAE;YACxF,IAAI;gBACF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACtC,OAAO,GAAG,CAAC,IAAS,CAAC;aACtB;YAAC,OAAO,KAAc,EAAE;gBACvB,IAAI,KAAK,YAAY,KAAK,EAAE;oBAC1B,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;iBAC9B;gBACD,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE;oBACpB,MAAM,KAAK,CAAC;iBACb;gBACD,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC;aAC5B;QACH,CAAC,CAAC;QAEQ,YAAO,GAAG,KAAK,EAAE,GAAgB,EAAE,EAAE;YAC7C,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,KAAc,EAAE,EAAE;gBACvC,MAAM,OAAO,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe,CAAC;gBACzE,OAAO;oBACL,IAAI,EAAE,EAAE;oBACR,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE;oBAC7B,KAAK,EAAE,OAAO;oBACd,KAAK,EAAE,EAAE;iBACV,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC,CAAC;QAYF,WAAM,GAAG,CAAC,EAAC,KAAK,EAAE,QAAQ,EAAa,EAAE,EAAE;YACzC,OAAO,IAAI,CAAC,OAAO,CAAC;gBAClB,GAAG,EAAE,IAAI,IAAI,CAAC,SAAS,WAAW,IAAI,CAAC,IAAI,EAAE;gBAC7C,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,EAAC,KAAK,EAAE,QAAQ,EAAC;aACxB,CAAC,CAAC;QACL,CAAC,CAAC;QAXA,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;CASF;AA7CD,8CA6CC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ampsec/platform-client",
-  "version": "57.0.0",
+  "version": "57.1.0",
   "description": "",
   "main": "build/src/index.js",
   "runkitExampleFilename": "example/main.js",
@@ -28,7 +28,8 @@
     "immutable": "^5.0.0-beta.4",
     "lodash": "^4.17.21",
     "lodash.omit": "^4.5.0",
-    "qs": "^6.11.2"
+    "qs": "^6.11.2",
+    "zod": "^3.22.4"
   },
   "devDependencies": {
     "@jest/types": "^29.5.0",

package/src/dto/enums/aiServiceModelName.ts

@@ -0,0 +1,7 @@
+/* eslint-disable no-unused-vars */
+export enum AIServiceModelName {
+  GPT3 = 'gpt-3.5-turbo-16k',
+  GPT4 = 'gpt-4',
+  TEXT_BISON = 'text-bison-32k',
+  CODE_BISON = 'code-bison-32k',
+}

package/src/dto/enums/index.ts

@@ -23,3 +23,4 @@ export * from './saasComponentKind';
 export * from './supportedMessageChannelKind.enum';
 export * from './workflowStep.kind';
 export * from './workflowTrigger.kind';
+export * from './aiServiceModelName';

package/src/services/AmpApi.ts

@@ -1,17 +1,4 @@
-import {
-  AgentDto,
-  AgentUpsertDto,
-  AssetDto,
-  FindingsInsightsDto,
-  ProviderDto,
-  ReportResultDto,
-  SaasAssetDto,
-  SaasComponentDto,
-  SaasUserDto,
-  TenantDto,
-  TenantUpsertDto,
-  UserDto,
-} from '../dto';
+import {AgentDto, AgentUpsertDto, AssetDto, ProviderDto, ReportResultDto, SaasAssetDto, SaasComponentDto, SaasUserDto, TenantDto, TenantUpsertDto, UserDto} from '../dto';
 import {
   AmpEntityService,
   AmpEntityServiceImpl,
@@ -21,8 +8,10 @@ import {
   AmpReportServiceImpl,
   AmpSettingsService,
   ConnectorsService,
+  FindingsInsightsService,
   FindingsService,
   NotificationService,
+  PredictionService,
 } from '.';
 import {KIND, TARGET_API_AGENT} from './constants';
 import {getAmpRestClient, AmpRestClientOptions, AgentIdentityService, ConnectorInstallService, DefaultEnumService, EnumService, RestClient} from './rest';
@@ -47,7 +36,7 @@ export class AmpApi {
   readonly connectors: ConnectorsService;
   readonly enums: EnumService;
   readonly findings: FindingsService;
-  readonly findingsInsights: AmpDataService<FindingsInsightsDto>;
+  readonly findingsInsights: FindingsInsightsService;
   readonly identity: AgentIdentityService;
   readonly install: ConnectorInstallService;
   readonly notifications: NotificationService;
@@ -60,6 +49,7 @@ export class AmpApi {
   readonly settings: AmpSettingsService;
   readonly tenants: AmpEntityService<TenantUpsertDto, TenantDto>;
   readonly users: AmpDataService<UserDto>;
+  readonly prediction: PredictionService;
 
   constructor(rest: RestClient) {
     this.agents = new AmpEntityServiceImpl<AgentUpsertDto, AgentDto>(rest, KIND.AGENTS);
@@ -67,7 +57,7 @@ export class AmpApi {
     this.connectors = new ConnectorsService(rest);
     this.enums = new DefaultEnumService(rest);
     this.findings = new FindingsService(rest);
-    this.findingsInsights = new AmpDataServiceImpl<FindingsInsightsDto>(rest, KIND.FINDINGS_INSIGHTS);
+    this.findingsInsights = new FindingsInsightsService(rest, KIND.FINDINGS_INSIGHTS);
     this.identity = new AgentIdentityService(rest);
     this.install = new ConnectorInstallService(rest, TARGET_API_AGENT);
     this.notifications = new NotificationService(rest);
@@ -80,6 +70,7 @@ export class AmpApi {
     this.settings = new AmpSettingsService(rest);
     this.tenants = new AmpEntityServiceImpl<TenantUpsertDto, TenantDto>(rest, KIND.TENANTS);
     this.users = new AmpDataServiceImpl<UserDto>(rest, KIND.USERS);
+    this.prediction = new PredictionService(rest, KIND.FINDINGS);
   }
 
   static instance(options: AmpApiOptions): AmpApi {

package/src/services/AmpSdk.ts

@@ -27,7 +27,7 @@ import {DefaultEnumService, EnumService} from './rest/EnumService';
 import {AmpSdkSettingsService} from './settings.service';
 import {AmpSaaSEntityService, AmpSaaSEntityServiceImpl, AmpSdkSaasAssetService, AmpSdkSaasComponentService, AmpSdkSaasUserService} from './saasEntity.service';
 import {ContentfulService} from './contentful.service';
-import {PlatformFindingsInsightsDto, PlatformFindingsInsightsUpsertDto} from '../dto/platform/platform.findingsInsights.dto';
+import {FindingsInsightsService} from './findingsInsights.service';
 
 export type AmpSdkOptions = AmpRestClientOptions;
 
@@ -47,7 +47,7 @@ export class AmpSdkServices {
   readonly contentful: ContentfulService;
   readonly enums: EnumService;
   readonly findings: AmpSaaSEntityService<PlatformFindingUpsertDto, PlatformFindingDto>;
-  readonly findingsInsights: AmpSaaSEntityService<PlatformFindingsInsightsUpsertDto, PlatformFindingsInsightsDto>;
+  readonly findingsInsights: FindingsInsightsService;
   readonly jobExecutions: AmpEntityService<PlatformJobExecutionStateUpsertDto, PlatformJobExecutionStateDto>;
   readonly jobs: AmpEntityService<PlatformJobSpecUpsertDto, PlatformJobSpecDto>;
   readonly notifications: AmpEntityService<PlatformNotificationUpsertDto, PlatformNotificationDto>;
@@ -69,7 +69,7 @@ export class AmpSdkServices {
     this.contentful = new ContentfulService();
     this.enums = new DefaultEnumService(rest, TARGET_API_PLATFORM);
     this.findings = new AmpSaaSEntityServiceImpl<PlatformFindingUpsertDto, PlatformFindingDto>(rest, KIND.FINDINGS, TARGET_API_PLATFORM);
-    this.findingsInsights = new AmpSaaSEntityServiceImpl<PlatformFindingsInsightsUpsertDto, PlatformFindingsInsightsDto>(rest, KIND.FINDINGS_INSIGHTS, TARGET_API_PLATFORM);
+    this.findingsInsights = new FindingsInsightsService(rest, KIND.FINDINGS_INSIGHTS, TARGET_API_PLATFORM);
     this.jobExecutions = new AmpEntityServiceImpl<PlatformJobExecutionStateUpsertDto, PlatformJobExecutionStateDto>(rest, KIND.JOB_EXECUTIONS, TARGET_API_PLATFORM);
     this.jobs = new AmpEntityServiceImpl<PlatformJobSpecUpsertDto, PlatformJobSpecDto>(rest, KIND.JOB_SPEC, TARGET_API_PLATFORM);
     this.notifications = new AmpEntityServiceImpl<PlatformNotificationUpsertDto, PlatformNotificationDto>(rest, KIND.NOTIFICATIONS, TARGET_API_PLATFORM);

package/src/services/findingsInsights.service.ts

@@ -0,0 +1,66 @@
+import {RestClient, RestRequest} from './rest';
+import {ErrorHandler, TargetApi} from './constants';
+import {z} from 'zod';
+import {FilterCriteria} from '../FilterCriteria';
+
+export const AmpFindingsFilterCriteriaSchema: any = z.lazy(() =>
+  z.union([
+    z.record(z.string()).refine(obj => Object.keys(obj).every(key => typeof key === 'string')),
+    z.object({
+      $and: z.array(AmpFindingsFilterCriteriaSchema),
+    }),
+    z.object({
+      $or: z.array(AmpFindingsFilterCriteriaSchema),
+    }),
+  ])
+);
+
+export type AmpFindingsFilterCriteria = z.infer<typeof AmpFindingsFilterCriteriaSchema>;
+export class FindingsInsightsService {
+  protected call = async <T>(req: RestRequest, errorHandler: ErrorHandler<T>): Promise<T> => {
+    try {
+      const res = await this.rest.call(req);
+      return res.data as T;
+    } catch (error: unknown) {
+      if (error instanceof Error) {
+        console.error(error.message);
+      }
+      if (this.rest.strict) {
+        throw error;
+      }
+      return errorHandler(error);
+    }
+  };
+
+  protected getPage = async (req: RestRequest) => {
+    return this.call(req, (error: unknown) => {
+      const message = error instanceof Error ? error.message : 'Unknown error';
+      return {
+        data: [],
+        kind: this.kind.toUpperCase(),
+        error: message,
+        hints: {},
+      };
+    });
+  };
+
+  protected readonly rest: RestClient;
+  protected readonly kind: string;
+  protected readonly targetApi: string;
+
+  constructor(rest: RestClient, kind: string, targetApi: TargetApi = 'api') {
+    this.rest = rest;
+    this.kind = kind;
+    this.targetApi = targetApi;
+  }
+
+  create = (filterCriteria: AmpFindingsFilterCriteria, queryParamsFilter?: FilterCriteria) => {
+    const filterOptions = AmpFindingsFilterCriteriaSchema.parse(filterCriteria);
+    return this.getPage({
+      url: `/${this.targetApi}/v1/insights/findings`,
+      method: 'POST',
+      data: filterOptions,
+      params: queryParamsFilter,
+    });
+  };
+}

package/src/services/index.ts

@@ -4,6 +4,7 @@ export * from './AmpSdk';
 export * from './constants';
 export * from './data.service';
 export * from './entity.service';
+export * from './findingsInsights.service';
 export * from './findings.service';
 export * from './lookup.service';
 export * from './rest';
@@ -14,3 +15,4 @@ export * from './connectors.service';
 export * from './notification.service';
 export * from './reports.service';
 export * from './contentful.service';
+export * from './prediction.service';

package/src/services/prediction.service.ts

@@ -0,0 +1,58 @@
+import {RestClient, RestRequest} from './rest';
+import {ErrorHandler, TargetApi} from './constants';
+import {AIServiceModelName} from '../dto/enums/aiServiceModelName';
+import {z} from 'zod';
+
+const PREDICTION_SCHEMA = z.object({
+  question: z.array(z.string()),
+  model: z.nativeEnum(AIServiceModelName).optional(),
+});
+
+export type PREDICTION = z.infer<typeof PREDICTION_SCHEMA>;
+
+export class PredictionService {
+  protected call = async <T>(req: RestRequest, errorHandler: ErrorHandler<T>): Promise<T> => {
+    try {
+      const res = await this.rest.call(req);
+      return res.data as T;
+    } catch (error: unknown) {
+      if (error instanceof Error) {
+        console.error(error.message);
+      }
+      if (this.rest.strict) {
+        throw error;
+      }
+      return errorHandler(error);
+    }
+  };
+
+  protected getPage = async (req: RestRequest) => {
+    return this.call(req, (error: unknown) => {
+      const message = error instanceof Error ? error.message : 'Unknown error';
+      return {
+        data: [],
+        kind: this.kind.toUpperCase(),
+        error: message,
+        hints: {},
+      };
+    });
+  };
+
+  protected readonly rest: RestClient;
+  protected readonly kind: string;
+  protected readonly targetApi: string;
+
+  constructor(rest: RestClient, kind: string, targetApi: TargetApi = 'api') {
+    this.rest = rest;
+    this.kind = kind;
+    this.targetApi = targetApi;
+  }
+
+  create = ({model, question}: PREDICTION) => {
+    return this.getPage({
+      url: `/${this.targetApi}/v1/ask/${this.kind}`,
+      method: 'POST',
+      data: {model, question},
+    });
+  };
+}