refactoring_v1 (#391)

src/engine/flowise.ts

@@ -1,38 +1,40 @@
-import axios, { AxiosError } from 'axios';
-import { ChatCompletionRequestMessage } from 'openai';
-import { AiEngine } from './Engine';
+import axios, { AxiosInstance } from 'axios';
+import { OpenAI } from 'openai';
+import { AiEngine, AiEngineConfig } from './Engine';
 
-import {
-  getConfig
-} from '../commands/config';
-
-const config = getConfig();
+interface FlowiseAiConfig extends AiEngineConfig {}
 
 export class FlowiseAi implements AiEngine {
+  config: FlowiseAiConfig;
+  client: AxiosInstance;
+
+  constructor(config) {
+    this.config = config;
+    this.client = axios.create({
+      url: `${config.baseURL}/${config.apiKey}`,
+      headers: { 'Content-Type': 'application/json' }
+    });
+  }
 
   async generateCommitMessage(
-    messages: Array<ChatCompletionRequestMessage>
+    messages: Array<OpenAI.Chat.Completions.ChatCompletionMessageParam>
   ): Promise<string | undefined> {
+    const gitDiff = (messages[messages.length - 1]?.content as string)
+      .replace(/\\/g, '\\\\')
+      .replace(/"/g, '\\"')
+      .replace(/\n/g, '\\n')
+      .replace(/\r/g, '\\r')
+      .replace(/\t/g, '\\t');
 
-    const gitDiff = messages[ messages.length - 1 ]?.content?.replace(/\\/g, '\\\\')  
-                                                            .replace(/"/g, '\\"')    
-                                                            .replace(/\n/g, '\\n')   
-                                                            .replace(/\r/g, '\\r')   
-                                                            .replace(/\t/g, '\\t');  
-    const url = `http://${config?.OCO_FLOWISE_ENDPOINT}/api/v1/prediction/${config?.OCO_FLOWISE_API_KEY}`; 
     const payload = {
-        question : gitDiff,
-        overrideConfig : {
-          systemMessagePrompt: messages[0]?.content,
-        },
-        history : messages.slice( 1, -1 ) 
-    }
+      question: gitDiff,
+      overrideConfig: {
+        systemMessagePrompt: messages[0]?.content
+      },
+      history: messages.slice(1, -1)
+    };
     try {
-      const response = await axios.post(url, payload, {
-        headers: {
-          'Content-Type': 'application/json'
-        }
-      });
+      const response = await this.client.post('', payload);
       const message = response.data;
       return message?.text;
     } catch (err: any) {