From 9b283024b47f57ad7bbf032fa015eab42f8162a9 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Mon, 6 May 2024 08:19:42 +0800
Subject: feat: extract vertexai-claude client (#485)

---
 models.yaml | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'models.yaml')

diff --git a/models.yaml b/models.yaml
index 5b57dae..aa7fd45 100644
--- a/models.yaml
+++ b/models.yaml
@@ -238,7 +238,6 @@
   #   - https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/gemini
   # notes:
   #   - get max_output_tokens info from models doc
-  #   - claude models have not been tested
   models:
     - name: gemini-1.0-pro
       max_input_tokens: 24568
@@ -257,6 +256,14 @@
       input_price: 2.5
       output_price: 7.5
       supports_vision: true
+
+- platform: vertexai-claude
+  # docs:
+  #   - https://cloud.google.com/vertex-ai/generative-ai/docs/partner-models/use-claude
+  # notes:
+  #   - get max_output_tokens info from models doc
+  #   - claude models have not been tested
+  models:
     - name: claude-3-opus@20240229
       max_input_tokens: 200000
       max_output_tokens: 4096
-- 
cgit v1.2.3