com.volcengine.service.maas.MaasConfig Maven / Gradle / Ivy

Go to download

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

The VOLC Engine SDK for Java

There is a newer version: 1.0.192

package com.volcengine.service.maas;

import com.volcengine.helper.Const;
import com.volcengine.model.ApiInfo;
import com.volcengine.model.Credentials;
import com.volcengine.model.ServiceInfo;
import org.apache.http.Header;
import org.apache.http.message.BasicHeader;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

public class MaasConfig {
    public static ServiceInfo getServiceInfo(String host, String region, int connectionTimeout, int socketTimeout) {
        return new ServiceInfo(new HashMap() {
            {
                put(Const.CONNECTION_TIMEOUT, connectionTimeout);
                put(Const.SOCKET_TIMEOUT, socketTimeout);
                put(Const.Scheme, Const.HTTPS);
                put(Const.Host, host);
                put(Const.Header, new ArrayList() {
                    {
                        add(new BasicHeader("Accept", "application/json"));
                    }
                });
                put(Const.Credentials, new Credentials(region, "ml_maas"));
            }
        });
    }

    public static Map getApiInfoList() {
        return new HashMap() {
            {
                put(Const.MaasApiChat, new ApiInfo(new HashMap() {
                    {
                        put(Const.Method, "post");
                        put(Const.Path, "/api/v1/chat");
                    }
                }));

                put(Const.MaasApiTokenization, new ApiInfo(new HashMap() {
                    {
                        put(Const.Method, "post");
                        put(Const.Path, "/api/v1/tokenization");
                    }
                }));

                put(Const.MaasApiClassification, new ApiInfo(new HashMap() {
                    {
                        put(Const.Method, "post");
                        put(Const.Path, "/api/v1/classification");
                    }
                }));

                put(Const.MaasApiEmbeddings, new ApiInfo(new HashMap() {
                    {
                        put(Const.Method, "post");
                        put(Const.Path, "/api/v1/embeddings");
                    }
                }));
            }
        };
    }
}