Method resolvePersonalModelOverride

coderd/x/chatd/subagent.go:368–424

(
	ctx context.Context,
	overrideContext codersdk.ChatModelOverrideContext,
	ownerID uuid.UUID,
	modelConfigID uuid.UUID,
)

}

func (p *Server) resolvePersonalModelOverride(
	ctx context.Context,
	overrideContext codersdk.ChatModelOverrideContext,
	ownerID uuid.UUID,
	modelConfigID uuid.UUID,
) (database.ChatModelConfig, bool, error) {
	modelConfig, providerName, err := p.resolveModelConfigAndNormalizedProvider(
		ctx,
		modelConfigID,
	)
	if err != nil {
		switch {
		case xerrors.Is(err, sql.ErrNoRows):
			p.logger.Debug(ctx,
				"personal model override is unavailable, using deployment default",
				slog.F("override_context", overrideContext),
				slog.F("owner_id", ownerID),
				slog.F("model_config_id", modelConfigID),
			)
		case errors.Is(err, errInvalidModelOverrideMetadata):
			p.logger.Debug(ctx,
				"personal model override metadata is invalid, using deployment default",
				slog.F("override_context", overrideContext),
				slog.F("owner_id", ownerID),
				slog.F("model_config_id", modelConfigID),
				slog.Error(err),
			)
		default:
			p.logger.Warn(ctx,
				"failed to resolve personal model override, using deployment default",
				slog.F("override_context", overrideContext),
				slog.F("owner_id", ownerID),
				slog.F("model_config_id", modelConfigID),
				slog.Error(err),
			)
		}
		return database.ChatModelConfig{}, false, nil
	}
	providerKeys, err := p.resolveUserProviderAPIKeys(ctx, ownerID, modelConfigAIProviderID(modelConfig))
	if err != nil {
		return database.ChatModelConfig{}, false, xerrors.Errorf(
			"resolve provider API keys: %w",
			err,
		)
	}
	if !userCanUseProviderKeys(providerKeys, providerName) {
		p.logger.Debug(ctx,
			"personal model override credentials are unavailable, using deployment default",
			slog.F("override_context", overrideContext),
			slog.F("owner_id", ownerID),
			slog.F("model_config_id", modelConfigID),
			slog.F("provider", providerName),
		)
		return database.ChatModelConfig{}, false, nil
	}
	return modelConfig, true, nil
}