<commit_before>package se.arbetsformedlingen.venice.probe;

import se.arbetsformedlingen.venice.common.Application;
import se.arbetsformedlingen.venice.common.Host;
import se.arbetsformedlingen.venice.common.Status;
import se.arbetsformedlingen.venice.common.Version;

import java.time.LocalDateTime;
import java.util.Objects;

public class ProbeResponse {
    private Host host;
    private Application application;
    private Version version;
    private Status status;
    private LocalDateTime checkTimestamp = LocalDateTime.now();

    ProbeResponse(Application application, Host host, Status status, Version version) {
        this.application = application;
        this.host = host;
        this.status = status;
        this.version = version;
    }

    String getHostName() {
        return host.getName();
    }

    String getApplicationName() {
        return application.getApplicationName();
    }

    public String getVersion() {
        return version.getVersion();
    }

    public String getStatus() {
        return status.getStatus();
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        ProbeResponse that = (ProbeResponse) o;
        return Objects.equals(host, that.host) &&
                Objects.equals(application, that.application) &&
                Objects.equals(version, that.version) &&
                Objects.equals(status, that.status);
    }

    @Override
    public int hashCode() {
        return Objects.hash(host, application, version, status);
    }

    @Override
    public String toString() {
        return "ProbeResponse{" +
                "host=" + host +
                ", application=" + application +
                ", version=" + version +
                ", status=" + status +
                ", checkTimestamp=" + checkTimestamp +
                '}';
    }
}
<commit_msg>Return the actual type and not a String<commit_after>package se.arbetsformedlingen.venice.probe;

import se.arbetsformedlingen.venice.common.Application;
import se.arbetsformedlingen.venice.common.Host;
import se.arbetsformedlingen.venice.common.Status;
import se.arbetsformedlingen.venice.common.Version;

import java.time.LocalDateTime;
import java.util.Objects;

public class ProbeResponse {
    private Host host;
    private Application application;
    private Version version;
    private Status status;
    private LocalDateTime checkTimestamp = LocalDateTime.now();

    ProbeResponse(Application application, Host host, Status status, Version version) {
        this.application = application;
        this.host = host;
        this.status = status;
        this.version = version;
    }

    Host getHost() {
        return host;
    }

    Application getApplication() {
        return application;
    }

    public String getVersion() {
        return version.toString();
    }

    public String getStatus() {
        return status.toString();
    }

    @Override
    public boolean equals(Object o) {
        if (this == o) return true;
        if (o == null || getClass() != o.getClass()) return false;
        ProbeResponse that = (ProbeResponse) o;
        return Objects.equals(host, that.host) &&
                Objects.equals(application, that.application) &&
                Objects.equals(version, that.version) &&
                Objects.equals(status, that.status);
    }

    @Override
    public int hashCode() {
        return Objects.hash(host, application, version, status);
    }

    @Override
    public String toString() {
        return "ProbeResponse{" +
                "host=" + host +
                ", application=" + application +
                ", version=" + version +
                ", status=" + status +
                ", checkTimestamp=" + checkTimestamp +
                '}';
    }
}

{'fraction_non_alphanumeric': 0.10326226560493193, 'fraction_numerical': 0.0, 'mean_word_length': 2.3743500866551126, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '5012670', 'n_tokens_mistral': 1042, 'n_tokens_neox': 999, 'n_words': 251}