From 4b54843ed74c3ba20fa42a0dbcde3e79b1076da0 Mon Sep 17 00:00:00 2001 From: "Pan, Wen-Ming" Date: Sun, 16 Jun 2024 09:36:31 +0800 Subject: [PATCH] fix: run agent with Vertex AI Gemini models (#5260) Co-authored-by: Wenming Pan --- .../model_providers/vertex_ai/llm/gemini-1.0-pro-vision.yaml | 3 +-- .../model_providers/vertex_ai/llm/gemini-1.0-pro.yaml | 2 -- .../model_providers/vertex_ai/llm/gemini-1.5-flash.yaml | 5 ++--- .../model_providers/vertex_ai/llm/gemini-1.5-pro.yaml | 4 +--- 4 files changed, 4 insertions(+), 10 deletions(-) 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