Parcourir la source

feat(llm/models): add gemini-1.5-pro (#2925)

Eric Wang il y a 1 an
Parent
commit
e0df7505f6

+ 37 - 0
api/core/model_runtime/model_providers/google/llm/gemini-1.5-pro-latest.yaml

@@ -0,0 +1,37 @@
+model: gemini-1.5-pro-latest
+label:
+  en_US: Gemini 1.5 Pro
+model_type: llm
+features:
+  - agent-thought
+  - vision
+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

+ 7 - 2
api/requirements.txt

@@ -39,10 +39,16 @@ resend~=0.7.0
 pyjwt~=2.8.0
 anthropic~=0.23.1
 newspaper3k==0.2.8
-google-api-python-client==2.90.0
 wikipedia==1.4.0
 readabilipy==0.2.0
+google-ai-generativelanguage==0.6.1
+google-api-core==2.18.0
+google-api-python-client==2.90.0
+google-auth==2.29.0
+google-auth-httplib2==0.2.0
+google-generativeai==0.5.0
 google-search-results==2.4.2
+googleapis-common-protos==1.63.0
 replicate~=0.22.0
 websocket-client~=1.7.0
 dashscope[tokenizer]~=1.14.0
@@ -62,7 +68,6 @@ numpy~=1.25.2
 unstructured[docx,pptx,msg,md,ppt]~=0.10.27
 bs4~=0.0.1
 markdown~=3.5.1
-google-generativeai~=0.3.2
 httpx[socks]~=0.24.1
 matplotlib~=3.8.2
 yfinance~=0.2.35