From 9ba3454faf80a6ae8fc97478e82111e84e3e78a9 Mon Sep 17 00:00:00 2001 From: Toshiaki Maki Date: Mon, 28 Oct 2024 23:24:16 +0900 Subject: [PATCH] Fix flaky tests in `PostgresMlEmbeddingModelIT` --- .../ai/postgresml/PostgresMlEmbeddingModel.java | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/models/spring-ai-postgresml/src/main/java/org/springframework/ai/postgresml/PostgresMlEmbeddingModel.java b/models/spring-ai-postgresml/src/main/java/org/springframework/ai/postgresml/PostgresMlEmbeddingModel.java index 14ad9d41c47..664be2b1281 100644 --- a/models/spring-ai-postgresml/src/main/java/org/springframework/ai/postgresml/PostgresMlEmbeddingModel.java +++ b/models/spring-ai-postgresml/src/main/java/org/springframework/ai/postgresml/PostgresMlEmbeddingModel.java @@ -135,7 +135,7 @@ public float[] embed(String text) { return this.jdbcTemplate.queryForObject( "SELECT pgml.embed(?, ?, ?::JSONB)" + this.defaultOptions.getVectorType().cast + " AS embedding", this.defaultOptions.getVectorType().rowMapper, this.defaultOptions.getTransformer(), text, - this.defaultOptions.getKwargs()); + ModelOptionsUtils.toJsonString(this.defaultOptions.getKwargs())); } @Override @@ -203,7 +203,6 @@ PostgresMlEmbeddingOptions mergeOptions(EmbeddingOptions requestOptions) { @Override public void afterPropertiesSet() { this.jdbcTemplate.execute("CREATE EXTENSION IF NOT EXISTS pgml"); - this.jdbcTemplate.execute("CREATE EXTENSION IF NOT EXISTS hstore"); if (StringUtils.hasText(this.defaultOptions.getVectorType().extensionName)) { this.jdbcTemplate .execute("CREATE EXTENSION IF NOT EXISTS " + this.defaultOptions.getVectorType().extensionName);