fix: [ML-505]: Fix generate step API input type (#3120)

* fix: [ML-505]: Fix generate step API input type
pull/3597/head
Yogesh Chauhan 2024-12-06 10:18:29 +00:00 committed by Harness
parent cfa4e351c4
commit f82dd44d57
2 changed files with 2 additions and 2 deletions

View File

@ -24,7 +24,7 @@ import (
func (c *Controller) GeneratePipelineStep(
ctx context.Context,
in *controllertypes.GeneratePipelineInput,
in *controllertypes.GeneratePipelineStepInput,
) (*controllertypes.GeneratePipelineStepOutput, error) {
generateRequest := &aitypes.PipelineStepGenerateRequest{
Prompt: in.Prompt,

View File

@ -27,7 +27,7 @@ func HandleGeneratePipelineStep(aiagentCtrl *aiagent.Controller) http.HandlerFun
return func(w http.ResponseWriter, r *http.Request) {
ctx := r.Context()
in := new(controllertypes.GeneratePipelineInput)
in := new(controllertypes.GeneratePipelineStepInput)
err := json.NewDecoder(r.Body).Decode(in)
if err != nil {
render.BadRequestf(ctx, w, "Invalid Request Body: %s.", err)