Package org.springframework.ai.embedding

Class PostgresMlEmbeddingClient

java.lang.Object

org.springframework.ai.embedding.PostgresMlEmbeddingClient

All Implemented Interfaces:

EmbeddingClient, org.springframework.beans.factory.InitializingBean

public class PostgresMlEmbeddingClient
extends Object
implements EmbeddingClient, org.springframework.beans.factory.InitializingBean

PostgresML EmbeddingClient

Author:

Toshiaki Maki

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static enum	PostgresMlEmbeddingClient.VectorType

Constructor Summary

Constructors

Constructor	Description
PostgresMlEmbeddingClient(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate)	a constructor
PostgresMlEmbeddingClient(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate, String transformer)	a constructor
PostgresMlEmbeddingClient(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate, String transformer, PostgresMlEmbeddingClient.VectorType vectorType)	a constructor
PostgresMlEmbeddingClient(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate, String transformer, PostgresMlEmbeddingClient.VectorType vectorType, Map<String,Object> kwargs, MetadataMode metadataMode)	a constructor

Method Summary

Modifier and Type	Method	Description
void	afterPropertiesSet()
int	dimensions()
List<Double>	embed(String text)
List<List<Double>>	embed(List<String> texts)
List<Double>	embed(Document document)
EmbeddingResponse	embedForResponse(List<String> texts)

Methods inherited from class java.lang.Object

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

Constructor Details

PostgresMlEmbeddingClient

public PostgresMlEmbeddingClient(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate)

a constructor

Parameters:

jdbcTemplate - JdbcTemplate

PostgresMlEmbeddingClient

public PostgresMlEmbeddingClient(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate,
 String transformer)

a constructor

Parameters:

jdbcTemplate - JdbcTemplate

transformer - huggingface sentence-transformer name

PostgresMlEmbeddingClient

public PostgresMlEmbeddingClient(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate,
 String transformer,
 PostgresMlEmbeddingClient.VectorType vectorType)

a constructor

Parameters:

jdbcTemplate - JdbcTemplate

transformer - huggingface sentence-transformer name

vectorType - vector type in PostgreSQL

PostgresMlEmbeddingClient

public PostgresMlEmbeddingClient(org.springframework.jdbc.core.JdbcTemplate jdbcTemplate,
 String transformer,
 PostgresMlEmbeddingClient.VectorType vectorType,
 Map<String,Object> kwargs,
 MetadataMode metadataMode)

a constructor

Parameters:

jdbcTemplate - JdbcTemplate

transformer - huggingface sentence-transformer name

vectorType - vector type in PostgreSQL

kwargs - optional arguments

Method Details

embed

public List<Double> embed(String text)

Specified by:

embed in interface EmbeddingClient

embed

public List<Double> embed(Document document)

Specified by:

embed in interface EmbeddingClient

embed

public List<List<Double>> embed(List<String> texts)

Specified by:

embed in interface EmbeddingClient

embedForResponse

public EmbeddingResponse embedForResponse(List<String> texts)

Specified by:

embedForResponse in interface EmbeddingClient

dimensions

public int dimensions()

Specified by:

dimensions in interface EmbeddingClient

afterPropertiesSet

public void afterPropertiesSet()

Specified by:

afterPropertiesSet in interface org.springframework.beans.factory.InitializingBean