Method resolveCreateChatModelConfigID

coderd/exp_chats.go:4454–4539

(
	ctx context.Context,
	userID uuid.UUID,
	req codersdk.CreateChatRequest,
)

}

func (api *API) resolveCreateChatModelConfigID(
	ctx context.Context,
	userID uuid.UUID,
	req codersdk.CreateChatRequest,
) (uuid.UUID, int, *codersdk.Response) {
	if req.ModelConfigID != nil {
		if *req.ModelConfigID == uuid.Nil {
			return uuid.Nil, http.StatusBadRequest, &codersdk.Response{
				Message: "Invalid model config ID.",
			}
		}
		return *req.ModelConfigID, 0, nil
	}

	personalOverridesEnabled, err := api.Database.GetChatPersonalModelOverridesEnabled(ctx)
	if err != nil {
		return uuid.Nil, http.StatusInternalServerError, &codersdk.Response{
			Message: "Failed to resolve chat model config.",
			Detail:  err.Error(),
		}
	}
	if !personalOverridesEnabled {
		return api.defaultCreateChatModelConfigID(ctx)
	}

	raw, err := api.Database.GetUserChatPersonalModelOverride(ctx, database.GetUserChatPersonalModelOverrideParams{
		UserID: userID,
		Key:    chatd.ChatPersonalModelOverrideKey(codersdk.ChatPersonalModelOverrideContextRoot),
	})
	if err != nil && !errors.Is(err, sql.ErrNoRows) {
		return uuid.Nil, http.StatusInternalServerError, &codersdk.Response{
			Message: "Failed to resolve chat model config.",
			Detail:  err.Error(),
		}
	}
	if err == nil {
		parsed := parseChatPersonalModelOverrideValue(
			raw,
			codersdk.ChatPersonalModelOverrideContextRoot,
		)
		if parsed.Malformed {
			api.Logger.Debug(
				ctx,
				"unsupported personal root model override mode, using default model",
				slog.F("user_id", userID),
				slog.F("raw_value", raw),
			)
		}
		switch parsed.Mode {
		case codersdk.ChatPersonalModelOverrideModeChatDefault:
			// For root context, chat_default and the defensive default
			// case both fall through to the deployment default model below.
		case codersdk.ChatPersonalModelOverrideModeModel:
			reason, err := api.userCanUseChatModelConfig(
				ctx,
				userID,
				parsed.ModelConfigID,
			)