Skip to content

Draft: Add method for retry content blocked request param

Gosia Ksionek requested to merge 454544-refactor-an-ai-gateway-client-for-using-it-with-individual-features-4 into master

What does this MR do and why?

The code changes introduce a new optional parameter retry_content_blocked_requests to the initialize method of the Gitlab::Llm::AiGateway::Client class. This parameter controls whether the client should retry requests that are blocked due to content restrictions. Additionally, the retry_immediately? method has been overridden to allow immediate retries for content-blocked requests.

This new method will require refactor when we decide how to handle blocked requests from Vertex in AI GW.

MR acceptance checklist

Please evaluate this MR against the MR acceptance checklist. It helps you analyze changes to reduce risks in quality, performance, reliability, security, and maintainability.

Screenshots or screen recordings

Screenshots are required for UI changes, and strongly recommended for all other merge requests.

Before After

How to set up and validate locally

Numbered steps to set up and validate the change are strongly suggested.

Related to #454544 (closed)

Merge request reports