doomiaichat 6.1.0 → 7.0.1

package/dist/aimp.d.ts

@@ -0,0 +1,34 @@
+import GptBase from "./gptbase";
+import { ChatReponse } from './declare';
+export default class AIMiddlePlatform extends GptBase {
+    protected apikey: string;
+    protected agent: {
+        endpoint: string;
+        agentid: string;
+    };
+    /**
+     *
+     * @param apikey 调用AI中台 的key
+     * @param agent  智能体信息
+     */
+    constructor(apikey: string, agent: {
+        endpoint: string;
+        agentid: string;
+    });
+    /**
+      * 非流式传输聊天请求
+      * @param chatText
+      * @param callChatOption
+      * @param axiosOption
+      */
+    chatRequest(chatText: string, callChatOption: any, axiosOption?: any): Promise<ChatReponse>;
+    /**
+     * 流式传输聊天请求
+     * @param chatText
+     * @param callChatOption
+     * @param attach
+     * @param axiosOption
+     * @returns
+     */
+    chatRequestInStream(chatText: string | Array<any>, callChatOption: any, attach?: any, axiosOption?: any): Promise<any>;
+}

package/dist/aimp.js

@@ -0,0 +1,133 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+/**
+ * 接入AI平台的中间层
+ */
+const axios_1 = __importDefault(require("axios"));
+const declare_1 = require("./declare");
+const gptbase_1 = __importDefault(require("./gptbase"));
+const stream_1 = require("stream");
+class AIMiddlePlatform extends gptbase_1.default {
+    /**
+     *
+     * @param apikey 调用AI中台 的key
+     * @param agent  智能体信息
+     */
+    constructor(apikey, agent) {
+        super();
+        this.apikey = apikey;
+        this.agent = agent;
+    }
+    /**
+      * 非流式传输聊天请求
+      * @param chatText
+      * @param callChatOption
+      * @param axiosOption
+      */
+    chatRequest(chatText, callChatOption, axiosOption = {}) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!chatText)
+                this.emit('chaterror', { successed: false, error: 'no text in chat' });
+            axiosOption = Object.assign({}, axiosOption || { timeout: 60000 });
+            const opts = Object.assign({ headers: {
+                    'Content-Type': 'application/json',
+                    'authorization': `Bearer ${this.apikey}`
+                }, method: 'post', url: `${this.agent.endpoint}/api/v1/agents/${this.agent.agentid}/completions`, data: {
+                    question: chatText,
+                    session_id: callChatOption.session_id,
+                    stream: false
+                } }, axiosOption);
+            const response = yield (0, declare_1.request)(opts);
+            if (!response.successed)
+                return { successed: false, error: 'failed' };
+            return { successed: true, message: response.data.answer };
+        });
+    }
+    /**
+     * 流式传输聊天请求
+     * @param chatText
+     * @param callChatOption
+     * @param attach
+     * @param axiosOption
+     * @returns
+     */
+    chatRequestInStream(chatText, callChatOption, attach, axiosOption) {
+        var _a, e_1, _b, _c;
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!chatText)
+                this.emit('chaterror', { successed: false, error: 'no text in chat' });
+            axiosOption = Object.assign({}, axiosOption || { timeout: 60000 });
+            let requestid = Math.ceil(Math.random() * (new Date().getTime() * Math.random()) / 1000);
+            try {
+                const opts = Object.assign({ headers: {
+                        'Content-Type': 'application/json',
+                        'authorization': `Bearer ${this.apikey}`
+                    }, method: 'post', url: `${this.agent.endpoint}/api/v1/agents/${this.agent.agentid}/completions`, data: {
+                        question: chatText,
+                        session_id: callChatOption.session_id,
+                        stream: true,
+                        optional: callChatOption.optional,
+                    }, responseType: 'stream' }, axiosOption);
+                // console.log('opts', opts)
+                const response = yield (0, axios_1.default)(opts);
+                const readableStream = stream_1.Readable.from(response.data);
+                let index = 0, session_id;
+                try {
+                    for (var _d = true, readableStream_1 = __asyncValues(readableStream), readableStream_1_1; readableStream_1_1 = yield readableStream_1.next(), _a = readableStream_1_1.done, !_a;) {
+                        _c = readableStream_1_1.value;
+                        _d = false;
+                        try {
+                            const chunk = _c;
+                            const answerData = JSON.parse(chunk.toString().replace('data:', ''));
+                            const { answer, running_status } = answerData.data;
+                            if (running_status === true)
+                                continue;
+                            if (!session_id)
+                                session_id = answerData.data.session_id;
+                            const finished = answerData.data === true;
+                            let output = { successed: true, requestid, segment: answer, text: answer, finish_reason: finished ? 'stop' : null, index: index++, session_id };
+                            if (attach)
+                                output = Object.assign({}, output, attach);
+                            this.emit(finished ? 'chatdone' : 'chattext', output);
+                        }
+                        finally {
+                            _d = true;
+                        }
+                    }
+                }
+                catch (e_1_1) { e_1 = { error: e_1_1 }; }
+                finally {
+                    try {
+                        if (!_d && !_a && (_b = readableStream_1.return)) yield _b.call(readableStream_1);
+                    }
+                    finally { if (e_1) throw e_1.error; }
+                }
+                return { successed: true, requestid };
+            }
+            catch (error) {
+                this.emit('requesterror', { successed: false, requestid, error: 'call axios faied ' + error });
+                return { successed: false, requestid };
+            }
+        });
+    }
+}
+exports.default = AIMiddlePlatform;

package/dist/declare.d.ts

@@ -73,6 +73,12 @@ export interface RpcResult extends ApiResult {
  * @returns
  */
 export declare function request(opts?: any): Promise<RpcResult>;
+/**
+ *
+ * @param opts
+ * @returns
+ */
+export declare function requestStream(opts: any, processChunkData: Function): void;
 /**
  * 数据缓存提供者接口
  */

package/dist/declare.js

@@ -12,7 +12,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.request = void 0;
+exports.requestStream = exports.request = void 0;
 const axios_1 = __importDefault(require("axios"));
 /**
  * Axios远程请求封装
@@ -33,3 +33,19 @@ function request(opts = {}) {
     });
 }
 exports.request = request;
+/**
+ *
+ * @param opts
+ * @returns
+ */
+function requestStream(opts = {}, processChunkData) {
+    if (!opts.data)
+        opts.data = opts.body;
+    (0, axios_1.default)(opts)
+        .then((res) => {
+        res.data.on('data', (chunk) => {
+            processChunkData(chunk);
+        });
+    });
+}
+exports.requestStream = requestStream;

package/dist/doubaoai.d.ts

@@ -0,0 +1,34 @@
+import GptBase from "./gptbase";
+export default class DouBaoAI extends GptBase {
+    protected apiKey: string;
+    protected apiOption: any;
+    /**
+     * 构造函数
+     */
+    constructor(apiKey: string, apiOption?: any);
+    /**
+     * 请求接口
+     */
+    chatRequest(chatText: string | Array<any>, callChatOption: any, axiosOption?: any): Promise<any>;
+    /**
+     * 组装最后的调用参数
+     * @param callChatOption
+     * @returns
+     */
+    private assembleApiParams;
+    /**
+     * 流式的聊天模式
+     * @param chatText
+     * @param _paramOption
+     * @param axiosOption
+     */
+    chatRequestInStream(chatText: string | Array<any>, callChatOption: any, attach?: any, axiosOption?: any): Promise<any>;
+    /**
+     * 处理每次流式返回的对话片段
+     * @param chunks
+     * @param requestid
+     * @param replytext
+     * @param attach
+     */
+    processChunkData(chunks: string[], requestid: Number, replytext: string[], attach: any): void;
+}

package/dist/doubaoai.js

@@ -0,0 +1,149 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+/**
+ * 火山方舟-豆包大模型引擎
+ */
+const declare_1 = require("./declare");
+const gptbase_1 = __importDefault(require("./gptbase"));
+class DouBaoAI extends gptbase_1.default {
+    /**
+     * 构造函数
+     */
+    constructor(apiKey, apiOption = {}) {
+        super();
+        this.apiOption = {};
+        this.apiKey = apiKey;
+        this.apiOption = apiOption;
+    }
+    /**
+     * 请求接口
+     */
+    chatRequest(chatText, callChatOption, axiosOption = {}) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!chatText)
+                return { successed: false, error: { errcode: 2, errmsg: '缺失聊天的内容' } };
+            const callParams = this.assembleApiParams(chatText, false, callChatOption, axiosOption);
+            try {
+                const response = yield (0, declare_1.request)(callParams);
+                if (response.successed && !response.data.code)
+                    return { successed: true, message: response.data.choices, usage: response.data.usage };
+                return Object.assign({ successed: false }, response.data);
+            }
+            catch (error) {
+                console.log('result is error ', error);
+                return { successed: false, error };
+            }
+        });
+    }
+    /**
+     * 组装最后的调用参数
+     * @param callChatOption
+     * @returns
+     */
+    assembleApiParams(chatText, streamCall = false, callChatOption, axiosOption = {}) {
+        let messages = typeof (chatText) == 'string' ? [{ role: 'user', content: chatText }] : chatText;
+        let params = {};
+        if ((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.temperature) || this.apiOption.temperature)
+            params.temperature = Number((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.temperature) || this.apiOption.temperature);
+        params.max_tokens = Number((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.maxtoken) || this.apiOption.maxtoken);
+        if ((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.top_p) || this.apiOption.top_p)
+            params.top_p = Number((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.top_p) || this.apiOption.top_p);
+        if ((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.presence_penalty) || this.apiOption.presence_penalty)
+            params.presence_penalty = Number((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.presence_penalty) || this.apiOption.presence_penalty);
+        if ((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.frequency_penalty) || this.apiOption.frequency_penalty)
+            params.frequency_penalty = Number((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.frequency_penalty) || this.apiOption.frequency_penalty);
+        if ((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.top_logprobs) || this.apiOption.top_logprobs) {
+            params.logprobs = true;
+            params.top_logprobs = Number((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.top_logprobs) || this.apiOption.top_logprobs);
+        }
+        params.tools = ((callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.enableToolCall) === 1 && (callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.tools)) ? callChatOption.tools : undefined;
+        params.tool_choice = (callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.enableToolCall) === 1 ? 'auto' : undefined;
+        const axiosParams = Object.assign(Object.assign({}, axiosOption), { method: "post", headers: {
+                'Content-Type': 'application/json',
+                'authorization': `Bearer ${this.apiKey}`
+            }, data: Object.assign(Object.assign({ model: (callChatOption === null || callChatOption === void 0 ? void 0 : callChatOption.model) || this.apiOption.model }, params), { messages, stream: streamCall }), url: 'https://ark.cn-beijing.volces.com/api/v3/chat/completions' });
+        if (streamCall)
+            axiosParams.responseType = 'stream';
+        return axiosParams;
+    }
+    /**
+     * 流式的聊天模式
+     * @param chatText
+     * @param _paramOption
+     * @param axiosOption
+     */
+    chatRequestInStream(chatText, callChatOption, attach, axiosOption) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!chatText)
+                this.emit('chaterror', { successed: false, error: 'no text in chat' });
+            axiosOption = Object.assign({}, axiosOption || { timeout: 10000 });
+            const callParams = this.assembleApiParams(chatText, true, callChatOption, axiosOption);
+            let requestid = Math.ceil(Math.random() * (new Date().getTime() * Math.random()) / 1000), replytext = [];
+            try {
+                (0, declare_1.requestStream)(callParams, (chunk) => {
+                    let streamText = chunk.toString().replace('[DONE]', '').replace(/[\r\n]+/gm, '');
+                    this.processChunkData(streamText.split(/data: /), requestid, replytext, attach);
+                });
+                return { successed: true, requestid };
+            }
+            catch (error) {
+                this.emit('requesterror', { successed: false, requestid, error: 'call axios faied ' + error });
+                return { successed: false, requestid };
+            }
+        });
+    }
+    /**
+     * 处理每次流式返回的对话片段
+     * @param chunks
+     * @param requestid
+     * @param replytext
+     * @param attach
+     */
+    processChunkData(chunks, requestid, replytext, attach) {
+        let has_tool_calls = 0, currentIndex, previous_index = -1, tool_calls = []; // 使用数组来存储工具调用
+        for (const splitString of chunks) {
+            if (!splitString)
+                continue;
+            const chunk = JSON.parse(splitString);
+            const [choice] = chunk.choices, { finish_reason: finishreason, index, usage } = choice, { content, tool_calls: toolCalls } = choice.delta;
+            if (toolCalls && toolCalls.length) {
+                currentIndex = toolCalls[0].index;
+                has_tool_calls = 1;
+                if (currentIndex !== previous_index) {
+                    tool_calls.push({
+                        id: toolCalls[0].id,
+                        type: 'function',
+                        function: {
+                            name: toolCalls[0].function.name,
+                            arguments: toolCalls[0].function.arguments
+                        }
+                    });
+                    // 更新previousIndex以供下次比较
+                    previous_index = currentIndex;
+                }
+                else {
+                    tool_calls[previous_index].function.arguments += toolCalls[0].function.arguments;
+                }
+            }
+            else
+                replytext.push(content);
+            let output = { successed: true, requestid, segment: content, text: replytext.join(''), finish_reason: finishreason, index, usage, has_tool_calls: has_tool_calls, tool_calls: tool_calls };
+            if (attach)
+                output = Object.assign({}, output, attach);
+            this.emit(finishreason ? 'chatdone' : 'chattext', output);
+        }
+    }
+}
+exports.default = DouBaoAI;

package/dist/gptprovider.d.ts

@@ -7,7 +7,9 @@ export declare const GptProviderEnum: {
     readonly OPENAI: "openai";
     readonly OPENAIPROXY: "openaiproxy";
     readonly MICROSOFT: "microsoft";
+    readonly AIMP: "aimp";
     readonly BAIDU: "baidu";
+    readonly DOUBAO: "doubao";
     readonly GOOGLE: "google";
     readonly STABILITY: "stability";
     readonly STABILITY2: "stability2";

package/dist/gptprovider.js

@@ -14,6 +14,8 @@ const azureai_1 = __importDefault(require("./azureai"));
 const stabilityai_1 = __importDefault(require("./stabilityai"));
 const stabilityplusai_1 = __importDefault(require("./stabilityplusai"));
 const baiduai_1 = __importDefault(require("./baiduai"));
+const aimp_1 = __importDefault(require("./aimp"));
+const doubaoai_1 = __importDefault(require("./doubaoai"));
 /**
  * OpenAI/NLP 的服务提供商 OpenAI,微软，百度文心(待接入)，google(待接入)
  */
@@ -21,7 +23,9 @@ exports.GptProviderEnum = {
     OPENAI: 'openai',
     OPENAIPROXY: 'openaiproxy',
     MICROSOFT: 'microsoft',
+    AIMP: 'aimp',
     BAIDU: 'baidu',
+    DOUBAO: 'doubao',
     GOOGLE: 'google',
     STABILITY: 'stability',
     STABILITY2: 'stability2',
@@ -34,7 +38,7 @@ exports.GptProviderEnum = {
  * @returns
  */
 function createGpt(provider, apikey, setting) {
-    let { model, maxtoken, temperature, serviceurl, endpoint, engine, version, embedding, top_p, presence_penalty, frequency_penalty } = setting || {};
+    let { model, agentid, maxtoken, temperature, serviceurl, endpoint, engine, version, embedding, top_p, presence_penalty, frequency_penalty } = setting || {};
     switch (provider) {
         case exports.GptProviderEnum.OPENAI:
             return new openai_1.default(apikey + '', { model, maxtoken, temperature, embedding, top_p, presence_penalty, frequency_penalty });
@@ -45,6 +49,10 @@ function createGpt(provider, apikey, setting) {
         case exports.GptProviderEnum.BAIDU:
             let cred = typeof (apikey) === 'string' ? { apikey, securitykey: apikey } : apikey;
             return new baiduai_1.default(cred);
+        case exports.GptProviderEnum.AIMP:
+            return new aimp_1.default(apikey + '', { endpoint, agentid });
+        case exports.GptProviderEnum.DOUBAO:
+            return new doubaoai_1.default(apikey + '', { model, maxtoken, temperature, top_p, presence_penalty, frequency_penalty });
         case exports.GptProviderEnum.STABILITY:
             return new stabilityai_1.default(apikey + '', { endpoint, engine }, setting);
         case exports.GptProviderEnum.STABILITY2:

package/package.json

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

package/src/aimp.ts

@@ -0,0 +1,100 @@
+/**
+ * 接入AI平台的中间层
+ */
+import axios from 'axios';
+import { request } from "./declare";
+import GptBase from "./gptbase"
+import { Readable } from 'stream';
+import { ChatReponse } from './declare';
+export default class AIMiddlePlatform extends GptBase {
+
+    protected apikey: string;
+    protected agent: { endpoint: string, agentid: string };
+    /**
+     * 
+     * @param apikey 调用AI中台 的key
+     * @param agent  智能体信息 
+     */
+    constructor(apikey: string, agent: { endpoint: string, agentid: string }) {
+        super();
+        this.apikey = apikey;
+        this.agent = agent;
+    }
+
+    /**
+      * 非流式传输聊天请求
+      * @param chatText 
+      * @param callChatOption 
+      * @param axiosOption 
+      */
+    public async chatRequest(chatText: string, callChatOption: any, axiosOption: any = {}): Promise<ChatReponse> {
+        if (!chatText) this.emit('chaterror', { successed: false, error: 'no text in chat' });
+        axiosOption = Object.assign({}, axiosOption || { timeout: 60000 })
+        const opts: any = {
+            headers: {
+                'Content-Type': 'application/json',
+                'authorization': `Bearer ${this.apikey}`
+            },
+            method: 'post',
+            url: `${this.agent.endpoint}/api/v1/agents/${this.agent.agentid}/completions`,
+            data: {
+                question: chatText,
+                session_id: callChatOption.session_id,
+                stream: false
+            },
+            ...axiosOption
+        }
+        const response = await request(opts);
+        if (!response.successed) return { successed: false, error: 'failed' };
+        return { successed: true, message: response.data.answer };
+    }
+    /**
+     * 流式传输聊天请求
+     * @param chatText 
+     * @param callChatOption 
+     * @param attach 
+     * @param axiosOption 
+     * @returns 
+     */
+    override async chatRequestInStream(chatText: string | Array<any>, callChatOption: any, attach?: any, axiosOption?: any): Promise<any> {
+        if (!chatText) this.emit('chaterror', { successed: false, error: 'no text in chat' });
+        axiosOption = Object.assign({}, axiosOption || { timeout: 60000 })
+        let requestid = Math.ceil(Math.random() * (new Date().getTime() * Math.random()) / 1000);
+        try {
+            const opts: any = {
+                headers: {
+                    'Content-Type': 'application/json',
+                    'authorization': `Bearer ${this.apikey}`
+                },
+                method: 'post',
+                url: `${this.agent.endpoint}/api/v1/agents/${this.agent.agentid}/completions`,
+                data: {
+                    question: chatText,
+                    session_id: callChatOption.session_id,
+                    stream: true,
+                    optional: callChatOption.optional,
+                },
+                responseType: 'stream',
+                ...axiosOption
+            }
+            // console.log('opts', opts)
+            const response = await axios(opts);
+            const readableStream = Readable.from(response.data);
+            let index = 0, session_id;
+            for await (const chunk of readableStream) {
+                const answerData = JSON.parse(chunk.toString().replace('data:',''));
+                const { answer, running_status } = answerData.data;
+                if (running_status === true) continue;
+                if (!session_id) session_id = answerData.data.session_id;
+                const finished = answerData.data === true;
+                let output = { successed: true, requestid, segment: answer, text: answer, finish_reason: finished ? 'stop' : null, index: index++, session_id };
+                if (attach) output = Object.assign({}, output, attach);
+                this.emit(finished ? 'chatdone' : 'chattext', output)
+            }
+            return { successed: true, requestid }
+        } catch (error) {
+            this.emit('requesterror', { successed: false, requestid, error: 'call axios faied ' + error });
+            return { successed: false, requestid }
+        }
+    }
+}

package/src/declare.ts

@@ -1,6 +1,7 @@
 
 import { EmbeddingItem } from '@azure/openai';
 import axios from 'axios';
+
 export interface ApiResult {
     /**
      * return the result of api called
@@ -25,29 +26,29 @@ export interface ChatReponse extends ApiResult {
      * @memberof ChatReponse
      */
     'message'?: Array<any>;
-    'usage'?:any;
+    'usage'?: any;
 }
 /**
  * 调用OpenAI Api的参数约定
  */
 export interface OpenAIApiParameters {
-    'embedding'?:string,       ///模型引擎，兼容Azure
+    'embedding'?: string,       ///模型引擎，兼容Azure
     'model'?: string,      ///模型名称
     'maxtoken'?: number;    ///返回的最大token
     'temperature'?: number;
-    'top_p'?:number;
+    'top_p'?: number;
     'presence_penalty'?: number;
     'frequency_penalty'?: number;
     'replyCounts'?: number;  ///返回多少答案
-    'tools'?:Array<any>,
+    'tools'?: Array<any>,
     'tool_choice'?: string,
     'enableToolCall'?: number ///是否允许调用toolfunction
 }
 /**
  * Azure 上的OpenAI的链接参数
  */
-export interface ProxyPatameters{
-    'serviceurl':string;  ///端点
+export interface ProxyPatameters {
+    'serviceurl': string;  ///端点
 }
 
 /**
@@ -84,9 +85,23 @@ export async function request(opts: any = {}): Promise<RpcResult> {
         let result = await axios(opts);
         return { successed: true, data: result.data }
     } catch (err) {
-        return { successed: false, error: err,data:err }
+        return { successed: false, error: err, data: err }
     }
 }
+/**
+ * 
+ * @param opts 
+ * @returns 
+ */
+export function requestStream(opts: any = {},processChunkData:Function) {
+    if (!opts.data) opts.data = opts.body;
+    axios(opts)
+    .then((res: any) => {
+        res.data.on('data', (chunk: any) => {
+            processChunkData(chunk);
+        });
+    })
+}
 
 
 /**

package/src/doubaoai.ts

@@ -0,0 +1,129 @@
+/**
+ * 火山方舟-豆包大模型引擎
+ */
+import { request, requestStream } from "./declare";
+import GptBase from "./gptbase"
+export default class DouBaoAI extends GptBase {
+    protected apiKey: string;
+    protected apiOption: any = {}
+    /**
+     * 构造函数
+     */
+    constructor(apiKey: string, apiOption: any = {}) {
+        super();
+        this.apiKey = apiKey;
+        this.apiOption = apiOption;
+    }
+    /**
+     * 请求接口
+     */
+    public async chatRequest(chatText: string | Array<any>, callChatOption: any, axiosOption: any = {}): Promise<any> {
+        if (!chatText) return { successed: false, error: { errcode: 2, errmsg: '缺失聊天的内容' } };
+        const callParams = this.assembleApiParams(chatText, false, callChatOption, axiosOption);
+        try {
+            const response = await request(callParams)
+            if (response.successed && !response.data.code)  return { successed: true, message: response.data.choices, usage: response.data.usage }
+            return { successed: false, ...response.data };
+        } catch (error) {
+            console.log('result is error ', error)
+            return { successed: false, error };
+        }
+    }
+    /**
+     * 组装最后的调用参数
+     * @param callChatOption 
+     * @returns 
+     */
+    private assembleApiParams(chatText: string | Array<any>, streamCall: boolean = false, callChatOption: any, axiosOption: any = {}): any {
+        let messages: Array<any> = typeof (chatText) == 'string' ? [{ role: 'user', content: chatText }] : chatText;
+        let params: any = {};
+        if (callChatOption?.temperature || this.apiOption.temperature) params.temperature = Number(callChatOption?.temperature || this.apiOption.temperature);
+        params.max_tokens = Number(callChatOption?.maxtoken || this.apiOption.maxtoken);
+        if (callChatOption?.top_p || this.apiOption.top_p) params.top_p = Number(callChatOption?.top_p || this.apiOption.top_p);
+        if (callChatOption?.presence_penalty || this.apiOption.presence_penalty) params.presence_penalty = Number(callChatOption?.presence_penalty || this.apiOption.presence_penalty);
+        if (callChatOption?.frequency_penalty || this.apiOption.frequency_penalty) params.frequency_penalty = Number(callChatOption?.frequency_penalty || this.apiOption.frequency_penalty);
+        if (callChatOption?.top_logprobs || this.apiOption.top_logprobs) {
+            params.logprobs = true;
+            params.top_logprobs = Number(callChatOption?.top_logprobs || this.apiOption.top_logprobs);
+        }
+        params.tools = (callChatOption?.enableToolCall === 1 && callChatOption?.tools) ? callChatOption.tools : undefined;
+        params.tool_choice = callChatOption?.enableToolCall === 1 ? 'auto' : undefined;
+        const axiosParams = {
+            ...axiosOption,
+            method: "post",
+            headers: {
+                'Content-Type': 'application/json',
+                'authorization': `Bearer ${this.apiKey}`
+            },
+            data: {
+                model: callChatOption?.model || this.apiOption.model,
+                ...params,
+                messages,
+                stream: streamCall
+            },
+            url: 'https://ark.cn-beijing.volces.com/api/v3/chat/completions'
+        };
+        if (streamCall) axiosParams.responseType = 'stream';
+        return axiosParams;
+    }
+    /**
+     * 流式的聊天模式
+     * @param chatText 
+     * @param _paramOption 
+     * @param axiosOption 
+     */
+    override async chatRequestInStream(chatText: string | Array<any>, callChatOption: any, attach?: any, axiosOption?: any): Promise<any> {
+        if (!chatText) this.emit('chaterror', { successed: false, error: 'no text in chat' });
+        axiosOption = Object.assign({}, axiosOption || { timeout: 10000 })
+        const callParams = this.assembleApiParams(chatText, true, callChatOption, axiosOption);
+        let requestid = Math.ceil(Math.random() * (new Date().getTime() * Math.random()) / 1000), replytext: string[] = [];
+        try {
+            requestStream(callParams, (chunk: any) => {
+                let streamText = chunk.toString().replace('[DONE]', '').replace(/[\r\n]+/gm, '')
+                this.processChunkData(streamText.split(/data: /), requestid, replytext, attach)
+            })
+            return { successed: true, requestid }
+        } catch (error) {
+            this.emit('requesterror', { successed: false, requestid, error: 'call axios faied ' + error });
+            return { successed: false, requestid }
+        }
+    }
+    /**
+     * 处理每次流式返回的对话片段
+     * @param chunks 
+     * @param requestid 
+     * @param replytext 
+     * @param attach 
+     */
+    processChunkData(chunks: string[], requestid: Number, replytext: string[], attach: any) {
+        let has_tool_calls = 0, currentIndex, previous_index = -1, tool_calls: any[] = [];// 使用数组来存储工具调用
+        for (const splitString of chunks) {
+            if (!splitString) continue;
+            const chunk = JSON.parse(splitString);
+            const [choice] = chunk.choices,
+                { finish_reason: finishreason, index, usage } = choice,
+                { content, tool_calls: toolCalls } = choice.delta;
+            if (toolCalls && toolCalls.length) {
+                currentIndex = toolCalls[0].index;
+                has_tool_calls = 1;
+                if (currentIndex !== previous_index) {
+                    tool_calls.push({
+                        id: toolCalls[0].id,
+                        type: 'function',
+                        function: {
+                            name: toolCalls[0].function.name,
+                            arguments: toolCalls[0].function.arguments
+                        }
+                    });
+                    // 更新previousIndex以供下次比较
+                    previous_index = currentIndex;
+                } else {
+                    tool_calls[previous_index].function.arguments += toolCalls[0].function.arguments
+                }
+            } else replytext.push(content);
+            let output = { successed: true, requestid, segment: content, text: replytext.join(''), finish_reason: finishreason, index, usage, has_tool_calls: has_tool_calls, tool_calls: tool_calls };
+            if (attach) output = Object.assign({}, output, attach);
+            this.emit(finishreason ? 'chatdone' : 'chattext', output)
+        }
+    }
+}

package/src/gptprovider.ts

@@ -8,6 +8,8 @@ import AzureAI from './azureai'
 import StabilityAI from './stabilityai'
 import StabilityPlusAI from './stabilityplusai'
 import BaiduWenXinAI, { ApiCredential } from './baiduai'
+import AIMiddlePlatform  from './aimp';
+import DouBaoAI from './doubaoai'
 import GptBase from './gptbase';
 /**
  * OpenAI/NLP 的服务提供商 OpenAI,微软，百度文心(待接入)，google(待接入)
@@ -16,7 +18,9 @@ export const GptProviderEnum = {
     OPENAI: 'openai',
     OPENAIPROXY:'openaiproxy',
     MICROSOFT: 'microsoft',
+    AIMP: 'aimp',   ///AI 中台业务
     BAIDU: 'baidu',
+    DOUBAO:'doubao',
     GOOGLE:'google',
     STABILITY:'stability',
     STABILITY2: 'stability2',
@@ -30,7 +34,7 @@ export type GptProviderEnum = typeof GptProviderEnum[keyof typeof GptProviderEnu
  * @returns 
  */
 export function createGpt(provider: GptProviderEnum, apikey: string|ApiCredential, setting: any): GptBase | null {
-    let { model, maxtoken, temperature, serviceurl,endpoint, engine, version, embedding, top_p, presence_penalty, frequency_penalty } = setting || {};
+    let { model,agentid, maxtoken, temperature, serviceurl,endpoint, engine, version, embedding, top_p, presence_penalty, frequency_penalty } = setting || {};
     switch (provider) {
         case GptProviderEnum.OPENAI: 
             return new OpenAIGpt(apikey + '', { model, maxtoken, temperature, embedding, top_p, presence_penalty, frequency_penalty });
@@ -39,8 +43,12 @@ export function createGpt(provider: GptProviderEnum, apikey: string|ApiCredentia
         case GptProviderEnum.MICROSOFT: 
             return new AzureAI(apikey + '', { endpoint, engine, version }, { model, maxtoken, temperature, embedding, top_p, presence_penalty, frequency_penalty }, );
         case GptProviderEnum.BAIDU:
-            let cred: ApiCredential = typeof (apikey) === 'string' ? { apikey, securitykey: apikey } : apikey
+            let cred: ApiCredential = typeof (apikey) === 'string' ? { apikey, securitykey: apikey } : apikey;
             return new BaiduWenXinAI(cred);
+        case GptProviderEnum.AIMP:
+            return new AIMiddlePlatform(apikey+'',{ endpoint,agentid });
+        case GptProviderEnum.DOUBAO:
+            return new DouBaoAI(apikey + '', { model, maxtoken, temperature, top_p, presence_penalty, frequency_penalty })
         case GptProviderEnum.STABILITY:
             return new StabilityAI(apikey + '', { endpoint, engine }, setting);
         case GptProviderEnum.STABILITY2: