@terranlabs/appflow-langchain
Version:
Use langchain in appflow
20 lines (17 loc) • 682 B
JavaScript
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;