UNPKG

@terranlabs/appflow-langchain

Version:

Use langchain in appflow

19 lines (17 loc) 584 B
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