com.dropchop.textonic.model.dto.doc.output.result.TextVecResult Maven / Gradle / Ivy

Go to download
package com.dropchop.textonic.model.dto.doc.output.result;

import com.dropchop.textonic.model.api.ml.Engine;
import com.dropchop.textonic.model.api.ml.Model;
import com.dropchop.textonic.model.api.ml.StepCode;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.Getter;
import lombok.NoArgsConstructor;
import lombok.Setter;
import lombok.experimental.SuperBuilder;
import org.eclipse.microprofile.openapi.annotations.media.Schema;

import java.util.List;

import static com.fasterxml.jackson.annotation.JsonInclude.Include.NON_NULL;

/**
 * @author Nikola Ivačič  on 11. 08. 22.
 */
@Getter
@Setter
@SuperBuilder
@NoArgsConstructor
@JsonInclude(NON_NULL)
public class TextVecResult extends AnalyzedResult {

  @JsonProperty("v")
  @Schema(
    description = "Vector component values."
  )
  List values;

  public TextVecResult(Engine engine, Model model, StepCode stepCode, String stepDetail, int size) {
    super(engine, model, stepCode, stepDetail, size);
  }

  public TextVecResult(String engineCode, String modelCode, StepCode stepCode, String stepDetail, int size) {
    super(engineCode, modelCode, stepCode, stepDetail, size);
  }

  public TextVecResult(Engine engine, Model model, StepCode stepCode, int size) {
    super(engine, model, stepCode, size);
  }

  @Override
  public String getType() {
    return Type.vec;
  }
}