Quellcode durchsuchen

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

Co-authored-by: Wenming Pan <pwm@google.com>
Pan, Wen-Ming vor 10 Monaten
Ursprung
Commit
4b54843ed7

+ 1 - 2
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

+ 0 - 2
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

+ 2 - 3
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

+ 1 - 3
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