From e430737c665d3ce0c9ad139db9520c6ee2d087ad Mon Sep 17 00:00:00 2001
From: Kevin Hermawan <84965338+kevinhermawan@users.noreply.github.com>
Date: Tue, 10 Dec 2024 20:00:11 +0700
Subject: [PATCH] feat: adds new Google model (`gemini-1.0-pro-vision`) (#9)

---
 Sources/AIModelRetriever/AIModelRetriever.swift | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/Sources/AIModelRetriever/AIModelRetriever.swift b/Sources/AIModelRetriever/AIModelRetriever.swift
index 9955ba2..f58da31 100644
--- a/Sources/AIModelRetriever/AIModelRetriever.swift
+++ b/Sources/AIModelRetriever/AIModelRetriever.swift
@@ -117,17 +117,16 @@ public extension AIModelRetriever {
 public extension AIModelRetriever {
     /// Retrieves a list of AI models from Google.
     ///
-    /// This method returns a predefined list of Google's AI models.
+    /// The list of available models is sourced from Google's official documentation:
+    /// [Google Models Documentation](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models)
     ///
     /// - Returns: An array of ``AIModel`` that represents Google's available models.
     func google() -> [AIModel] {
         return [
             AIModel(id: "gemini-1.5-flash", name: "Gemini 1.5 Flash"),
-            AIModel(id: "gemini-1.5-flash-8b", name: "Gemini 1.5 Flash-8B"),
             AIModel(id: "gemini-1.5-pro", name: "Gemini 1.5 Pro"),
             AIModel(id: "gemini-1.0-pro", name: "Gemini 1.0 Pro"),
-            AIModel(id: "text-embedding-004", name: "Text Embedding"),
-            AIModel(id: "aqa", name: "AQA")
+            AIModel(id: "gemini-1.0-pro-vision", name: "Gemini 1.0 Pro Vision")
         ]
     }
 }