Enhance ChromaVectorStore initialization and add test for disabled schema

 - Improve ChromaVectorStore to throw exception on missing collection with disabled schema
 - Add test case to verify exception is thrown when collection doesn't exist and schema init is false
 - Correctly set Chroma collection id when initializeSchema set to false

 Co-authored-by: Christian Tzolov <ctzolov@vmware.com>

Author: alexcheng1982

spring-ai-spring-boot-autoconfigure/src/test/java/org/springframework/ai/autoconfigure/vectorstore/chroma/ChromaVectorStoreAutoConfigurationIT.java

@@ -41,6 +41,7 @@
 import org.springframework.context.annotation.Configuration;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.junit.Assert.assertThrows;
 import static org.springframework.ai.autoconfigure.vectorstore.observation.ObservationTestUtil.assertObservationRegistry;
 
 /**
@@ -60,13 +61,12 @@ public class ChromaVectorStoreAutoConfigurationIT {
 		.withUserConfiguration(Config.class)
 		.withPropertyValues("spring.ai.vectorstore.chroma.client.host=http://" + chroma.getHost(),
 				"spring.ai.vectorstore.chroma.client.port=" + chroma.getMappedPort(8000),
-				"spring.ai.vectorstore.chroma.initializeSchema=true",
 				"spring.ai.vectorstore.chroma.collectionName=TestCollection");
 
 	@Test
 	public void addAndSearchWithFilters() {
 
-		contextRunner.run(context -> {
+		contextRunner.withPropertyValues("spring.ai.vectorstore.chroma.initializeSchema=true").run(context -> {
 
 			VectorStore vectorStore = context.getBean(VectorStore.class);
 			TestObservationRegistry observationRegistry = context.getBean(TestObservationRegistry.class);
@@ -125,6 +125,16 @@ public void addAndSearchWithFilters() {
 		});
 	}
 
+	@Test
+	public void throwExceptionOnMissingCollectionAndDisabledInitializedSchema() {
+
+		contextRunner.withPropertyValues("spring.ai.vectorstore.chroma.initializeSchema=false").run(context -> {
+			assertThrows(
+					"Collection TestCollection doesn't exist and won't be created as the initializeSchema is set to false.",
+					java.lang.RuntimeException.class, () -> context.getBean(VectorStore.class));
+		});
+	}
+
 	@Configuration(proxyBeanMethods = false)
 	static class Config {
 

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

@@ -50,6 +50,10 @@
  * their similarity to a query, using the {@link ChromaApi} and {@link EmbeddingModel} for
  * embedding calculations. For more information about how it does this, see the official
  * <a href="https://www.trychroma.com/">Chroma website</a>.
+ * 
+ * @author Christian Tzolov
+ * @author Fu Cheng
+ * 
  */
 public class ChromaVectorStore extends AbstractObservationVectorStore implements InitializingBean {
 
@@ -185,13 +189,16 @@ public String getCollectionId() {
 
 	@Override
 	public void afterPropertiesSet() throws Exception {
-
-		if (!this.initializeSchema)
-			return;
-
 		var collection = this.chromaApi.getCollection(this.collectionName);
 		if (collection == null) {
-			collection = this.chromaApi.createCollection(new ChromaApi.CreateCollectionRequest(this.collectionName));
+			if (initializeSchema) {
+				collection = this.chromaApi
+					.createCollection(new ChromaApi.CreateCollectionRequest(this.collectionName));
+			}
+			else {
+				throw new RuntimeException("Collection " + this.collectionName
+						+ " doesn't exist and won't be created as the initializeSchema is set to false.");
+			}
 		}
 		this.collectionId = collection.id();
 	}