diff --git a/service/prompt/prompt_build_service.go b/service/prompt/prompt_build_service.go
index b2d772b..2822bfb 100644
--- a/service/prompt/prompt_build_service.go
+++ b/service/prompt/prompt_build_service.go
@@ -69,6 +69,7 @@ func compileToProviderRequest(ctx context.Context, ir *IR, chatModel *gateway.As
 		"callbackUrl":    utils.GetCallbackURL(ctx, "/prompt/callback"),
 		"requestPayload": providerReq,
 		"buildType":      req.BuildType,
+		"buildModelName": req.ModelName,
 	}, nil
 }