Skip to content

Add retry support to VertexAI embedding and chat models #1437

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

Closed
wants to merge 2 commits into from

Conversation

markpollack
Copy link
Member

Introduces retry functionality to VertexAI embedding and chat models, enhancing their resilience against transient failures.

It also corrects a typo in the VertexAiEmbeddingConnectionDetails class name.

Key changes:

  • Add RetryTemplate to VertexAiTextEmbeddingModel and VertexAiGeminiChatModel
  • Introduce spring-ai-retry dependency
  • Refactor code to support retry logic
  • Update auto-configuration classes to incorporate retry functionality
  • Fix typo in VertexAiEmbeddingConnectionDetails class name

Addresses #832

Introduces retry functionality to VertexAI embedding and
chat models, enhancing their resilience against transient failures.

It also corrects a typo in the VertexAiEmbeddingConnectionDetails
class name.

Key changes:

* Add RetryTemplate to VertexAiTextEmbeddingModel and VertexAiGeminiChatModel
* Introduce spring-ai-retry dependency
* Refactor code to support retry logic
* Update auto-configuration classes to incorporate retry functionality
* Fix typo in VertexAiEmbeddingConnectionDetails class name
@sobychacko
Copy link
Contributor

Merged via bb88e2f.

@sobychacko sobychacko closed this Oct 2, 2024
@matosoma123
Copy link

@sobychacko/ @markpollack , can someone delete the branch - vertex-retry - since the changes have been merged ? ( for cleanup purposes ? )

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants