115 lines
4.8 KiB
TypeScript
115 lines
4.8 KiB
TypeScript
/*****************************************************************************\
|
|
| █████ █████ ██ █ █████ █████ ████ ██ ████ █████ █████ ███ ® |
|
|
| ██ █ ███ █ █ ██ ██ ██ ██ ██ ██ █ ██ ██ █ █ |
|
|
| ██ ███ ████ █ ██ █ ████ █████ ██████ ██ ████ █ █ █ ██ |
|
|
| ██ ██ █ █ ██ █ █ ██ ██ ██ ██ ██ ██ █ ██ ██ █ █ |
|
|
| █████ █████ █ ███ █████ ██ ██ ██ ██ █████ ████ █████ █ ███ |
|
|
| |
|
|
| General Bots Copyright (c) pragmatismo.com.br. All rights reserved. |
|
|
| Licensed under the AGPL-3.0. |
|
|
| |
|
|
| According to our dual licensing model, this program can be used either |
|
|
| under the terms of the GNU Affero General Public License, version 3, |
|
|
| or under a proprietary license. |
|
|
| |
|
|
| The texts of the GNU Affero General Public License with an additional |
|
|
| permission and of our proprietary license can be found at and |
|
|
| in the LICENSE file you have received along with this program. |
|
|
| |
|
|
| This program is distributed in the hope that it will be useful, |
|
|
| but WITHOUT ANY WARRANTY, without even the implied warranty of |
|
|
| MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
|
| GNU Affero General Public License for more details. |
|
|
| |
|
|
| "General Bots" is a registered trademark of pragmatismo.com.br. |
|
|
| The licensing of the program under the AGPLv3 does not imply a |
|
|
| trademark license. Therefore any rights, title and interest in |
|
|
| our trademarks remain entirely with us. |
|
|
| |
|
|
\*****************************************************************************/
|
|
|
|
'use strict';
|
|
|
|
import { GBMinInstance } from 'botlib';
|
|
//import OpenAI from "openai";
|
|
import { ChatGPTAPIBrowser, getOpenAIAuth } from 'chatgpt'
|
|
import { CollectionUtil } from 'pragmatismo-io-framework';
|
|
import { DialogKeywords } from '../../basic.gblib/services/DialogKeywords.js';
|
|
import Path from 'path';
|
|
import * as Fs from 'fs';
|
|
|
|
export class ChatServices {
|
|
|
|
public static async sendMessage(min: GBMinInstance, text: string) {
|
|
let key;
|
|
if (process.env.OPENAI_KEY) {
|
|
key = process.env.OPENAI_KEY;
|
|
}
|
|
else {
|
|
key = min.core.getParam(min.instance, 'Open AI Key', null);
|
|
}
|
|
|
|
if (!key) {
|
|
throw new Error('Open AI Key not configured in .gbot.');
|
|
}
|
|
let functions = [];
|
|
|
|
// Adds .gbdialog as functions if any to GPT Functions.
|
|
|
|
await CollectionUtil.asyncForEach(Object.values(min.scriptMap), async script => {
|
|
const path = DialogKeywords.getGBAIPath(min.botId, "gbdialog", null);
|
|
const localFolder = Path.join('work', path, `${script}.json`);
|
|
|
|
if (Fs.existsSync(localFolder)) {
|
|
const func = Fs.readFileSync(localFolder).toJSON();
|
|
functions.push(func);
|
|
}
|
|
|
|
});
|
|
|
|
// Calls Model.
|
|
|
|
// const openai = new OpenAI({
|
|
// apiKey: key
|
|
// });
|
|
// const chatCompletion = await openai.chat.completions.create({
|
|
// model: "gpt-3.5-turbo",
|
|
// messages: [{ role: "user", content: text }],
|
|
// functions: functions
|
|
// });
|
|
// return chatCompletion.choices[0].message.content;
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
* Generate text
|
|
*
|
|
* CONTINUE keword.
|
|
*
|
|
* result = CONTINUE text
|
|
*
|
|
*/
|
|
public static async continue(min: GBMinInstance, text: string, chatId) {
|
|
let key;
|
|
if (process.env.OPENAI_KEY) {
|
|
key = process.env.OPENAI_KEY;
|
|
}
|
|
else {
|
|
key = min.core.getParam(min.instance, 'Open AI Key', null);
|
|
}
|
|
|
|
if (!key) {
|
|
throw new Error('Open AI Key not configured in .gbot.');
|
|
}
|
|
// const openai = new OpenAI({
|
|
// apiKey: key
|
|
// });
|
|
// const chatCompletion = await openai.chat.completions.create({
|
|
// model: "gpt-3.5-turbo",
|
|
// messages: [{ role: "user", content: text }]
|
|
|
|
// });
|
|
// return chatCompletion.choices[0].message.content;
|
|
}
|
|
}
|