Skip to content

Commit 5ca18f9

Browse files
committed
feat: register vectorizer adapters in the type adapter factory
1 parent 484c0ba commit 5ca18f9

1 file changed

Lines changed: 54 additions & 2 deletions

File tree

src/main/java/io/weaviate/client6/v1/api/collections/VectorConfig.java

Lines changed: 54 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,35 @@
1616
import com.google.gson.stream.JsonWriter;
1717

1818
import io.weaviate.client6.v1.api.collections.vectorizers.Img2VecNeuralVectorizer;
19+
import io.weaviate.client6.v1.api.collections.vectorizers.Multi2MultiVecJinaAiVectorizer;
20+
import io.weaviate.client6.v1.api.collections.vectorizers.Multi2VecAwsVectorizer;
21+
import io.weaviate.client6.v1.api.collections.vectorizers.Multi2VecBindVectorizer;
1922
import io.weaviate.client6.v1.api.collections.vectorizers.Multi2VecClipVectorizer;
23+
import io.weaviate.client6.v1.api.collections.vectorizers.Multi2VecCohereVectorizer;
24+
import io.weaviate.client6.v1.api.collections.vectorizers.Multi2VecGoogleVectorizer;
25+
import io.weaviate.client6.v1.api.collections.vectorizers.Multi2VecJinaAiVectorizer;
26+
import io.weaviate.client6.v1.api.collections.vectorizers.Multi2VecNvidiaVectorizer;
27+
import io.weaviate.client6.v1.api.collections.vectorizers.Multi2VecVoyageAiVectorizer;
28+
import io.weaviate.client6.v1.api.collections.vectorizers.Ref2VecCentroidVectorizer;
2029
import io.weaviate.client6.v1.api.collections.vectorizers.SelfProvidedVectorizer;
30+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2MultiVecJinaAiVectorizer;
31+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecAwsVectorizer;
32+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecAzureOpenAiVectorizer;
33+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecCohereVectorizer;
2134
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecContextionaryVectorizer;
35+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecDatabricksVectorizer;
36+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecGoogleAiStudioVectorizer;
37+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecGoogleVectorizer;
38+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecHuggingfaceVectorizer;
39+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecJinaAiVectorizer;
40+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecMistralVectorizer;
41+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecModel2VecVectorizer;
42+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecMorphVectorizer;
43+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecNvidiaVectorizer;
44+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecOllamaVectorizer;
45+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecOpenAiVectorizer;
46+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecTransformersVectorizer;
47+
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecVoyageAiVectorizer;
2248
import io.weaviate.client6.v1.api.collections.vectorizers.Text2VecWeaviateVectorizer;
2349
import io.weaviate.client6.v1.internal.ObjectBuilder;
2450
import io.weaviate.client6.v1.internal.TaggedUnion;
@@ -271,10 +297,36 @@ private final void addAdapter(Gson gson, VectorConfig.Kind kind, Class<? extends
271297

272298
private final void init(Gson gson) {
273299
addAdapter(gson, VectorConfig.Kind.NONE, SelfProvidedVectorizer.class);
300+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_AWS, Text2VecAwsVectorizer.class);
301+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_CONTEXTIONARY, Text2VecContextionaryVectorizer.class);
302+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_COHERE, Text2VecCohereVectorizer.class);
303+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_DATABRICKS, Text2VecDatabricksVectorizer.class);
304+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_GOOGLE, Text2VecGoogleVectorizer.class);
305+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_GOOGLEAISTUDIO, Text2VecGoogleAiStudioVectorizer.class);
306+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_HUGGINGFACE, Text2VecHuggingfaceVectorizer.class);
307+
addAdapter(gson, VectorConfig.Kind.REF2VEC_CENTROID, Ref2VecCentroidVectorizer.class);
308+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_JINAAI, Text2VecJinaAiVectorizer.class);
309+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_MISTRAL, Text2VecMistralVectorizer.class);
310+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_MORPH, Text2VecMorphVectorizer.class);
311+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_MODEL2VEC, Text2VecModel2VecVectorizer.class);
312+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_NVIDIA, Text2VecNvidiaVectorizer.class);
313+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_OPENAI, Text2VecOpenAiVectorizer.class);
314+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_AZURE_OPENAI, Text2VecAzureOpenAiVectorizer.class);
315+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_OLLAMA, Text2VecOllamaVectorizer.class);
316+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_TRANSFORMERS, Text2VecTransformersVectorizer.class);
317+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_VOYAGEAI, Text2VecVoyageAiVectorizer.class);
318+
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_WEAVIATE, Text2VecWeaviateVectorizer.class);
274319
addAdapter(gson, VectorConfig.Kind.IMG2VEC_NEURAL, Img2VecNeuralVectorizer.class);
320+
addAdapter(gson, VectorConfig.Kind.MULTI2VEC_AWS, Multi2VecAwsVectorizer.class);
321+
addAdapter(gson, VectorConfig.Kind.MULTI2VEC_BIND, Multi2VecBindVectorizer.class);
275322
addAdapter(gson, VectorConfig.Kind.MULTI2VEC_CLIP, Multi2VecClipVectorizer.class);
276-
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_WEAVIATE, Text2VecWeaviateVectorizer.class);
277-
addAdapter(gson, VectorConfig.Kind.TEXT2VEC_CONTEXTIONARY, Text2VecContextionaryVectorizer.class);
323+
addAdapter(gson, VectorConfig.Kind.MULTI2VEC_GOOGLE, Multi2VecGoogleVectorizer.class);
324+
addAdapter(gson, VectorConfig.Kind.MULTI2VEC_COHERE, Multi2VecCohereVectorizer.class);
325+
addAdapter(gson, VectorConfig.Kind.MULTI2VEC_JINAAI, Multi2VecJinaAiVectorizer.class);
326+
addAdapter(gson, VectorConfig.Kind.MULTI2VEC_NVIDIA, Multi2VecNvidiaVectorizer.class);
327+
addAdapter(gson, VectorConfig.Kind.MULTI2VEC_VOYAGEAI, Multi2VecVoyageAiVectorizer.class);
328+
addAdapter(gson, VectorConfig.Kind.TEXT2MULTIVEC_JINAAI, Text2MultiVecJinaAiVectorizer.class);
329+
addAdapter(gson, VectorConfig.Kind.MULTI2MULTIVEC_JINAAI, Multi2MultiVecJinaAiVectorizer.class);
278330
}
279331

280332
@SuppressWarnings("unchecked")

0 commit comments

Comments
 (0)