Package com.cockroachlabs.data.ai

Class VectorSearchTemplate

java.lang.Object

com.cockroachlabs.data.ai.VectorSearchTemplate

public class VectorSearchTemplate
extends Object

Constructor Summary

Constructors

Constructor	Description
VectorSearchTemplate(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate, org.springframework.transaction.PlatformTransactionManager txManager, AIMetrics metrics)

Method Summary

Modifier and Type	Method	Description
List<VectorSearchResult<Object>>	search(String table, String vectorColumn, float[] queryVector, DistanceMetric metric, int limit)
List<VectorSearchResult<Object>>	search(String table, String vectorColumn, float[] queryVector, DistanceMetric metric, int limit, List<String> columns, String filter, int beamSize)
void	storeEmbedding(String table, String idColumn, Object idValue, String vectorColumn, float[] embedding)
void	storeEmbeddingWithMetadata(String table, String idColumn, Object idValue, String vectorColumn, float[] embedding, Map<String,Object> metadata)
static String	toVectorLiteral(float[] vector)

Methods inherited from class java.lang.Object

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

Constructor Details

VectorSearchTemplate

public VectorSearchTemplate(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate,
 org.springframework.transaction.PlatformTransactionManager txManager,
 AIMetrics metrics)

Method Details

search

public List<VectorSearchResult<Object>> search(String table,
 String vectorColumn,
 float[] queryVector,
 DistanceMetric metric,
 int limit)

search

public List<VectorSearchResult<Object>> search(String table,
 String vectorColumn,
 float[] queryVector,
 DistanceMetric metric,
 int limit,
 List<String> columns,
 String filter,
 int beamSize)

storeEmbedding

public void storeEmbedding(String table,
 String idColumn,
 Object idValue,
 String vectorColumn,
 float[] embedding)

storeEmbeddingWithMetadata

public void storeEmbeddingWithMetadata(String table,
 String idColumn,
 Object idValue,
 String vectorColumn,
 float[] embedding,
 Map<String,Object> metadata)

toVectorLiteral

public static String toVectorLiteral(float[] vector)