Method userCanUseChatModelConfig

coderd/exp_chats.go:905–961

userCanUseChatModelConfig returns chatModelConfigAvailable when the user can use the model config. If err is non-nil, callers must ignore the returned reason because it may be the zero-value availability sentinel.

(
	ctx context.Context,
	userID uuid.UUID,
	modelConfigID uuid.UUID,
)

// use the model config. If err is non-nil, callers must ignore the returned
// reason because it may be the zero-value availability sentinel.
func (api *API) userCanUseChatModelConfig(
	ctx context.Context,
	userID uuid.UUID,
	modelConfigID uuid.UUID,
) (chatModelConfigUnavailableReason, error) {
	if modelConfigID == uuid.Nil {
		return chatModelConfigUnavailableModelNotFoundOrDisabled, nil
	}
	//nolint:gocritic // Non-admin users need deployment config validation.
	model, err := api.Database.GetChatModelConfigByID(
		dbauthz.AsSystemRestricted(ctx),
		modelConfigID,
	)
	if err != nil {
		if errors.Is(err, sql.ErrNoRows) || httpapi.Is404Error(err) {
			return chatModelConfigUnavailableModelNotFoundOrDisabled, nil
		}
		return chatModelConfigAvailable, err
	}
	if !model.Enabled {
		return chatModelConfigUnavailableModelNotFoundOrDisabled, nil
	}

	availability, err := api.getUserChatProviderAvailability(ctx, userID)
	if err != nil {
		return chatModelConfigAvailable, err
	}
	if model.AIProviderID.Valid {
		providerID := model.AIProviderID.UUID
		if _, ok := availability.enabledProviderIDs[providerID]; !ok {
			return chatModelConfigUnavailableProviderDisabled, nil
		}
		providerStatus, ok := availability.providerStatusByID[providerID]
		if !ok {
			return chatModelConfigUnavailableProviderDisabled, nil
		}
		if !providerStatus.Available {
			return chatModelConfigUnavailableCredentialsMissing, nil
		}
		return chatModelConfigAvailable, nil
	}
	provider, _, err := chatprovider.ResolveModelWithProviderHint(model.Model, model.Provider)
	if err != nil {
		return chatModelConfigUnavailableProviderDisabled, nil
	}
	if _, ok := availability.enabledProviderNames[provider]; !ok {
		return chatModelConfigUnavailableProviderDisabled, nil
	}
	providerStatus, ok := availability.providerStatus[provider]
	if !ok {
		return chatModelConfigUnavailableProviderDisabled, nil
	}
	if !providerStatus.Available {
		return chatModelConfigUnavailableCredentialsMissing, nil
	}
	return chatModelConfigAvailable, nil
}