Skip to content

Commit f379f43

Browse files
yossiovadiaclaude
andcommitted
fix(e2e): configure aibrix profile to use correct model for PII tests
The aibrix profile deploys vllm-llama3-8b-instruct as its LLM model, but the PII detection test was hardcoded to use "general-expert" which doesn't exist in aibrix deployments. This commit configures the aibrix profile to set E2E_TEST_MODEL environment variable to "vllm-llama3-8b-instruct" during setup, ensuring PII tests use the correct model name for aibrix deployments. Related: fix(e2e): make PII test model name configurable via environment variable 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> Signed-off-by: Yossi Ovadia <yovadia@redhat.com>
1 parent d61eeeb commit f379f43

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

e2e/profiles/aibrix/profile.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,9 @@ func (p *Profile) Setup(ctx context.Context, opts *framework.SetupOptions) error
8181
p.verbose = opts.Verbose
8282
p.log("Setting up AIBrix test environment")
8383

84+
// Configure PII test to use the correct model for aibrix
85+
os.Setenv("E2E_TEST_MODEL", deploymentDemoLLM)
86+
8487
deployer := helm.NewDeployer(opts.KubeConfig, opts.Verbose)
8588

8689
// Track what we've deployed for cleanup on error

0 commit comments

Comments
 (0)