update example to use Claude sonnet 3.5 (GoogleCloudPlatform#11873)

huythngo · huyngo-google · web-flow · commit 56b73c049a42 · 2024-06-20T14:21:20.000-04:00
Co-authored-by: Huy Ngo &lt;huyngo@google.com&gt;
diff --git a/generative_ai/anthropic_claude_3_streaming.py b/generative_ai/anthropic_claude_3_streaming.py
@@ -26,7 +26,7 @@ def generate_text_streaming(project_id: str, region: str) -> str:
     result = []
 
     with client.messages.stream(
-        model="claude-3-sonnet@20240229",
+        model="claude-3-5-sonnet@20240620",
         max_tokens=1024,
         messages=[
             {
diff --git a/generative_ai/anthropic_claude_3_streaming_test.py b/generative_ai/anthropic_claude_3_streaming_test.py
@@ -21,7 +21,7 @@
 
 
 _PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
-_LOCATION = "us-central1"
+_LOCATION = "us-east5"
 
 
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=10)
diff --git a/generative_ai/anthropic_claude_3_tool_use.py b/generative_ai/anthropic_claude_3_tool_use.py
@@ -24,7 +24,7 @@
 def tool_use(project_id: str, region: str) -> object:
     client = AnthropicVertex(region=region, project_id=project_id)
     message = client.messages.create(
-        model="claude-3-opus@20240229",
+        model="claude-3-5-sonnet@20240620",
         max_tokens=1024,
         tools=[
             {
diff --git a/generative_ai/anthropic_claude_3_unary.py b/generative_ai/anthropic_claude_3_unary.py
@@ -24,7 +24,7 @@ def generate_text(project_id: str, region: str) -> object:
 
     client = AnthropicVertex(region=region, project_id=project_id)
     message = client.messages.create(
-        model="claude-3-sonnet@20240229",
+        model="claude-3-5-sonnet@20240620",
         max_tokens=1024,
         messages=[
             {
diff --git a/generative_ai/anthropic_claude_3_unary_test.py b/generative_ai/anthropic_claude_3_unary_test.py
@@ -20,7 +20,7 @@
 import anthropic_claude_3_unary
 
 _PROJECT_ID = os.getenv("GOOGLE_CLOUD_PROJECT")
-_LOCATION = "us-central1"
+_LOCATION = "us-east5"
 
 
 @backoff.on_exception(backoff.expo, ResourceExhausted, max_time=10)

Original file line number	Diff line number	Diff line change
`@@ -26,7 +26,7 @@ def generate_text_streaming(project_id: str, region: str) -> str:`
`26`	`26`	`result = []`
`27`	`27`
`28`	`28`	`with client.messages.stream(`
`29`		`- model="claude-3-sonnet@20240229",`
	`29`	`+ model="claude-3-5-sonnet@20240620",`
`30`	`30`	`max_tokens=1024,`
`31`	`31`	`messages=[`
`32`	`32`	`{`
Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@`
`24`	`24`	`def tool_use(project_id: str, region: str) -> object:`
`25`	`25`	`client = AnthropicVertex(region=region, project_id=project_id)`
`26`	`26`	`message = client.messages.create(`
`27`		`- model="claude-3-opus@20240229",`
	`27`	`+ model="claude-3-5-sonnet@20240620",`
`28`	`28`	`max_tokens=1024,`
`29`	`29`	`tools=[`
`30`	`30`	`{`
Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@ def generate_text(project_id: str, region: str) -> object:`
`24`	`24`
`25`	`25`	`client = AnthropicVertex(region=region, project_id=project_id)`
`26`	`26`	`message = client.messages.create(`
`27`		`- model="claude-3-sonnet@20240229",`
	`27`	`+ model="claude-3-5-sonnet@20240620",`
`28`	`28`	`max_tokens=1024,`
`29`	`29`	`messages=[`
`30`	`30`	`{`