apiVersion: agentgateway.dev/v1alpha1
kind: AgentgatewayPolicy
metadata:
  name: tier-models
  namespace: agentgateway-system
spec:
  targetRefs:
    - group: gateway.networking.k8s.io
      kind: HTTPRoute
      name: model-catalog
  traffic:
    transformation:
      response:
        body: >-
          toJson(json(response.body).merge({"data":
            json(response.body).data +
            json(response.body).data.filter(m, m.id.startsWith("vertex_ai/")).map(m,
              m.merge({"id": "vertex_ai_flex/" + m.id.stripPrefix("vertex_ai/")})) +
            json(response.body).data.filter(m, m.id.startsWith("vertex_ai/")).map(m,
              m.merge({"id": "vertex_ai_batch/" + m.id.stripPrefix("vertex_ai/")}))
          }))