Fix Chroma API 0.4.22+ compatiblity issues

 - Makes upsert a void metod.
 - Fix affected test.
 - Replac getEmebedding with complex filters test by query Embedding.

 Resolves #224

Author: tzolov

spring-ai-docs/src/main/antora/modules/ROOT/pages/api/vectordbs/chroma.adoc

@@ -36,7 +36,7 @@ Add these dependencies to your project:
 <dependency>
  <groupId>org.springframework.ai</groupId>
  <artifactId>spring-ai-openai-spring-boot-starter</artifactId>
- <version>0.7.0-SNAPSHOT</version>
+ <version>0.8.0-SNAPSHOT</version>
 </dependency>
 ----
 
@@ -47,7 +47,7 @@ Add these dependencies to your project:
 <dependency>
   <groupId>org.springframework.ai</groupId>
   <artifactId>spring-ai-chroma-store</artifactId>
-  <version>0.7.0-SNAPSHOT</version>
+  <version>0.8.0-SNAPSHOT</version>
 </dependency>
 ----
 

vector-stores/spring-ai-chroma/pom.xml

@@ -62,29 +62,4 @@
 
 	</dependencies>
 
-	<!-- <build>
-		<plugins>
-			<plugin>
-				<groupId>io.swagger.codegen.v3</groupId>
-				<artifactId>swagger-codegen-maven-plugin</artifactId>
-				<version>3.0.50</version>
-				<executions>
-					<execution>
-						<goals>
-							<goal>generate</goal>
-						</goals>
-						<configuration>
-							<inputSpec>src/test/resources/api.yaml</inputSpec>
-							<language>java</language>
-							<configOptions>
-								<sourceFolder>src/gen/java/main</sourceFolder>
-							</configOptions>
-						</configuration>
-					</execution>
-				</executions>
-			</plugin>
-		</plugins>
-	</build> -->
-
-
 </project>

vector-stores/spring-ai-chroma/src/main/java/org/springframework/ai/chroma/ChromaApi.java

@@ -316,9 +316,9 @@ public List<Collection> listCollections() {
 	// Chroma Collection API (https://docs.trychroma.com/js_reference/Collection)
 	//
 
-	public Boolean upsertEmbeddings(String collectionId, AddEmbeddingsRequest embedding) {
+	public void upsertEmbeddings(String collectionId, AddEmbeddingsRequest embedding) {
 
-		return this.restTemplate
+		this.restTemplate
 			.exchange(this.baseUrl + "/api/v1/collections/{collection_id}/upsert", HttpMethod.POST,
 					this.getHttpEntityFor(embedding), Boolean.class, collectionId)
 			.getBody();

vector-stores/spring-ai-chroma/src/main/java/org/springframework/ai/vectorsore/ChromaVectorStore.java

@@ -100,12 +100,8 @@ public void add(List<Document> documents) {
 			embeddings.add(JsonUtils.toFloatArray(document.getEmbedding()));
 		}
 
-		var success = this.chromaApi.upsertEmbeddings(this.collectionId,
+		this.chromaApi.upsertEmbeddings(this.collectionId,
 				new AddEmbeddingsRequest(ids, embeddings, metadatas, contents));
-
-		if (!success) {
-			throw new RuntimeException("Unsuccessful storing!");
-		}
 	}
 
 	@Override

vector-stores/spring-ai-chroma/src/test/java/org/springframework/ai/chroma/ChromaApiIT.java

@@ -45,7 +45,7 @@
 public class ChromaApiIT {
 
 	@Container
-	static GenericContainer<?> chromaContainer = new GenericContainer<>("ghcr.io/chroma-core/chroma:0.4.15")
+	static GenericContainer<?> chromaContainer = new GenericContainer<>("ghcr.io/chroma-core/chroma:0.4.22.dev44")
 		.withExposedPorts(8000);
 
 	@Autowired
@@ -85,9 +85,7 @@ public void testCollection() {
 				List.of(Map.of(), Map.of("key1", "value1", "key2", true, "key3", 23.4)),
 				List.of("Hello World", "Big World"));
 
-		var success = chroma.upsertEmbeddings(newCollection.id(), addEmbeddingRequest);
-
-		assertThat(success).isTrue();
+		chroma.upsertEmbeddings(newCollection.id(), addEmbeddingRequest);
 
 		var addEmbeddingRequest2 = new AddEmbeddingsRequest("id3", new float[] { 3f, 3f, 3f },
 				Map.of("key1", "value1", "key2", true, "key3", 23.4), "Big World");
@@ -96,18 +94,30 @@ public void testCollection() {
 
 		assertThat(chroma.countEmbeddings(newCollection.id())).isEqualTo(3);
 
+		var queryResult = chroma.queryCollection(newCollection.id(),
+				new QueryRequest(List.of(1f, 1f, 1f), 3, chroma.where("""
+						{
+							"key2" : { "$eq": true }
+						}
+						""")));
+		assertThat(queryResult.ids().get(0)).hasSize(2);
+		assertThat(queryResult.ids().get(0)).containsExactlyInAnyOrder("id2", "id3");
+
 		// Update existing embedding.
 		chroma.upsertEmbeddings(newCollection.id(), new AddEmbeddingsRequest("id3", new float[] { 6f, 6f, 6f },
 				Map.of("key1", "value2", "key2", false, "key4", 23.4), "Small World"));
 
 		var result = chroma.getEmbeddings(newCollection.id(), new GetEmbeddingsRequest(List.of("id2")));
 		assertThat(result.ids().get(0)).isEqualTo("id2");
 
-		result = chroma.getEmbeddings(newCollection.id(), new GetEmbeddingsRequest(List.of(), chroma.where("""
-				{ "key2" : { "$eq": true} }
-				""")));
-
-		assertThat(result.ids()).containsExactlyInAnyOrder("id2");
+		queryResult = chroma.queryCollection(newCollection.id(),
+				new QueryRequest(List.of(1f, 1f, 1f), 3, chroma.where("""
+						{
+							"key2" : { "$eq": true }
+						}
+						""")));
+		assertThat(queryResult.ids().get(0)).hasSize(1);
+		assertThat(queryResult.ids().get(0)).containsExactlyInAnyOrder("id2");
 	}
 
 	@Test

vector-stores/spring-ai-chroma/src/test/java/org/springframework/ai/vectorstore/BasicAuthChromaWhereIT.java

@@ -55,7 +55,7 @@ public class BasicAuthChromaWhereIT {
 	 * https://docs.trychroma.com/usage-guide#basic-authentication
 	 */
 	@Container
-	static GenericContainer<?> chromaContainer = new GenericContainer<>("ghcr.io/chroma-core/chroma:0.4.15")
+	static GenericContainer<?> chromaContainer = new GenericContainer<>("ghcr.io/chroma-core/chroma:0.4.22")
 		.withEnv("CHROMA_SERVER_AUTH_CREDENTIALS_FILE", "server.htpasswd")
 		.withEnv("CHROMA_SERVER_AUTH_CREDENTIALS_PROVIDER",
 				"chromadb.auth.providers.HtpasswdFileServerAuthCredentialsProvider")

vector-stores/spring-ai-chroma/src/test/java/org/springframework/ai/vectorstore/ChromaVectorStoreIT.java

@@ -46,7 +46,7 @@
 public class ChromaVectorStoreIT {
 
 	@Container
-	static GenericContainer<?> chromaContainer = new GenericContainer<>("ghcr.io/chroma-core/chroma:0.4.15")
+	static GenericContainer<?> chromaContainer = new GenericContainer<>("ghcr.io/chroma-core/chroma:0.4.22")
 		.withExposedPorts(8000);
 
 	List<Document> documents = List.of(

vector-stores/spring-ai-chroma/src/test/java/org/springframework/ai/vectorstore/TokenSecuredChromaWhereIT.java

@@ -57,7 +57,7 @@ public class TokenSecuredChromaWhereIT {
 	 * https://docs.trychroma.com/usage-guide#static-api-token-authentication
 	 */
 	@Container
-	static GenericContainer<?> chromaContainer = new GenericContainer<>("ghcr.io/chroma-core/chroma:0.4.15")
+	static GenericContainer<?> chromaContainer = new GenericContainer<>("ghcr.io/chroma-core/chroma:0.4.22")
 		.withEnv("CHROMA_SERVER_AUTH_CREDENTIALS", CHROMA_SERVER_AUTH_CREDENTIALS)
 		.withEnv("CHROMA_SERVER_AUTH_CREDENTIALS_PROVIDER",
 				"chromadb.auth.token.TokenConfigServerAuthCredentialsProvider")