com.volcengine.service.vikingDB

Class VikingDBService

java.lang.Object

com.volcengine.service.BaseServiceImpl

com.volcengine.service.vikingDB.VikingDBService

All Implemented Interfaces:

IBaseService

public class VikingDBService
extends BaseServiceImpl

Field Summary

Fields inherited from class com.volcengine.service.BaseServiceImpl

apiInfoList, credentials, ISigner, serviceInfo

Constructor Summary

Constructors

Constructor and Description
VikingDBService(String host, String region, String ak, String sk, String scheme)
VikingDBService(String host, String region, String ak, String sk, String scheme, org.apache.http.HttpHost proxy)

Method Summary

Modifier and Type	Method and Description
List<Double>	batchRerank(List<Map<String,Object>> datas)
double	calculateRetryTimeout(int remainingRetries)
HashMap<String,Object>	convertLinkedTreeMapToHashMap(com.google.gson.internal.LinkedTreeMap<String,Object> linkedTreeMap)
static com.google.gson.internal.LinkedTreeMap<String,Object>	convertStringToLinkedTreeMap(String x)
Collection	createCollection(CreateCollectionParam createCollectionParam)
Index	createIndex(CreateIndexParam createIndexParam)
String	createTask(CreateTaskParam createTaskParam)
com.google.gson.internal.LinkedTreeMap<String,Object>	doRequest(String api, List<org.apache.http.NameValuePair> params, HashMap<String,Object> body)
void	dropCollection(String collectionName)
void	dropIndex(String collectionName, String indexName)
void	dropTask(String taskId)
List<List<Double>>	embedding(EmbModel embModel, List<RawData> rawDatas)
List<Double>	embedding(EmbModel embModel, RawData rawData)
Map<String,Object>	embeddingV2(EmbModel embModel, List<RawData> rawDatas)
Map<String,Object>	embeddingV2(EmbModel embModel, RawData rawData)
ExceptionDetails	extractExceptionDetails(String exceptionMessage)
static Map<String,ApiInfo>	getApiInfo()
Collection	getCollection(String collectionName)
Index	getIndex(String collectionName, String indexName)
static ServiceInfo	getServiceInfo(String host, String region, String ak, String sk, String scheme)
Task	getTask(String taskId)
List<Collection>	listCollections()
List<Index>	listIndexes(String collectionName)
List<Task>	listTasks()
Task	packageTask(Map<String,Object> res)
void	requestPing()
Double	rerank(String query, String content, String title)
com.google.gson.internal.LinkedTreeMap<String,Object>	retryRequest(String api, List<org.apache.http.NameValuePair> params, HashMap<String,Object> body, Integer remainingRetries)
void	setHeader(HashMap<String,String> header)
void	updateCollection(UpdateCollectionParam updateCollectionParam)
void	updateIndex(UpdateIndexParam updateIndexParam)
void	updateTask(String taskId, String task_status)

Methods inherited from class com.volcengine.service.BaseServiceImpl

destroy, getAccessKey, getApiInfoList, getHttpClient, getISigner, getRegion, getSecretKey, getServiceInfo, getSessionToken, getSignUrl, json, makeRequest, post, postDataWithResponse, postFileMultiPart, prepareRequest, proto, put, putData, putData, putDataWithResponse, putDataWithResponse, query, setAccessKey, setClientNoReuse, setConnectionTimeout, setHost, setHttpClient, setRegion, setScheme, setSecretKey, setServiceInfo, setSessionToken, setSocketTimeout, signSts2

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

VikingDBService

public VikingDBService(String host,
                       String region,
                       String ak,
                       String sk,
                       String scheme)
                throws Exception

Throws:

Exception

VikingDBService

public VikingDBService(String host,
                       String region,
                       String ak,
                       String sk,
                       String scheme,
                       org.apache.http.HttpHost proxy)
                throws Exception

Throws:

Exception

Method Detail

requestPing

public void requestPing()
                 throws Exception

Throws:

Exception

getServiceInfo

public static ServiceInfo getServiceInfo(String host,
                                         String region,
                                         String ak,
                                         String sk,
                                         String scheme)

getApiInfo

public static Map<String,ApiInfo> getApiInfo()

setHeader

public void setHeader(HashMap<String,String> header)

doRequest

public com.google.gson.internal.LinkedTreeMap<String,Object> doRequest(String api,
                                                                       List<org.apache.http.NameValuePair> params,
                                                                       HashMap<String,Object> body)
                                                                throws Exception

Throws:

Exception

retryRequest

public com.google.gson.internal.LinkedTreeMap<String,Object> retryRequest(String api,
                                                                          List<org.apache.http.NameValuePair> params,
                                                                          HashMap<String,Object> body,
                                                                          Integer remainingRetries)
                                                                   throws Exception

Throws:

Exception

calculateRetryTimeout

public double calculateRetryTimeout(int remainingRetries)

convertStringToLinkedTreeMap

public static com.google.gson.internal.LinkedTreeMap<String,Object> convertStringToLinkedTreeMap(String x)

convertLinkedTreeMapToHashMap

public HashMap<String,Object> convertLinkedTreeMapToHashMap(com.google.gson.internal.LinkedTreeMap<String,Object> linkedTreeMap)

extractExceptionDetails

public ExceptionDetails extractExceptionDetails(String exceptionMessage)

createCollection

public Collection createCollection(CreateCollectionParam createCollectionParam)
                            throws Exception

Throws:

Exception

getCollection

public Collection getCollection(String collectionName)
                         throws Exception

Throws:

Exception

dropCollection

public void dropCollection(String collectionName)
                    throws Exception

Throws:

Exception

listCollections

public List<Collection> listCollections()
                                 throws Exception

Throws:

Exception

createIndex

public Index createIndex(CreateIndexParam createIndexParam)
                  throws Exception

Throws:

Exception

getIndex

public Index getIndex(String collectionName,
                      String indexName)
               throws Exception

Throws:

Exception

dropIndex

public void dropIndex(String collectionName,
                      String indexName)
               throws Exception

Throws:

Exception

listIndexes

public List<Index> listIndexes(String collectionName)
                        throws Exception

Throws:

Exception

updateCollection

public void updateCollection(UpdateCollectionParam updateCollectionParam)
                      throws Exception

Throws:

Exception

updateIndex

public void updateIndex(UpdateIndexParam updateIndexParam)
                 throws Exception

Throws:

Exception

embedding

public List<Double> embedding(EmbModel embModel,
                              RawData rawData)
                       throws Exception

Throws:

Exception

embedding

public List<List<Double>> embedding(EmbModel embModel,
                                    List<RawData> rawDatas)
                             throws Exception

Throws:

Exception

rerank

public Double rerank(String query,
                     String content,
                     String title)
              throws Exception

Throws:

Exception

batchRerank

public List<Double> batchRerank(List<Map<String,Object>> datas)
                         throws Exception

Throws:

Exception

embeddingV2

public Map<String,Object> embeddingV2(EmbModel embModel,
                                      List<RawData> rawDatas)
                               throws Exception

Throws:

Exception

embeddingV2

public Map<String,Object> embeddingV2(EmbModel embModel,
                                      RawData rawData)
                               throws Exception

Throws:

Exception

packageTask

public Task packageTask(Map<String,Object> res)
                 throws Exception

Throws:

Exception

createTask

public String createTask(CreateTaskParam createTaskParam)
                  throws Exception

Throws:

Exception

getTask

public Task getTask(String taskId)
             throws Exception

Throws:

Exception

listTasks

public List<Task> listTasks()
                     throws Exception

Throws:

Exception

dropTask

public void dropTask(String taskId)
              throws Exception

Throws:

Exception

updateTask

public void updateTask(String taskId,
                       String task_status)
                throws Exception

Throws:

Exception