Restord Gemini Flash

devoxx · Jul 4, 2024 · 6e87d2d · 6e87d2d
1 parent 358562e
commit 6e87d2d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/java/com/devoxx/genie/service/LLMModelRegistryService.java b/src/main/java/com/devoxx/genie/service/LLMModelRegistryService.java
@@ -264,10 +264,10 @@ private void addGeminiModels() {
 
         models.add(LanguageModel.builder()
             .provider(ModelProvider.Gemini)
-            .modelName("gemini-1.5-pro-latest")
-            .displayName("Gemini 1.5 Pro")
-            .inputCost(7.0)
-            .outputCost(21.0)
+            .modelName("gemini-1.5-flash-latest")
+            .displayName("Gemini 1.5 Flash")
+            .inputCost(0.7)
+            .outputCost(2.1)
             .contextWindow(1_000_000)
             .apiKeyUsed(true)
             .build());