Method putChatModelOverride

coderd/exp_chats.go:4788–4832

EXPERIMENTAL: this endpoint is experimental and is subject to change.

(rw http.ResponseWriter, r *http.Request)

// EXPERIMENTAL: this endpoint is experimental and is subject to change.
func (api *API) putChatModelOverride(rw http.ResponseWriter, r *http.Request) {
	ctx := r.Context()
	if !api.Authorize(r, policy.ActionUpdate, rbac.ResourceDeploymentConfig) {
		httpapi.Forbidden(rw)
		return
	}
	overrideContext, ok := readChatModelOverrideContext(rw, r)
	if !ok {
		return
	}

	var req codersdk.UpdateChatModelOverrideRequest
	if !httpapi.Read(ctx, rw, r, &req) {
		return
	}

	modelConfigID, err := parseChatModelOverride(req.ModelConfigID)
	if err != nil {
		httpapi.Write(ctx, rw, http.StatusBadRequest, codersdk.Response{
			Message: "Invalid model_config_id.",
			Detail:  fmt.Sprintf("Value %q is not a valid UUID.", req.ModelConfigID),
		})
		return
	}

	status, resp := validateChatModelOverrideID(ctx, api.Database, modelConfigID)
	if resp != nil {
		httpapi.Write(ctx, rw, status, *resp)
		return
	}

	label, err := api.upsertChatModelOverrideConfig(ctx, overrideContext, modelConfigID)
	if err != nil {
		if label == "" {
			label = string(overrideContext)
		}
		httpapi.Write(ctx, rw, http.StatusInternalServerError, codersdk.Response{
			Message: fmt.Sprintf("Internal error updating %s model override.", label),
			Detail:  err.Error(),
		})
		return
	}

	rw.WriteHeader(http.StatusNoContent)
}

func readChatPersonalModelOverrideContext(
	rw http.ResponseWriter,