Skip to main content

mistral_openapi_client/models/
embedding_request.rs

1/*
2 * Mistral AI API
3 *
4 * Our Chat Completion and Embeddings APIs specification. Create your account on [La Plateforme](https://console.mistral.ai) to get access and read the [docs](https://docs.mistral.ai) to learn how to use it.
5 *
6 * The version of the OpenAPI document: 1.0.0
7 * 
8 * Generated by: https://openapi-generator.tech
9 */
10
11use crate::models;
12use serde::{Deserialize, Serialize};
13
14#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
15pub struct EmbeddingRequest {
16    /// ID of the model to use.
17    #[serde(rename = "model")]
18    pub model: String,
19    #[serde(rename = "metadata", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
20    pub metadata: Option<Option<std::collections::HashMap<String, serde_json::Value>>>,
21    #[serde(rename = "input")]
22    pub input: Box<models::Input2>,
23    #[serde(rename = "output_dimension", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
24    pub output_dimension: Option<Option<i32>>,
25    /// The data type of the output embeddings when feature available. If not provided, a default output data type will be used.
26    #[serde(rename = "output_dtype", skip_serializing_if = "Option::is_none")]
27    pub output_dtype: Option<models::EmbeddingDtype>,
28    /// The format of embeddings in the response.
29    #[serde(rename = "encoding_format", skip_serializing_if = "Option::is_none")]
30    pub encoding_format: Option<models::EncodingFormat>,
31}
32
33impl EmbeddingRequest {
34    pub fn new(model: String, input: models::Input2) -> EmbeddingRequest {
35        EmbeddingRequest {
36            model,
37            metadata: None,
38            input: Box::new(input),
39            output_dimension: None,
40            output_dtype: None,
41            encoding_format: None,
42        }
43    }
44}
45