@terranlabs/appflow-langchain
Version:
Use langchain in appflow
19 lines (17 loc) • 584 B
JavaScript
const OpenAI = require("openai");
class OpenAILLM {
constructor(config) {
const api_key = config?.api_key || process.env.OPENAI_API_KEY;
console.log("api_key", api_key)
this.model = new OpenAI({ apiKey: api_key });
}
async generate(prompt) {
const completion = await this.model.chat.completions.create({
messages: [{ role: 'user', content: prompt }],
model: 'gpt-3.5-turbo-1106',
});
const response = completion.choices[0].message.content
return response;
}
}
module.exports = OpenAILLM