com.alibaba.dashscope.embeddings.TextEmbeddingResult Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dashscope-sdk-java Show documentation

The newest version!

// Copyright (c) Alibaba, Inc. and its affiliates.

package com.alibaba.dashscope.embeddings;

import com.alibaba.dashscope.common.DashScopeResult;
import com.alibaba.dashscope.utils.JsonUtils;
import com.google.gson.JsonObject;
import lombok.Data;

@Data
public class TextEmbeddingResult {
  private String requestId;
  private TextEmbeddingOutput output;
  private TextEmbeddingUsage usage;

  private TextEmbeddingResult() {}

  public static TextEmbeddingResult fromDashScopeResult(DashScopeResult dashScopeResult) {
    TextEmbeddingResult res = new TextEmbeddingResult();
    res.output =
        JsonUtils.fromJson((JsonObject) dashScopeResult.getOutput(), TextEmbeddingOutput.class);
    res.usage = JsonUtils.fromJson(dashScopeResult.getUsage(), TextEmbeddingUsage.class);
    res.requestId = dashScopeResult.getRequestId();
    // Type listType = new TypeToken>(){}.getType();
    // res.embeddings = JsonUtils.fromJson(res.output.get("embeddings"), listType);
    return res;
  }
}