Package com.dtsx.astra.sdk.cassio

Class ClusteredMetadataVectorTable

java.lang.Object

com.dtsx.astra.sdk.cassio.AbstractCassandraTable<ClusteredMetadataVectorRecord>

com.dtsx.astra.sdk.cassio.ClusteredMetadataVectorTable

public class ClusteredMetadataVectorTable
extends AbstractCassandraTable<ClusteredMetadataVectorRecord>

Table representing persistence for LangChain operations. - parition key: partitionId - clustering key: rowId - column: value

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	ClusteredMetadataVectorTable.Builder	Builder class for creating instances of ClusteredMetadataVectorTable.

Field Summary

Fields inherited from class com.dtsx.astra.sdk.cassio.AbstractCassandraTable

ATTRIBUTES_BLOB, BODY_BLOB, COLUMN_SIMILARITY, cqlSession, DEFAULT_RECORD_COUNT, keyspaceName, METADATA_S, PARTITION_ID, ROW_ID, SAI_INDEX_CLASSNAME, tableName, VECTOR

Constructor Summary

Constructors

Constructor	Description
ClusteredMetadataVectorTable(@NonNull com.datastax.oss.driver.api.core.CqlSession session, @NonNull String keyspaceName, @NonNull String tableName, @NonNull Integer vectorDimension, @NonNull CassandraSimilarityMetric metric)	Constructor with mandatory parameters.
ClusteredMetadataVectorTable(com.datastax.oss.driver.api.core.CqlSession session, String keyspaceName, String tableName, int vectorDimension)	Constructor with mandatory parameters.

Method Summary

Modifier and Type	Method	Description
static ClusteredMetadataVectorTable.Builder	builder()	Builder for the class.
void	create()	Create table if not exist.
void	delete(String partitionId, UUID rowId)	Access a record by its id
void	deletePartition(@NonNull String partitionDd)	Delete a partition.
List<ClusteredMetadataVectorRecord>	findPartition(@NonNull String partitionDd)	Find a partition.
Optional<ClusteredMetadataVectorRecord>	get(String partitionId, UUID rowId)	Access a record by its id
ClusteredMetadataVectorRecord	mapRow(com.datastax.oss.driver.api.core.cql.Row cqlRow)	Should be table to map from a Cassandra row to a record.
void	put(ClusteredMetadataVectorRecord row)	Upsert a row of the table.
void	save(ClusteredMetadataVectorRecord row)
List<AnnResult<ClusteredMetadataVectorRecord>>	similaritySearch(AnnQuery query)	Compute Similarity Search.

Methods inherited from class com.dtsx.astra.sdk.cassio.AbstractCassandraTable

clear, delete, getCqlSession, putAsync

Methods inherited from class java.lang.Object

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

Constructor Details

ClusteredMetadataVectorTable

public ClusteredMetadataVectorTable(@NonNull
 @NonNull com.datastax.oss.driver.api.core.CqlSession session,
 @NonNull
 @NonNull String keyspaceName,
 @NonNull
 @NonNull String tableName,
 @NonNull
 @NonNull Integer vectorDimension,
 @NonNull
 @NonNull CassandraSimilarityMetric metric)

Constructor with mandatory parameters.

Parameters:

session - cassandra session

keyspaceName - keyspace name

tableName - table name

vectorDimension - vector dimension

metric - similarity metric

ClusteredMetadataVectorTable

public ClusteredMetadataVectorTable(com.datastax.oss.driver.api.core.CqlSession session,
 String keyspaceName,
 String tableName,
 int vectorDimension)

Constructor with mandatory parameters.

Parameters:

session - cassandra session

keyspaceName - keyspace name

tableName - table name

vectorDimension - vector dimension

Method Details

builder

public static ClusteredMetadataVectorTable.Builder builder()

Builder for the class.

Returns:

builder for the class

create

public void create()

Description copied from class: AbstractCassandraTable

Create table if not exist.

Specified by:

create in class AbstractCassandraTable<ClusteredMetadataVectorRecord>

findPartition

public List<ClusteredMetadataVectorRecord> findPartition(@NonNull
 @NonNull String partitionDd)

Find a partition.

Parameters:

partitionDd - partition id

Returns:

list of rows

deletePartition

public void deletePartition(@NonNull
 @NonNull String partitionDd)

Delete a partition.

Parameters:

partitionDd - partition id

get

public Optional<ClusteredMetadataVectorRecord> get(String partitionId,
 UUID rowId)

Access a record by its id

Parameters:

partitionId - partition id

rowId - rowId

Returns:

record if exists

delete

public void delete(String partitionId,
 UUID rowId)

Access a record by its id

Parameters:

partitionId - partition id

rowId - rowId

save

public void save(ClusteredMetadataVectorRecord row)

put

public void put(ClusteredMetadataVectorRecord row)

Description copied from class: AbstractCassandraTable

Upsert a row of the table.

Specified by:

put in class AbstractCassandraTable<ClusteredMetadataVectorRecord>

Parameters:

row - current row

mapRow

public ClusteredMetadataVectorRecord mapRow(com.datastax.oss.driver.api.core.cql.Row cqlRow)

Description copied from class: AbstractCassandraTable

Should be table to map from a Cassandra row to a record.

Specified by:

mapRow in class AbstractCassandraTable<ClusteredMetadataVectorRecord>

Parameters:

cqlRow - current cassandra row

Returns:

current record

similaritySearch

public List<AnnResult<ClusteredMetadataVectorRecord>> similaritySearch(AnnQuery query)

Compute Similarity Search.

Parameters:

query - current query

Returns:

results