-
Notifications
You must be signed in to change notification settings - Fork 536
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Support models: Anthropic Claude 3 on Vertex AI #964
Conversation
PR Reviewer Guide 🔍(Review updated until commit b0f9b96)
|
PR Code Suggestions ✨
|
@@ -23,6 +23,9 @@ | |||
'meta-llama/Llama-2-7b-chat-hf': 4096, | |||
'vertex_ai/codechat-bison': 6144, | |||
'vertex_ai/codechat-bison-32k': 32000, | |||
'vertex_ai/claude-3-haiku@20240307': 100000, | |||
'vertex_ai/claude-3-sonnet@20240229': 100000, | |||
'vertex_ai/claude-3-opus@20240229': 100000, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
... :
100000
Other existing claude settings use 100000
, then I set 100000
@@ -9,7 +10,7 @@ GitPython==3.1.32 | |||
google-cloud-aiplatform==1.35.0 | |||
google-cloud-storage==2.10.0 | |||
Jinja2==3.1.2 | |||
litellm==1.31.10 | |||
litellm==1.34.42 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Initial version to support claude3 on vertex ai is v1.34.28.dev12.
v1.34.42 is last version of v1.34.x and released on 11th April.
We may update to v1.37.19 without any efforts.
But if we update to v1.37.20, then we need to update openai's version.
# error message to use litellm>=1.37.20
ERROR: Cannot install pr-agent and pr-agent==0.2.2 because these package versions have conflicting dependencies.
The conflict is caused by:
pr-agent 0.2.2 depends on openai==1.13.3
litellm 1.37.20 depends on openai>=1.27.0
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Persistent review updated to latest commit b0f9b96 |
looks good. |
User description
This PR enables to use Claude 3 models (Haiku, Sonnet, Opus) on Vertex AI Model Garden.
Model names and versions are picked from LiteLLM,
https://github.com/BerriAI/litellm/blob/4bf33ef26732374668096b9c3b5c77d0a3345c58/docs/my-website/docs/providers/vertex.md#anthropic
Model descriptions (just for ref):
PR Type
enhancement, dependencies
Description
anthropic[vertex]==0.21.3
.litellm
dependency to version1.34.42
.Changes walkthrough 📝
__init__.py
Add support for Claude 3 models on Vertex AI
pr_agent/algo/init.py
vertex_ai/claude-3-haiku@20240307
,vertex_ai/claude-3-sonnet@20240229
, andvertex_ai/claude-3-opus@20240229
with a token limit of 100000.requirements.txt
Update dependencies for Anthropic Claude 3 support
requirements.txt
anthropic[vertex]==0.21.3
dependency.litellm
dependency to version1.34.42
.