Update Dockerfile

Dockerfile

@@ -5,14 +5,14 @@ WORKDIR /app
 
 COPY <<EOF /app/package.json
 {
-  "name": "gemini-proxy-server",
+  "name": "gemini-api-server",
   "version": "1.0.0",
   "main": "index.js",
   "scripts": {
     "start": "node index.js"
   },
   "dependencies": {
-    "axios": "^1.7.2",
+    "@google/generative-ai": "^0.15.0",
     "cors": "^2.8.5",
     "express": "^4.19.2"
   }
@@ -24,12 +24,11 @@ RUN npm install --only=production
 COPY <<EOF /app/index.js
 const express = require('express');
 const cors = require('cors');
-const axios = require('axios');
+const { GoogleGenerativeAI } = require('@google/generative-ai');
 
 const app = express();
 const PORT = 7860;
-const MODEL_ID = "gemini-flash-lite-latest";
-const API_URL_BASE = "https://generativelanguage.googleapis.com/v1beta/models";
+const MODEL_NAME = "gemini-flash-lite-latest";
 
 const apiKeys = [
     "AIzaSyCL6KXbZDbJ5fBgI4BOrdhJbEDLnuzUL-Y"
@@ -52,33 +51,34 @@ app.post('/api/generate', async (req, res) => {
 
     try {
         const apiKey = getRandomApiKey();
-        const apiUrl = `${API_URL_BASE}/${MODEL_ID}:streamGenerateContent?key=${apiKey}`;
-
-        const requestBody = {
-            contents: [{
-                role: "user",
-                parts: [{ "text": prompt }],
-            }],
-            generationConfig: {
-                thinkingConfig: {
-                    thinkingBudget: 0,
-                },
-            },
+        const genAI = new GoogleGenerativeAI(apiKey);
+        const model = genAI.getGenerativeModel({ model: MODEL_NAME });
+
+        const generationConfig = {
+            temperature: 0.9,
+            topK: 1,
+            topP: 1,
+            maxOutputTokens: 2048,
         };
 
-        const geminiResponse = await axios.post(apiUrl, requestBody, {
-            headers: { 'Content-Type': 'application/json' },
-            responseType: 'stream'
+        const result = await model.generateContentStream({
+            contents: [{ role: "user", parts: [{ text: prompt }] }],
+            generationConfig,
         });
+        
+        res.setHeader('Content-Type', 'text/plain; charset=utf-8');
+        res.setHeader('Transfer-Encoding', 'chunked');
 
-        res.setHeader('Content-Type', geminiResponse.headers['content-type']);
-        geminiResponse.data.pipe(res);
+        for await (const chunk of result.stream) {
+            const chunkText = chunk.text();
+            res.write(chunkText);
+        }
+
+        res.end();
 
     } catch (error) {
-        console.error("Error saat menghubungi Gemini API:", error.message);
-        const status = error.response ? error.response.status : 500;
-        const data = error.response ? error.response.data : { message: "Internal Server Error" };
-        res.status(status).json(data);
+        console.error("Error saat streaming dari Gemini API:", error);
+        res.status(500).send("Terjadi kesalahan pada server saat memproses permintaan Anda.");
     }
 });