com.newrelic.api.agent.LlmFeedbackEventAttributes Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of newrelic-api Show documentation
New Relics's Java Api Jar.
There is a newer version: 8.16.0
package com.newrelic.api.agent;

import java.util.HashMap;
import java.util.Map;
import java.util.UUID;

public class LlmFeedbackEventAttributes {
    private final String traceId;
    private final Object rating;
    private final String category;
    private final String message;
    private final Map metadata;
    private final UUID id;
    private final String ingestSource;
    private static final String INGEST_SOURCE = "Java";

    protected LlmFeedbackEventAttributes(String traceId, Object rating, String category, String message, Map metadata, UUID id, String ingestSource) {
        this.traceId = traceId;
        this.rating = rating;
        this.category = category;
        this.message = message;
        this.metadata = metadata;
        this.id = id;
        this.ingestSource = ingestSource;
    }

    public String getTraceId() {
        return traceId;
    }

    public Object getRating() {
        return rating;
    }


    public String getCategory() {
        return category;
    }

    public String getMessage() {
        return message;
    }

    public Map getMetadata() {
        return metadata;
    }

    public UUID getId() {
        return id;
    }

    public String getIngestSource() {
        return ingestSource;
    }

    public Map toMap() {
        Map feedbackAttributesMap = new HashMap<>();
        feedbackAttributesMap.put("trace_id", getTraceId());
        feedbackAttributesMap.put("rating", getRating());
        feedbackAttributesMap.put("id", getId());
        feedbackAttributesMap.put("ingest_source", getIngestSource());
        if (category != null) {
            feedbackAttributesMap.put("category", getCategory());
        }
        if (message != null) {
            feedbackAttributesMap.put("message", getMessage());
        }
        if (metadata != null) {
            feedbackAttributesMap.put("metadata", getMetadata());
        }
        return feedbackAttributesMap;
    }

    public static class Builder {
        private final String traceId;
        private final Object rating;
        private String category = null;
        private String message = null;
        private Map metadata = null;
        private final UUID id = UUID.randomUUID();

        public Builder(String traceId, Object rating) {
            this.traceId = traceId;
            this.rating = rating;
        }

        public Builder category(String category) {
            this.category = category;
            return this;
        }

        public Builder message(String message) {
            this.message = message;
            return this;
        }

        public Builder metadata(Map metadata) {
            this.metadata = metadata;
            return this;
        }

        public Map build() {
            return new LlmFeedbackEventAttributes(traceId, rating, category, message, metadata, id, INGEST_SOURCE).toMap();

        }
    }
}