Merge pull request #120 from VirdocsSoftware/develop

feat: 🎸 update model and token size (#119)

Author: robaone-redshelf

.github/actions/auto-pr-description/generate_pr_description.js

@@ -107,7 +107,7 @@ ${diffContent}`;
  * Call Gemini API with the given prompt
  */
 async function callGeminiAPI(prompt, apiKey) {
-  const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key=${apiKey}`, {
+  const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent?key=${apiKey}`, {
     method: 'POST',
     headers: { 'Content-Type': 'application/json' },
     body: JSON.stringify({
@@ -120,7 +120,7 @@ async function callGeminiAPI(prompt, apiKey) {
         temperature: 0.7,
         topK: 40,
         topP: 0.95,
-        maxOutputTokens: 2048,
+        maxOutputTokens: 8192,
       }
     })
   });

.github/actions/auto-release-description/generate_pr_description.js

@@ -43,7 +43,7 @@ Keep it concise and focused on the most important changes.`;
   const combinedPrompt = `${promptTemplate}\n\nHere is the git diff:\n\n${diffContent}`;
 
   try {
-    const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key=${apiKey}`, {
+    const response = await fetch(`https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent?key=${apiKey}`, {
       method: 'POST',
       headers: { 'Content-Type': 'application/json' },
       body: JSON.stringify({
@@ -56,7 +56,7 @@ Keep it concise and focused on the most important changes.`;
           temperature: 0.7,
           topK: 40,
           topP: 0.95,
-          maxOutputTokens: 1024,
+          maxOutputTokens: 8192,
         }
       })
     });

.github/actions/confluence-release-notes/generate_release_notes.js

@@ -205,7 +205,7 @@ class GeminiClient {
     const command =
       'Categorize the following as either Improvement or Feature and output to json using schema \'{"categories":[{"ticket_id":"$key","category":"$category"}]}\'';
     const url =
-      "https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-8b:generateContent?key=" +
+      "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent?key=" +
       this.api_key;
     const body = JSON.stringify({
       contents: [
@@ -217,6 +217,9 @@ class GeminiClient {
           ],
         },
       ],
+      generationConfig: {
+        maxOutputTokens: 8192,
+      },
     });
     const response = await this.fetch(url, {
       method: "POST",

.github/actions/llm/invoke_gemini.js

@@ -1,6 +1,11 @@
 import { GoogleGenerativeAI } from "@google/generative-ai";
 const genAI = new GoogleGenerativeAI(process.env.API_KEY);
-const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
+const model = genAI.getGenerativeModel({ 
+  model: "gemini-2.0-flash-exp",
+  generationConfig: {
+    maxOutputTokens: 8192,
+  }
+});
 const prompt = process.env.LLM_PROMPT;
 
 try {