@@ -17,22 +17,28 @@ limitations under the License.
17
17
package handlers
18
18
19
19
import (
20
+ "context"
20
21
"fmt"
21
22
"encoding/json"
22
23
24
+ "sigs.k8s.io/controller-runtime/pkg/log"
25
+ logutil "sigs.k8s.io/gateway-api-inference-extension/pkg/epp/util/logging"
23
26
basepb "github.com/envoyproxy/go-control-plane/envoy/config/core/v3"
24
27
eppb "github.com/envoyproxy/go-control-plane/envoy/service/ext_proc/v3"
25
28
)
26
29
27
30
// HandleRequestBody handles request bodies.
28
- func (s * Server ) HandleRequestBody (body * eppb.HttpBody ) (* eppb.ProcessingResponse , error ) {
31
+ func (s * Server ) HandleRequestBody (ctx context.Context , body * eppb.HttpBody ) (* eppb.ProcessingResponse , error ) {
32
+ logger := log .FromContext (ctx )
33
+
29
34
var data map [string ]any
30
35
if err := json .Unmarshal (body .GetBody (), & data ); err != nil {
31
36
return nil , err
32
37
}
33
38
34
39
modelVal , ok := data ["model" ]
35
40
if ! ok {
41
+ logger .V (logutil .DEFAULT ).Info ("Request body does not contain model parameter" )
36
42
return & eppb.ProcessingResponse {
37
43
Response : & eppb.ProcessingResponse_RequestBody {
38
44
RequestBody : & eppb.BodyResponse {},
@@ -42,6 +48,7 @@ func (s *Server) HandleRequestBody(body *eppb.HttpBody) (*eppb.ProcessingRespons
42
48
43
49
modelStr , ok := modelVal .(string )
44
50
if ! ok {
51
+ logger .V (logutil .DEFAULT ).Error ("Model parameter value is not a string" )
45
52
return & eppb.ProcessingResponse {
46
53
Response : & eppb.ProcessingResponse_RequestBody {
47
54
RequestBody : & eppb.BodyResponse {},
0 commit comments