doomiaichat 2.6.0 → 2.7.0

package/dist/azureai.js

@@ -50,7 +50,6 @@ class AzureAI extends openai_1.default {
                 let param = Object.assign(Object.assign({}, axiosOption), { method: "post", data: {
                         input: text
                     }, url: this.EmbeddingUrl });
-                console.log('param', param);
                 const response = yield (0, declare_1.request)(param);
                 if (response.data) {
                     return { successed: true, embedding: response.data.data[0].embedding };

package/dist/openai.d.ts

@@ -1,6 +1,6 @@
 import { OpenAIApi, ChatCompletionRequestMessage } from "openai";
 import GptBase from "./gptbase";
-import { OpenAIApiParameters, ChatReponse, SummaryReponse, FaqItem, ExaminationPaperResult, EmotionResult, SimilarityResult, QuestionItem, CommentResult } from './declare';
+import { OpenAIApiParameters, ChatReponse, SummaryReponse, FaqItem, ExaminationPaperResult, EmotionResult, SimilarityResult, QuestionItem, CommentResult, EmbeddingResult } from './declare';
 export default class OpenAIGpt extends GptBase {
     protected readonly apiKey: string;
     private aiApi;
@@ -21,7 +21,7 @@ export default class OpenAIGpt extends GptBase {
      * 获得文字的向量
      * @param text
      */
-    getTextEmbedding(_text: string, _axiosOption: any): Promise<any>;
+    getTextEmbedding(text: string, axiosOption: any): Promise<EmbeddingResult>;
     /**
      * 向OpenAI发送一个聊天请求
      * @param {*} chatText

package/dist/openai.js

@@ -48,9 +48,25 @@ class OpenAIGpt extends gptbase_1.default {
      * 获得文字的向量
      * @param text
      */
-    getTextEmbedding(_text, _axiosOption) {
+    getTextEmbedding(text, axiosOption) {
         return __awaiter(this, void 0, void 0, function* () {
-            return null;
+            if (!text)
+                return { successed: false, error: { errcode: 2, errmsg: 'content required' } };
+            if (!this.aiApi) {
+                this.aiApi = this.createOpenAI(this.apiKey);
+            }
+            // console.log('message', message)
+            try {
+                const response = yield this.aiApi.createEmbedding({
+                    model: 'text-embedding-ada-002',
+                    input: text,
+                }, axiosOption);
+                return { successed: true, embedding: response.data.data[0].embedding };
+            }
+            catch (error) {
+                console.log('result is error ', error);
+                return { successed: false, error };
+            }
         });
     }
     /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "doomiaichat",
-  "version": "2.6.0",
+  "version": "2.7.0",
   "description": "Doomisoft OpenAI",
   "main": "dist/index.js",
   "scripts": {

package/src/azureai.ts

@@ -44,7 +44,6 @@ export default class AzureAI extends OpenAIGpt {
                 },
                 url: this.EmbeddingUrl
             };
-            console.log('param', param)
             const response = await request(param)
             if (response.data) {
                 return { successed: true, embedding: response.data.data[0].embedding };

package/src/openai.ts

@@ -1,7 +1,7 @@
 import { Configuration, OpenAIApi, ChatCompletionRequestMessage } from "openai"
 // import { EventEmitter } from "events";
 import GptBase from "./gptbase"
-import { OpenAIApiParameters, ChatReponse, OutlineSummaryItem, SummaryReponse, FaqItem, ExaminationPaperResult, EmotionResult, SimilarityResult, QuestionItem, CommentResult } from './declare'
+import { OpenAIApiParameters, ChatReponse, OutlineSummaryItem, SummaryReponse, FaqItem, ExaminationPaperResult, EmotionResult, SimilarityResult, QuestionItem, CommentResult, EmbeddingResult } from './declare'
 const SECTION_LENGTH = 1600;     ///每2400个字符分成一组
 const MESSAGE_LENGTH = 1;      ///每次送8句话给openai 进行解析，送多了，会报错
 //请将答案放在最后,标记为答案：()
@@ -43,8 +43,22 @@ export default class OpenAIGpt extends GptBase {
      * 获得文字的向量
      * @param text 
      */
-    async getTextEmbedding(_text: string, _axiosOption: any): Promise<any>{
-        return null;
+    async getTextEmbedding(text: string, axiosOption: any): Promise<EmbeddingResult>{
+        if (!text) return { successed: false, error: { errcode: 2, errmsg: 'content required' } };
+        if (!this.aiApi) {
+            this.aiApi = this.createOpenAI(this.apiKey);
+        }
+        // console.log('message', message)
+        try {
+            const response:any = await this.aiApi.createEmbedding({
+                model:'text-embedding-ada-002',
+                input: text,
+            }, axiosOption);
+            return { successed: true, embedding: response.data.data[0].embedding };
+        } catch (error) {
+            console.log('result is error ', error)
+            return { successed: false, error };
+        }
     }
     /**
      * 向OpenAI发送一个聊天请求