Merge pull request #7 from avoidik/patch-1

Update llm_vertex.py - update models list

Author: justyns

README.md

@@ -8,18 +8,12 @@ For Gemini support using AI Studio, please see [llm-gemini](https://github.com/s
 
 Supported models:
 
-- gemini-2.5-pro-preview-06-05
-- gemini-2.5-pro-preview-05-06
-- gemini-2.5-pro-exp-03-25
-- gemini-2.5-flash-preview-05-20
-- gemini-2.5-flash-preview-04-17
-- gemini-2.0-flash-001
-- gemini-2.0-flash-lite-001
-- gemini-2.0-flash-thinking-exp-01-21
-- gemini-1.5-flash
+- gemini-2.5-flash
+- gemini-2.5-pro
+- gemini-2.0-flash-lite
+- gemini-2.0-flash
 - gemini-1.5-pro
-- gemini-1.0-pro
-- gemini-1.0-pro-vision
+- gemini-1.5-flash
 
 ## Installation
 

llm_vertex.py

@@ -10,18 +10,12 @@
 def register_models(register):
     # Source: https://cloud.google.com/vertex-ai/generative-ai/docs/learn/models
     models = [
-        'gemini-2.5-pro-preview-06-05',
-        'gemini-2.5-pro-preview-05-06',
-        'gemini-2.5-pro-exp-03-25',
-        'gemini-2.5-flash-preview-05-20',
-        'gemini-2.5-flash-preview-04-17',
-        'gemini-2.0-flash-001',
-        'gemini-2.0-flash-lite-001',
-        'gemini-2.0-flash-thinking-exp-01-21',
-        'gemini-1.5-flash',
+        'gemini-2.5-flash',
+        'gemini-2.5-pro',
+        'gemini-2.0-flash-lite',
+        'gemini-2.0-flash',
         'gemini-1.5-pro',
-        'gemini-1.0-pro',
-        'gemini-1.0-pro-vision',
+        'gemini-1.5-flash',
     ]
 
     for model in models:
@@ -30,8 +24,8 @@ def register_models(register):
     # TODO: How to register custom models?
 
 class Vertex(llm.Model):
-    model_id = "vertex-gemini-1.0-pro-vision-001"
-    model_name = "gemini-1.0-pro-vision-001"
+    model_id = ""
+    model_name = ""
     can_stream = True
 
     class Options(llm.Options):

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "llm-vertex"
-version = "0.2.2"
+version = "0.2.3"
 description = "Plugin for LLM adding support for Google Cloud Vertex AI"
 dependencies = [
     "llm",

tests/test_llm_vertex.py

@@ -28,7 +28,6 @@ def test_supported_models_list():
     key_models = [
         'gemini-1.5-pro',
         'gemini-1.5-flash',
-        'gemini-1.0-pro',
     ]
 
     registered_models = []