diff --git a/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro-vision.yaml b/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro-vision.yaml
index da3bc8a64a..ebb276b8af 100644
--- a/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro-vision.yaml
+++ b/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro-vision.yaml
@@ -3,9 +3,8 @@ label:
   en_US: Gemini 1.0 Pro Vision
 model_type: llm
 features:
+  - agent-thought
   - vision
-  - tool-call
-  - stream-tool-call
 model_properties:
   mode: chat
   context_size: 16384
diff --git a/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro.yaml b/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro.yaml
index 029fab718c..c325973846 100644
--- a/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro.yaml
+++ b/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro.yaml
@@ -4,8 +4,6 @@ label:
 model_type: llm
 features:
   - agent-thought
-  - tool-call
-  - stream-tool-call
 model_properties:
   mode: chat
   context_size: 32760
diff --git a/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-flash.yaml b/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-flash.yaml
index 72b8410aa1..c308f0a322 100644
--- a/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-flash.yaml
+++ b/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-flash.yaml
@@ -1,11 +1,10 @@
-model: gemini-1.5-flash-preview-0514
+model: gemini-1.5-flash-001
 label:
   en_US: Gemini 1.5 Flash
 model_type: llm
 features:
+  - agent-thought
   - vision
-  - tool-call
-  - stream-tool-call
 model_properties:
   mode: chat
   context_size: 1048576
diff --git a/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-pro.yaml b/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-pro.yaml
index 141f61aad6..744863e773 100644
--- a/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-pro.yaml
+++ b/api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-pro.yaml
@@ -1,12 +1,10 @@
-model: gemini-1.5-pro-preview-0514
+model: gemini-1.5-pro-001
 label:
   en_US: Gemini 1.5 Pro
 model_type: llm
 features:
   - agent-thought
   - vision
-  - tool-call
-  - stream-tool-call
 model_properties:
   mode: chat
   context_size: 1048576