Skip to content

Commit 8c84584

Browse files
committed
fix vertex embedding perf test
1 parent 5f51228 commit 8c84584

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/load_tests/test_vertex_embeddings_load_test.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ def load_vertex_ai_credentials():
5959

6060
async def create_async_vertex_embedding_task():
6161
load_vertex_ai_credentials()
62-
base_url = "https://exampleopenaiendpoint-production.up.railway.app/v1/projects/pathrise-convert-1606954137718/locations/us-central1/publishers/google/models/embedding-gecko-001:predict"
62+
base_url = "https://exampleopenaiendpoint-production.up.railway.app/v1/projects/pathrise-convert-1606954137718/locations/us-central1/publishers/google/models/textembedding-gecko@001"
6363
embedding_args = {
6464
"model": "vertex_ai/textembedding-gecko",
6565
"input": "This is a test sentence for embedding.",

0 commit comments

Comments
 (0)