com.konfigthis.carbonai.client.model.EmbeddingAndChunk Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of carbonai-java-sdk Show documentation
Show all versions of carbonai-java-sdk Show documentation
Connect external data to LLMs, no matter the source.
This library was generated by https://konfigthis.com
/*
* Carbon
* Connect external data to LLMs, no matter the source.
*
* The version of the OpenAPI document: 1.0.0
*
*
* NOTE: This class is auto generated by Konfig (https://konfigthis.com).
* Do not edit the class manually.
*/
package com.konfigthis.carbonai.client.model;
import java.util.Objects;
import java.util.Arrays;
import com.google.gson.TypeAdapter;
import com.google.gson.annotations.JsonAdapter;
import com.google.gson.annotations.SerializedName;
import com.google.gson.stream.JsonReader;
import com.google.gson.stream.JsonWriter;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonArray;
import com.google.gson.JsonDeserializationContext;
import com.google.gson.JsonDeserializer;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParseException;
import com.google.gson.TypeAdapterFactory;
import com.google.gson.reflect.TypeToken;
import org.apache.commons.lang3.StringUtils;
import java.lang.reflect.Type;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;
import com.konfigthis.carbonai.client.JSON;
/**
* EmbeddingAndChunk
*/@javax.annotation.Generated(value = "Generated by https://konfigthis.com")
public class EmbeddingAndChunk {
public static final String SERIALIZED_NAME_USER_FILE_ID = "user_file_id";
@SerializedName(SERIALIZED_NAME_USER_FILE_ID)
private Integer userFileId;
public static final String SERIALIZED_NAME_CHUNK_INDEX = "chunk_index";
@SerializedName(SERIALIZED_NAME_CHUNK_INDEX)
private Integer chunkIndex;
public static final String SERIALIZED_NAME_SOURCE_CONTENT = "source_content";
@SerializedName(SERIALIZED_NAME_SOURCE_CONTENT)
private String sourceContent;
public static final String SERIALIZED_NAME_EMBEDDING = "embedding";
@SerializedName(SERIALIZED_NAME_EMBEDDING)
private List embedding = new ArrayList<>();
public static final String SERIALIZED_NAME_CONTENT_METADATA = "content_metadata";
@SerializedName(SERIALIZED_NAME_CONTENT_METADATA)
private Object contentMetadata;
public EmbeddingAndChunk() {
}
public EmbeddingAndChunk userFileId(Integer userFileId) {
this.userFileId = userFileId;
return this;
}
/**
* Get userFileId
* @return userFileId
**/
@javax.annotation.Nonnull
@ApiModelProperty(required = true, value = "")
public Integer getUserFileId() {
return userFileId;
}
public void setUserFileId(Integer userFileId) {
this.userFileId = userFileId;
}
public EmbeddingAndChunk chunkIndex(Integer chunkIndex) {
this.chunkIndex = chunkIndex;
return this;
}
/**
* Get chunkIndex
* @return chunkIndex
**/
@javax.annotation.Nullable
@ApiModelProperty(required = true, value = "")
public Integer getChunkIndex() {
return chunkIndex;
}
public void setChunkIndex(Integer chunkIndex) {
this.chunkIndex = chunkIndex;
}
public EmbeddingAndChunk sourceContent(String sourceContent) {
this.sourceContent = sourceContent;
return this;
}
/**
* Get sourceContent
* @return sourceContent
**/
@javax.annotation.Nonnull
@ApiModelProperty(required = true, value = "")
public String getSourceContent() {
return sourceContent;
}
public void setSourceContent(String sourceContent) {
this.sourceContent = sourceContent;
}
public EmbeddingAndChunk embedding(List embedding) {
this.embedding = embedding;
return this;
}
public EmbeddingAndChunk addEmbeddingItem(Double embeddingItem) {
this.embedding.add(embeddingItem);
return this;
}
/**
* Get embedding
* @return embedding
**/
@javax.annotation.Nullable
@ApiModelProperty(required = true, value = "")
public List getEmbedding() {
return embedding;
}
public void setEmbedding(List embedding) {
this.embedding = embedding;
}
public EmbeddingAndChunk contentMetadata(Object contentMetadata) {
this.contentMetadata = contentMetadata;
return this;
}
/**
* Get contentMetadata
* @return contentMetadata
**/
@javax.annotation.Nullable
@ApiModelProperty(required = true, value = "")
public Object getContentMetadata() {
return contentMetadata;
}
public void setContentMetadata(Object contentMetadata) {
this.contentMetadata = contentMetadata;
}
/**
* A container for additional, undeclared properties.
* This is a holder for any undeclared properties as specified with
* the 'additionalProperties' keyword in the OAS document.
*/
private Map additionalProperties;
/**
* Set the additional (undeclared) property with the specified name and value.
* If the property does not already exist, create it otherwise replace it.
*
* @param key name of the property
* @param value value of the property
* @return the EmbeddingAndChunk instance itself
*/
public EmbeddingAndChunk putAdditionalProperty(String key, Object value) {
if (this.additionalProperties == null) {
this.additionalProperties = new HashMap();
}
this.additionalProperties.put(key, value);
return this;
}
/**
* Return the additional (undeclared) property.
*
* @return a map of objects
*/
public Map getAdditionalProperties() {
return additionalProperties;
}
/**
* Return the additional (undeclared) property with the specified name.
*
* @param key name of the property
* @return an object
*/
public Object getAdditionalProperty(String key) {
if (this.additionalProperties == null) {
return null;
}
return this.additionalProperties.get(key);
}
@Override
public boolean equals(Object o) {
if (this == o) {
return true;
}
if (o == null || getClass() != o.getClass()) {
return false;
}
EmbeddingAndChunk embeddingAndChunk = (EmbeddingAndChunk) o;
return Objects.equals(this.userFileId, embeddingAndChunk.userFileId) &&
Objects.equals(this.chunkIndex, embeddingAndChunk.chunkIndex) &&
Objects.equals(this.sourceContent, embeddingAndChunk.sourceContent) &&
Objects.equals(this.embedding, embeddingAndChunk.embedding) &&
Objects.equals(this.contentMetadata, embeddingAndChunk.contentMetadata)&&
Objects.equals(this.additionalProperties, embeddingAndChunk.additionalProperties);
}
@Override
public int hashCode() {
return Objects.hash(userFileId, chunkIndex, sourceContent, embedding, contentMetadata, additionalProperties);
}
@Override
public String toString() {
StringBuilder sb = new StringBuilder();
sb.append("class EmbeddingAndChunk {\n");
sb.append(" userFileId: ").append(toIndentedString(userFileId)).append("\n");
sb.append(" chunkIndex: ").append(toIndentedString(chunkIndex)).append("\n");
sb.append(" sourceContent: ").append(toIndentedString(sourceContent)).append("\n");
sb.append(" embedding: ").append(toIndentedString(embedding)).append("\n");
sb.append(" contentMetadata: ").append(toIndentedString(contentMetadata)).append("\n");
sb.append(" additionalProperties: ").append(toIndentedString(additionalProperties)).append("\n");
sb.append("}");
return sb.toString();
}
/**
* Convert the given object to string with each line indented by 4 spaces
* (except the first line).
*/
private String toIndentedString(Object o) {
if (o == null) {
return "null";
}
return o.toString().replace("\n", "\n ");
}
public static HashSet openapiFields;
public static HashSet openapiRequiredFields;
static {
// a set of all properties/fields (JSON key names)
openapiFields = new HashSet();
openapiFields.add("user_file_id");
openapiFields.add("chunk_index");
openapiFields.add("source_content");
openapiFields.add("embedding");
openapiFields.add("content_metadata");
// a set of required properties/fields (JSON key names)
openapiRequiredFields = new HashSet();
openapiRequiredFields.add("user_file_id");
openapiRequiredFields.add("chunk_index");
openapiRequiredFields.add("source_content");
openapiRequiredFields.add("embedding");
openapiRequiredFields.add("content_metadata");
}
/**
* Validates the JSON Object and throws an exception if issues found
*
* @param jsonObj JSON Object
* @throws IOException if the JSON Object is invalid with respect to EmbeddingAndChunk
*/
public static void validateJsonObject(JsonObject jsonObj) throws IOException {
if (jsonObj == null) {
if (!EmbeddingAndChunk.openapiRequiredFields.isEmpty()) { // has required fields but JSON object is null
throw new IllegalArgumentException(String.format("The required field(s) %s in EmbeddingAndChunk is not found in the empty JSON string", EmbeddingAndChunk.openapiRequiredFields.toString()));
}
}
// check to make sure all required properties/fields are present in the JSON string
for (String requiredField : EmbeddingAndChunk.openapiRequiredFields) {
if (jsonObj.get(requiredField) == null) {
throw new IllegalArgumentException(String.format("The required field `%s` is not found in the JSON string: %s", requiredField, jsonObj.toString()));
}
}
if (!jsonObj.get("source_content").isJsonPrimitive()) {
throw new IllegalArgumentException(String.format("Expected the field `source_content` to be a primitive type in the JSON string but got `%s`", jsonObj.get("source_content").toString()));
}
// ensure the required json array is present
if (jsonObj.get("embedding") == null) {
throw new IllegalArgumentException("Expected the field `linkedContent` to be an array in the JSON string but got `null`");
} else if (!jsonObj.get("embedding").isJsonArray()) {
throw new IllegalArgumentException(String.format("Expected the field `embedding` to be an array in the JSON string but got `%s`", jsonObj.get("embedding").toString()));
}
}
public static class CustomTypeAdapterFactory implements TypeAdapterFactory {
@SuppressWarnings("unchecked")
@Override
public TypeAdapter create(Gson gson, TypeToken type) {
if (!EmbeddingAndChunk.class.isAssignableFrom(type.getRawType())) {
return null; // this class only serializes 'EmbeddingAndChunk' and its subtypes
}
final TypeAdapter elementAdapter = gson.getAdapter(JsonElement.class);
final TypeAdapter thisAdapter
= gson.getDelegateAdapter(this, TypeToken.get(EmbeddingAndChunk.class));
return (TypeAdapter) new TypeAdapter() {
@Override
public void write(JsonWriter out, EmbeddingAndChunk value) throws IOException {
JsonObject obj = thisAdapter.toJsonTree(value).getAsJsonObject();
obj.remove("additionalProperties");
// serialize additonal properties
if (value.getAdditionalProperties() != null) {
for (Map.Entry entry : value.getAdditionalProperties().entrySet()) {
if (entry.getValue() instanceof String)
obj.addProperty(entry.getKey(), (String) entry.getValue());
else if (entry.getValue() instanceof Number)
obj.addProperty(entry.getKey(), (Number) entry.getValue());
else if (entry.getValue() instanceof Boolean)
obj.addProperty(entry.getKey(), (Boolean) entry.getValue());
else if (entry.getValue() instanceof Character)
obj.addProperty(entry.getKey(), (Character) entry.getValue());
else {
obj.add(entry.getKey(), gson.toJsonTree(entry.getValue()).getAsJsonObject());
}
}
}
elementAdapter.write(out, obj);
}
@Override
public EmbeddingAndChunk read(JsonReader in) throws IOException {
JsonObject jsonObj = elementAdapter.read(in).getAsJsonObject();
validateJsonObject(jsonObj);
// store additional fields in the deserialized instance
EmbeddingAndChunk instance = thisAdapter.fromJsonTree(jsonObj);
for (Map.Entry entry : jsonObj.entrySet()) {
if (!openapiFields.contains(entry.getKey())) {
if (entry.getValue().isJsonPrimitive()) { // primitive type
if (entry.getValue().getAsJsonPrimitive().isString())
instance.putAdditionalProperty(entry.getKey(), entry.getValue().getAsString());
else if (entry.getValue().getAsJsonPrimitive().isNumber())
instance.putAdditionalProperty(entry.getKey(), entry.getValue().getAsNumber());
else if (entry.getValue().getAsJsonPrimitive().isBoolean())
instance.putAdditionalProperty(entry.getKey(), entry.getValue().getAsBoolean());
else
throw new IllegalArgumentException(String.format("The field `%s` has unknown primitive type. Value: %s", entry.getKey(), entry.getValue().toString()));
} else if (entry.getValue().isJsonArray()) {
instance.putAdditionalProperty(entry.getKey(), gson.fromJson(entry.getValue(), List.class));
} else { // JSON object
instance.putAdditionalProperty(entry.getKey(), gson.fromJson(entry.getValue(), HashMap.class));
}
}
}
return instance;
}
}.nullSafe();
}
}
/**
* Create an instance of EmbeddingAndChunk given an JSON string
*
* @param jsonString JSON string
* @return An instance of EmbeddingAndChunk
* @throws IOException if the JSON string is invalid with respect to EmbeddingAndChunk
*/
public static EmbeddingAndChunk fromJson(String jsonString) throws IOException {
return JSON.getGson().fromJson(jsonString, EmbeddingAndChunk.class);
}
/**
* Convert an instance of EmbeddingAndChunk to an JSON string
*
* @return JSON string
*/
public String toJson() {
return JSON.getGson().toJson(this);
}
}