Package com.tnsai.intelligence.rag.vector

Class JdbcPgvectorTransport

java.lang.Object

com.tnsai.intelligence.rag.vector.JdbcPgvectorTransport

All Implemented Interfaces:

PgvectorTransport

public final class JdbcPgvectorTransport
extends Object

JDK JDBC transport for a pgvector table.

The PostgreSQL driver is loaded reflectively. It is not a compile or required runtime dependency of tnsai-intelligence.

Since:

0.14.0

Nested Class Summary

Nested classes/interfaces inherited from interface com.tnsai.intelligence.rag.vector.PgvectorTransport

PgvectorTransport.Config, PgvectorTransport.IndexKind, PgvectorTransport.Point, PgvectorTransport.ScoredPoint

Constructor Summary

Constructors

Constructor	Description
JdbcPgvectorTransport(PgvectorTransport.Config config)

Method Summary

Modifier and Type	Method	Description
void	closeOwner(String owner, String generation, long tombstoneUntil)
int	countByOwner(String owner)
int	countLeased(String owner, String generation)
boolean	delete(String pointId)
void	deleteByOwner(String owner)
void	deleteExpiredOwners(long expiresAtInclusive, long tombstoneUntil, long purgeTombstonesBefore)
boolean	deleteLeased(String pointId, String owner, String generation)
void	ensureSchema(int dimension)
Optional<PgvectorTransport.Point>	get(String pointId)
void	openOwner(String owner, String generation, long ownerExpiresAt)
void	renewOwner(String owner, String generation, long ownerExpiresAt)
List<PgvectorTransport.ScoredPoint>	search(float[] query, int topK, String owner)
List<PgvectorTransport.ScoredPoint>	searchLeased(float[] query, int topK, String owner, String generation)
void	upsert(PgvectorTransport.Point point)
void	upsertLeased(PgvectorTransport.Point point, String generation, long ownerExpiresAt)

Methods inherited from class java.lang.Object

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

Constructor Details

JdbcPgvectorTransport

public JdbcPgvectorTransport(PgvectorTransport.Config config)

Method Details

ensureSchema

public void ensureSchema(int dimension)

Specified by:

ensureSchema in interface PgvectorTransport

upsert

public void upsert(PgvectorTransport.Point point)

Specified by:

upsert in interface PgvectorTransport

openOwner

public void openOwner(String owner,
 String generation,
 long ownerExpiresAt)

upsertLeased

public void upsertLeased(PgvectorTransport.Point point,
 String generation,
 long ownerExpiresAt)

deleteLeased

public boolean deleteLeased(String pointId,
 String owner,
 String generation)

delete

public boolean delete(String pointId)

Specified by:

delete in interface PgvectorTransport

deleteByOwner

public void deleteByOwner(String owner)

Specified by:

deleteByOwner in interface PgvectorTransport

closeOwner

public void closeOwner(String owner,
 String generation,
 long tombstoneUntil)

deleteExpiredOwners

public void deleteExpiredOwners(long expiresAtInclusive,
 long tombstoneUntil,
 long purgeTombstonesBefore)

renewOwner

public void renewOwner(String owner,
 String generation,
 long ownerExpiresAt)

countByOwner

public int countByOwner(String owner)

Specified by:

countByOwner in interface PgvectorTransport

countLeased

public int countLeased(String owner,
 String generation)

get

public Optional<PgvectorTransport.Point> get(String pointId)

Specified by:

get in interface PgvectorTransport

search

public List<PgvectorTransport.ScoredPoint> search(float[] query,
 int topK,
 String owner)

Specified by:

search in interface PgvectorTransport

searchLeased

public List<PgvectorTransport.ScoredPoint> searchLeased(float[] query,
 int topK,
 String owner,
 String generation)