RD-2603: Add embeddings mode options to semantic vectors and similar terms endpoints in the model

Author: juliencherry

model/src/main/java/com/basistech/rosette/apimodel/EmbeddingsMode.java

@@ -0,0 +1,28 @@
+/*
+ * Copyright 2023 Basis Technology Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.basistech.rosette.apimodel;
+
+/**
+ * The generation of embeddings to use for all languages. Defaults to second generation unless the language only
+ * supports first generation embeddings
+ */
+public enum EmbeddingsMode {
+    DEFAULT,
+    GEN_1,
+    GEN_2,
+    /**/;
+}

model/src/main/java/com/basistech/rosette/apimodel/SemanticVectorsOptions.java

@@ -1,18 +1,18 @@
 /*
-* Copyright 2018 Basis Technology Corp.
-*
-* Licensed under the Apache License, Version 2.0 (the "License");
-* you may not use this file except in compliance with the License.
-* You may obtain a copy of the License at
-*
-* http://www.apache.org/licenses/LICENSE-2.0
-*
-* Unless required by applicable law or agreed to in writing, software
-* distributed under the License is distributed on an "AS IS" BASIS,
-* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-* See the License for the specific language governing permissions and
-* limitations under the License.
-*/
+ * Copyright 2023 Basis Technology Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
 package com.basistech.rosette.apimodel;
 
 import com.basistech.rosette.annotations.JacksonMixin;
@@ -32,4 +32,9 @@ public class SemanticVectorsOptions extends Options {
      */
     private Boolean perToken;
 
+    /**
+     * @return the generation of embeddings to use for all languages, defaulting to second generation embeddings unless
+     * the language supports first generation embeddings
+     */
+    private EmbeddingsMode embeddingsMode;
 }

model/src/main/java/com/basistech/rosette/apimodel/SimilarTermsOptions.java

@@ -1,5 +1,5 @@
 /*
- * Copyright 2018 Basis Technology Corp.
+ * Copyright 2023 Basis Technology Corp.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -42,4 +42,10 @@ public class SimilarTermsOptions extends Options {
      * @return the number of similar terms requested
      */
     private final Integer count;
+
+    /**
+     * @return the generation of embeddings to use for all languages, defaulting to second generation embeddings unless
+     * the language supports first generation embeddings
+     */
+    private EmbeddingsMode embeddingsMode;
 }