Add Codestral rails aigw support
What does this MR do and why?
issue: #480646 (closed)
We added support for Codestral-on-Vertex on the AI Gateway: Add Codestral on Vertex support for code comple... (gitlab-org/modelops/applied-ml/code-suggestions/ai-assist#564 - closed). We also added a feature flag (use_codestral_for_code_completions) that would toggle the use of Codestral-on-Vertex for direct connections requests (from the IDE to AI Gateway): Add Codestral-on-Vertex model details to /code_... (#479939 - closed).
We need to make sure that the Rails -> AIGW code completion requests specifies Codestral-on-Vertex when use_codestral_for_code_completions is enabled.
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
- Follow the steps here to setup code completion in VS code
- Disable direct connection follow the instruction here
- start type to trigger code completion
- check logs from AIGW side
We can see codestral model details are in the payload now.
Related to #480646 (closed)