File size: 659 Bytes
9705b6c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
const { ChatOpenAI } = require('langchain/chat_models/openai');
function createLLM({
modelOptions,
configOptions,
callbacks,
streaming = false,
openAIApiKey,
azure = {},
}) {
let credentials = { openAIApiKey };
let configuration = {
apiKey: openAIApiKey,
};
if (azure) {
credentials = {};
configuration = {};
}
// console.debug('createLLM: configOptions');
// console.debug(configOptions);
return new ChatOpenAI(
{
streaming,
verbose: true,
credentials,
configuration,
...azure,
...modelOptions,
callbacks,
},
configOptions,
);
}
module.exports = createLLM;
|