From 9ae72cdcf43371ff6e87508e62b260ea24081553 Mon Sep 17 00:00:00 2001
From: QuietRocket <56984521+QuietRocket@users.noreply.github.com>
Date: Fri, 24 May 2024 05:43:06 +0200
Subject: [PATCH] feat: Add Gemini Flash (#4616)

---
 .../google/llm/gemini-1.5-flash-latest.yaml   | 39 +++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 api/core/model_runtime/model_providers/google/llm/gemini-1.5-flash-latest.yaml

diff --git a/api/core/model_runtime/model_providers/google/llm/gemini-1.5-flash-latest.yaml b/api/core/model_runtime/model_providers/google/llm/gemini-1.5-flash-latest.yaml
new file mode 100644
index 000000000..24b1c5af8
--- /dev/null
+++ b/api/core/model_runtime/model_providers/google/llm/gemini-1.5-flash-latest.yaml
@@ -0,0 +1,39 @@
+model: gemini-1.5-flash-latest
+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
+parameter_rules:
+  - name: temperature
+    use_template: temperature
+  - name: top_p
+    use_template: top_p
+  - name: top_k
+    label:
+      zh_Hans: 取样数量
+      en_US: Top k
+    type: int
+    help:
+      zh_Hans: 仅从每个后续标记的前 K 个选项中采样。
+      en_US: Only sample from the top K options for each subsequent token.
+    required: false
+  - name: max_tokens_to_sample
+    use_template: max_tokens
+    required: true
+    default: 8192
+    min: 1
+    max: 8192
+  - name: response_format
+    use_template: response_format
+pricing:
+  input: '0.00'
+  output: '0.00'
+  unit: '0.000001'
+  currency: USD