Skip to content

Commit c64c504

Browse files
northdpolePRAteek-singHWY
authored andcommitted
Improve logging vertex (#738)
* attempt 1 at fixing the vertex bug * update embeddings model
1 parent 3d5020b commit c64c504

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

application/prompt_client/vertex_prompt_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,15 +71,15 @@ def get_text_embeddings(self, text: str) -> List[float]:
7171
values = []
7272
try:
7373
result = self.client.models.embed_content(
74-
model="gemini-embedding-exp-03-07",
74+
model="models/gemini-embedding-001",
7575
contents=text,
7676
config=types.EmbedContentConfig(task_type="SEMANTIC_SIMILARITY"),
7777
)
7878
if not result:
7979
return None
8080
values = result.embeddings[0].values
8181
except genai.errors.ClientError as e:
82-
logger.info("hit limit, sleeping for a minute")
82+
logger.info(f"hit limit, sleeping for a minute, error was: {repr(e)}")
8383
time.sleep(
8484
60
8585
) # Vertex's quota is per minute, so sleep for a full minute, then try again

0 commit comments

Comments
 (0)