Add AIGW completion migration for Resolve Vulnerability
What does this MR do and why?
Part of #482070, this MR adds the Gitlab::Llm::AiGateway::Completions::ResolveVulnerability
completion which moves the prompt to the AI Gateway (see feat: add prompt for resolve_vulnerability (gitlab-org/modelops/applied-ml/code-suggestions/ai-assist!1334 - merged)).
The changes in this MR are activated by the prompt_migration_resolve_vulnerability
feature flag.
This MR was originally started in Add AIGW service class for resolve_vulnerability (!165884 - closed) which was split up into this MR and Extracts Resolve Vulnerability completion helpers (!170475 - merged)
References
Please include cross links to any resources that are relevant to this MR This will give reviewers and future readers helpful context to give an efficient review of the changes introduced.
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
- Setup - If needed, follow the instructions in this MR to set up Resolve Vulnerability with AI in the GDK - https://handbook.gitlab.com/handbook/engineering/development/sec/govern/threat-insights/ve_vr_setup/
- First, ensure the original completion works by ensuring the
prompt_migration_resolve_vulnerability
feature flag is disabled and testing Resolve Vulnerability for any supported CWE listed in a project's vulnerability report. - Tail the AI Gateway logs with
gdk tail gitlab-ai-gateway
. - The path of the request in the logs should be
path=/v1/proxy/anthropic/v1/messages
if theprompt_migration_resolve_vulnerability
feature flag is disabled. - Next, enable the
prompt_migration_resolve_vulnerability
feature flag in the GDK. - Test Resolve Vulnerability for the same CWE you tested previously.
- Tail the AI Gateway logs with
gdk tail gitlab-ai-gateway
. - In the logs, you should see
Initializing prompt registry from local yaml
- The path of the request in the logs should be
path=/v1/prompts/resolve_vulnerability
if theprompt_migration_resolve_vulnerability
feature flag is enabled.