1111import io .weaviate .client6 .v1 .api .WeaviateClient ;
1212import io .weaviate .client6 .v1 .api .collections .CollectionConfig ;
1313import io .weaviate .client6 .v1 .api .collections .DataType ;
14+ import io .weaviate .client6 .v1 .api .collections .Generative ;
1415import io .weaviate .client6 .v1 .api .collections .InvertedIndex ;
1516import io .weaviate .client6 .v1 .api .collections .Property ;
1617import io .weaviate .client6 .v1 .api .collections .Quantization ;
1920import io .weaviate .client6 .v1 .api .collections .VectorConfig ;
2021import io .weaviate .client6 .v1 .api .collections .config .Shard ;
2122import io .weaviate .client6 .v1 .api .collections .config .ShardStatus ;
23+ import io .weaviate .client6 .v1 .api .collections .generative .DummyGenerative ;
2224import io .weaviate .client6 .v1 .api .collections .vectorindex .Hnsw ;
2325import io .weaviate .client6 .v1 .api .collections .vectorizers .SelfProvidedVectorizer ;
2426import io .weaviate .containers .Container ;
@@ -234,7 +236,7 @@ public void testNestedProperties() throws IOException {
234236 }
235237
236238 @ Test
237- public void test_updateQuantization () throws IOException {
239+ public void test_updateQuantization_uncompressed () throws IOException {
238240 // Arrange
239241 var nsThings = ns ("Things" );
240242
@@ -255,4 +257,22 @@ public void test_updateQuantization() throws IOException {
255257 .extracting (VectorConfig ::quantization )
256258 .returns (Quantization .Kind .BQ , Quantization ::_kind );
257259 }
260+
261+ @ Test
262+ public void test_updateGenerative () throws IOException {
263+ // Arrange
264+ var nsThings = ns ("Things" );
265+
266+ var things = client .collections .create (nsThings ,
267+ c -> c .vectorConfig (VectorConfig .selfProvided ()));
268+
269+ // Act
270+ things .config .update (c -> c .generativeModule (new DummyGenerative ()));
271+
272+ // Assert
273+ var config = things .config .get ();
274+ Assertions .assertThat (config ).get ()
275+ .extracting (CollectionConfig ::generativeModule ).isNotNull ()
276+ .returns (Generative .Kind .DUMMY , Generative ::_kind );
277+ }
258278}
0 commit comments