fix: run agent with Vertex AI Gemini models (#5260)

Co-authored-by: Wenming Pan <pwm@google.com>
2 years ago · 4b54843ed7
parent ef55d0da78
commit 4b54843ed7
4 changed files with 4 additions and 10 deletions
--- 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
--- 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
--- 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
--- 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