io.github.sinri.AiOnHttpMix.volces.v3.response.VolcesChatResponse Maven / Gradle / Ivy

Go to download
package io.github.sinri.AiOnHttpMix.volces.v3.response;

import io.github.sinri.keel.core.json.UnmodifiableJsonifiableEntity;
import io.vertx.core.json.JsonObject;

import java.util.ArrayList;
import java.util.List;

public interface VolcesChatResponse extends UnmodifiableJsonifiableEntity {
    static VolcesChatResponse wrap(JsonObject jsonObject) {
        return new VolcesChatResponseImpl(jsonObject);
    }

    /**
     * @return 一次 chat completion 接口调用的唯一标识。
     */
    default String getId() {
        return readString("id");
    }

    /**
     * @return 本次对话生成时间戳（秒）。
     */
    default Integer getCreated() {
        return readInteger("created");
    }

    /**
     * @return 实际使用的模型名称和版本。
     */
    default String getModel() {
        return readString("model");
    }

    /**
     * @return 固定为 chat.completion。
     */
    default String getObject() {
        return readString("object");
    }

    /**
     * @return 本次请求的 tokens 用量
     */
    default VolcesChatUsage getUsage() {
        return VolcesChatUsage.wrap(readJsonObject("usage"));
    }

    /**
     * @return 本次请求的模型输出内容
     */
    default List getChoices() {
        List a = readJsonObjectArray("choices");
        if (a == null) return null;
        List list = new ArrayList<>();
        a.forEach(x -> {
            VolcesChatResponseChoice y = VolcesChatResponseChoice.wrap(x);
            list.add(y);
        });
        return list;
    }

}