Merge pull request #45 from CJCrafter/embeddings-dimensions

Embeddings dimensions

Author: CJCrafter

src/main/kotlin/com/cjcrafter/openai/embeddings/EmbeddingsRequest.kt

@@ -12,13 +12,15 @@ import com.fasterxml.jackson.annotation.JsonProperty
  * @property input The input(s) to convert to embeddings.
  * @property model Which [model](https://platform.openai.com/docs/models/embeddings) to use to generate the embeddings.
  * @property encodingFormat Determines how the embeddings are encoded. Defaults to [EncodingFormat.FLOAT].
+ * @property dimensions The number of dimensions to use for the embeddings.
  * @property user The user ID to associate with this request.
  * @constructor Create empty Embeddings request
  */
 data class EmbeddingsRequest internal constructor(
     var input: Any,
     var model: String,
     @JsonProperty("encoding_format") var encodingFormat: EncodingFormat? = null,
+    var dimensions: Int? = null,
     var user: String? = null,
 ) {
 
@@ -30,20 +32,23 @@ data class EmbeddingsRequest internal constructor(
         private var input: Any? = null
         private var model: String? = null
         private var encodingFormat: EncodingFormat? = null
+        private var dimensions: Int? = null
         private var user: String? = null
 
         fun input(input: String) = apply { this.input = input }
         fun input(input: List<String>) = apply { this.input = input }
         fun model(model: String) = apply { this.model = model }
         fun encodingFormat(encodingFormat: EncodingFormat) = apply { this.encodingFormat = encodingFormat }
+        fun dimensions(dimensions: Int) = apply { this.dimensions = dimensions }
         fun user(user: String) = apply { this.user = user }
 
         fun build(): EmbeddingsRequest {
             return EmbeddingsRequest(
                 input = input ?: throw IllegalStateException("input must be defined to use EmbeddingsRequest"),
                 model = model ?: throw IllegalStateException("model must be defined to use EmbeddingsRequest"),
                 encodingFormat = encodingFormat,
-                user = user
+                dimensions = dimensions,
+                user = user,
             )
         }
     }