mistral_rouille/models/
embedding_usage.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 * Build date: 2024-06-15T23:41:00.377209-06:00[America/Mexico_City]
7 * Generated using: https://openapi-generator.tech
8 * Open API specification v0.0.2 provided by Mistral @ https://docs.mistral.ai/redocusaurus/plugin-redoc-0.yaml
9 * Custom generation templates by [GovCraft Ai](https://www.govcraft.ai)
10 * Contact: roland@govcraft.ai
11 */
12
13use crate::models;
14use serde::{Deserialize, Serialize};
15
16#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
17pub struct EmbeddingUsage {
18    #[serde(rename = "prompt_tokens")]
19    pub prompt_tokens: i32,
20    #[serde(rename = "total_tokens")]
21    pub total_tokens: i32,
22}
23
24impl EmbeddingUsage {
25    /// Creates a new `EmbeddingUsage` instance.
26    ///
27    /// # Arguments
28    /// * `prompt_tokens`
29    /// * `total_tokens`
30    ///
31    /// # Returns
32    ///
33    /// A new `EmbeddingUsage` instance.
34    pub fn new(prompt_tokens: i32, total_tokens: i32) -> EmbeddingUsage {
35        EmbeddingUsage {
36            prompt_tokens,
37            total_tokens,
38        }
39    }
40}