Skip to content

Commit e430737

Browse files
feat: adds new Google model (gemini-1.0-pro-vision) (#9)
1 parent 075d52e commit e430737

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

Sources/AIModelRetriever/AIModelRetriever.swift

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -117,17 +117,16 @@ public extension AIModelRetriever {
117117
public extension AIModelRetriever {
118118
/// Retrieves a list of AI models from Google.
119119
///
120-
/// This method returns a predefined list of Google's AI models.
120+
/// The list of available models is sourced from Google's official documentation:
121+
/// [Google Models Documentation](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models)
121122
///
122123
/// - Returns: An array of ``AIModel`` that represents Google's available models.
123124
func google() -> [AIModel] {
124125
return [
125126
AIModel(id: "gemini-1.5-flash", name: "Gemini 1.5 Flash"),
126-
AIModel(id: "gemini-1.5-flash-8b", name: "Gemini 1.5 Flash-8B"),
127127
AIModel(id: "gemini-1.5-pro", name: "Gemini 1.5 Pro"),
128128
AIModel(id: "gemini-1.0-pro", name: "Gemini 1.0 Pro"),
129-
AIModel(id: "text-embedding-004", name: "Text Embedding"),
130-
AIModel(id: "aqa", name: "AQA")
129+
AIModel(id: "gemini-1.0-pro-vision", name: "Gemini 1.0 Pro Vision")
131130
]
132131
}
133132
}

0 commit comments

Comments
 (0)