AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-06-30 19:15:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: Wenming Pan <pwm@google.com>
This commit is contained in:
Pan, Wen-Ming 2024-06-16 09:36:31 +08:00 committed by GitHub
parent ef55d0da78
commit 4b54843ed7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 4 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro-vision.yaml
View File

@ -3,9 +3,8 @@ label:
en_US: Gemini 1.0 Pro Vision en_US: Gemini 1.0 Pro Vision
model_type: llm model_type: llm
features: features:
- agent-thought
- vision - vision
- tool-call
- stream-tool-call
model_properties: model_properties:
mode: chat mode: chat
context_size: 16384 context_size: 16384

2
api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.0-pro.yaml
View File

@ -4,8 +4,6 @@ label:
model_type: llm model_type: llm
features: features:
- agent-thought - agent-thought
- tool-call
- stream-tool-call
model_properties: model_properties:
mode: chat mode: chat
context_size: 32760 context_size: 32760

5
api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-flash.yaml
View File

@ -1,11 +1,10 @@
model: gemini-1.5-flash-preview-0514 model: gemini-1.5-flash-001
label: label:
en_US: Gemini 1.5 Flash en_US: Gemini 1.5 Flash
model_type: llm model_type: llm
features: features:
- agent-thought
- vision - vision
- tool-call
- stream-tool-call
model_properties: model_properties:
mode: chat mode: chat
context_size: 1048576 context_size: 1048576

4
api/core/model_runtime/model_providers/vertex_ai/llm/gemini-1.5-pro.yaml
View File

@ -1,12 +1,10 @@
model: gemini-1.5-pro-preview-0514 model: gemini-1.5-pro-001
label: label:
en_US: Gemini 1.5 Pro en_US: Gemini 1.5 Pro
model_type: llm model_type: llm
features: features:
- agent-thought - agent-thought
- vision - vision
- tool-call
- stream-tool-call
model_properties: model_properties:
mode: chat mode: chat
context_size: 1048576 context_size: 1048576