more

stephmilovic · Sep 26, 2023 · 71d2b0c · 71d2b0c
1 parent bc6c2a2
commit 71d2b0c
Show file tree

Hide file tree

Showing 5 changed files with 18 additions and 18 deletions.
diff --git a/x-pack/plugins/stack_connectors/server/connector_types/open_ai/index.ts b/x-pack/plugins/stack_connectors/server/connector_types/open_ai/index.ts
@@ -21,13 +21,13 @@ import {
 } from '../../../common/open_ai/constants';
 import { ConfigSchema, SecretsSchema } from '../../../common/open_ai/schema';
 import { Config, Secrets } from '../../../common/open_ai/types';
-import { GenAiConnector } from './open_ai';
+import { OpenAIConnector } from './open_ai';
 import { renderParameterTemplates } from './render';
 
 export const getConnectorType = (): SubActionConnectorType<Config, Secrets> => ({
   id: OPEN_AI_CONNECTOR_ID,
   name: OPEN_AI_TITLE,
-  Service: GenAiConnector,
+  Service: OpenAIConnector,
   schema: {
     config: ConfigSchema,
     secrets: SecretsSchema,

diff --git a/x-pack/plugins/stack_connectors/server/connector_types/open_ai/open_ai.test.ts b/x-pack/plugins/stack_connectors/server/connector_types/open_ai/open_ai.test.ts
@@ -5,7 +5,7 @@
  * 2.0.
  */
 
-import { GenAiConnector } from './open_ai';
+import { OpenAIConnector } from './open_ai';
 import { actionsConfigMock } from '@kbn/actions-plugin/server/actions_config.mock';
 import {
   DEFAULT_OPENAI_MODEL,
@@ -18,7 +18,7 @@ import { RunActionResponseSchema, StreamingResponseSchema } from '../../../commo
 import { initDashboard } from './create_dashboard';
 jest.mock('./create_dashboard');
 
-describe('GenAiConnector', () => {
+describe('OpenAIConnector', () => {
   let mockRequest: jest.Mock;
   let mockError: jest.Mock;
   const mockResponseString = 'Hello! How can I assist you today?';
@@ -46,7 +46,7 @@ describe('GenAiConnector', () => {
   });
 
   describe('OpenAI', () => {
-    const connector = new GenAiConnector({
+    const connector = new OpenAIConnector({
       configurationUtilities: actionsConfigMock.create(),
       connector: { id: '1', type: OPEN_AI_CONNECTOR_ID },
       config: {
@@ -285,7 +285,7 @@ describe('GenAiConnector', () => {
   });
 
   describe('AzureAI', () => {
-    const connector = new GenAiConnector({
+    const connector = new OpenAIConnector({
       configurationUtilities: actionsConfigMock.create(),
       connector: { id: '1', type: OPEN_AI_CONNECTOR_ID },
       config: {
@@ -455,7 +455,7 @@ describe('GenAiConnector', () => {
   });
 
   describe('Token dashboard', () => {
-    const connector = new GenAiConnector({
+    const connector = new OpenAIConnector({
       configurationUtilities: actionsConfigMock.create(),
       connector: { id: '1', type: OPEN_AI_CONNECTOR_ID },
       config: { apiUrl: 'https://example.com/api', apiProvider: OpenAiProviderType.AzureAi },

diff --git a/x-pack/plugins/stack_connectors/server/connector_types/open_ai/open_ai.ts b/x-pack/plugins/stack_connectors/server/connector_types/open_ai/open_ai.ts
@@ -37,7 +37,7 @@ import {
   sanitizeRequest,
 } from './lib/utils';
 
-export class GenAiConnector extends SubActionConnector<Config, Secrets> {
+export class OpenAIConnector extends SubActionConnector<Config, Secrets> {
   private url;
   private provider;
   private key;

diff --git a/...ns_simulators/server/gen_ai_simulation.ts → ...s_simulators/server/open_ai_simulation.ts b/...ns_simulators/server/gen_ai_simulation.ts → ...s_simulators/server/open_ai_simulation.ts
@@ -9,7 +9,7 @@ import http from 'http';
 
 import { ProxyArgs, Simulator } from './simulator';
 
-export class GenAiSimulator extends Simulator {
+export class OpenAISimulator extends Simulator {
   private readonly returnError: boolean;
 
   constructor({ returnError = false, proxy }: { returnError?: boolean; proxy?: ProxyArgs }) {
@@ -24,10 +24,10 @@ export class GenAiSimulator extends Simulator {
     data: Record<string, unknown>
   ) {
     if (this.returnError) {
-      return GenAiSimulator.sendErrorResponse(response);
+      return OpenAISimulator.sendErrorResponse(response);
     }
 
-    return GenAiSimulator.sendResponse(response);
+    return OpenAISimulator.sendResponse(response);
   }
 
   private static sendResponse(response: http.ServerResponse) {

diff --git a/...2/tests/actions/connector_types/gen_ai.ts → .../tests/actions/connector_types/open_ai.ts b/...2/tests/actions/connector_types/gen_ai.ts → .../tests/actions/connector_types/open_ai.ts
@@ -8,9 +8,9 @@
 import expect from '@kbn/expect';
 
 import {
-  GenAiSimulator,
+  OpenAISimulator,
   genAiSuccessResponse,
-} from '@kbn/actions-simulators-plugin/server/gen_ai_simulation';
+} from '@kbn/actions-simulators-plugin/server/open_ai_simulation';
 import { FtrProviderContext } from '../../../../../common/ftr_provider_context';
 import { getUrlPrefix, ObjectRemover } from '../../../../../common/lib';
 
@@ -51,7 +51,7 @@ export default function genAiTest({ getService }: FtrProviderContext) {
       objectRemover.removeAll();
     });
     describe('action creation', () => {
-      const simulator = new GenAiSimulator({
+      const simulator = new OpenAISimulator({
         returnError: false,
         proxy: {
           config: configService.get('kbnTestServer.serverArgs'),
@@ -217,7 +217,7 @@ export default function genAiTest({ getService }: FtrProviderContext) {
 
     describe('executor', () => {
       describe('validation', () => {
-        const simulator = new GenAiSimulator({
+        const simulator = new OpenAISimulator({
           proxy: {
             config: configService.get('kbnTestServer.serverArgs'),
           },
@@ -272,7 +272,7 @@ export default function genAiTest({ getService }: FtrProviderContext) {
 
       describe('execution', () => {
         describe('successful response simulator', () => {
-          const simulator = new GenAiSimulator({
+          const simulator = new OpenAISimulator({
             proxy: {
               config: configService.get('kbnTestServer.serverArgs'),
             },
@@ -377,7 +377,7 @@ export default function genAiTest({ getService }: FtrProviderContext) {
           });
         });
         describe('non-default space simulator', () => {
-          const simulator = new GenAiSimulator({
+          const simulator = new OpenAISimulator({
             proxy: {
               config: configService.get('kbnTestServer.serverArgs'),
             },
@@ -428,7 +428,7 @@ export default function genAiTest({ getService }: FtrProviderContext) {
         });
 
         describe('error response simulator', () => {
-          const simulator = new GenAiSimulator({
+          const simulator = new OpenAISimulator({
             returnError: true,
             proxy: {
               config: configService.get('kbnTestServer.serverArgs'),