UNPKG

@terranlabs/appflow-langchain

Version:

Use langchain in appflow

20 lines (17 loc) 682 B
const { GoogleGenerativeAI } = require("@google/generative-ai"); // Access your API key as an environment variable (see "Set up your API key" above) class GeminiLLM { constructor(config) { const api_key = config?.api_key || process.env.GEMINI_API_KEY; console.log("api_key", api_key) const genAI = new GoogleGenerativeAI(api_key); this.model = genAI.getGenerativeModel({ model: "gemini-1.5-flash", generationConfig: { temperature: 0.2 } }); } async generate(prompt) { const result = await this.model.generateContent(prompt); const response = await result.response; const text = response.text(); return text; } } module.exports = GeminiLLM;