vapi_client/models/
google_model.rs

1/*
2 * Vapi API
3 *
4 * API for building voice assistants
5 *
6 * The version of the OpenAPI document: 1.0
7 *
8 * Generated by: https://openapi-generator.tech
9 */
10
11use serde::{Deserialize, Serialize};
12use utoipa::ToSchema;
13
14
15use crate::models;
16
17#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize, ToSchema)]
18pub struct GoogleModel {
19    /// This is the starting state for the conversation.
20    #[serde(rename = "messages", skip_serializing_if = "Option::is_none")]
21    pub messages: Option<Vec<models::OpenAiMessage>>,
22    /// These are the tools that the assistant can use during the call. To use existing tools, use `toolIds`.  Both `tools` and `toolIds` can be used together.
23    #[serde(rename = "tools", skip_serializing_if = "Option::is_none")]
24    pub tools: Option<Vec<models::AnyscaleModelToolsInner>>,
25    /// These are the tools that the assistant can use during the call. To use transient tools, use `tools`.  Both `tools` and `toolIds` can be used together.
26    #[serde(rename = "toolIds", skip_serializing_if = "Option::is_none")]
27    pub tool_ids: Option<Vec<String>>,
28    #[serde(rename = "knowledgeBase", skip_serializing_if = "Option::is_none")]
29    pub knowledge_base: Option<models::AnyscaleModelKnowledgeBase>,
30    /// This is the ID of the knowledge base the model will use.
31    #[serde(rename = "knowledgeBaseId", skip_serializing_if = "Option::is_none")]
32    pub knowledge_base_id: Option<String>,
33    /// This is the Google model that will be used.
34    #[serde(rename = "model")]
35    pub model: Model,
36    #[serde(rename = "provider")]
37    pub provider: Provider,
38    /// This is the session configuration for the Gemini Flash 2.0 Multimodal Live API. Only applicable if the model `gemini-2.0-flash-realtime-exp` is selected.
39    #[serde(rename = "realtimeConfig", skip_serializing_if = "Option::is_none")]
40    pub realtime_config: Option<models::GoogleRealtimeConfig>,
41    /// This is the temperature that will be used for calls. Default is 0 to leverage caching for lower latency.
42    #[serde(rename = "temperature", skip_serializing_if = "Option::is_none")]
43    pub temperature: Option<f64>,
44    /// This is the max number of tokens that the assistant will be allowed to generate in each turn of the conversation. Default is 250.
45    #[serde(rename = "maxTokens", skip_serializing_if = "Option::is_none")]
46    pub max_tokens: Option<f64>,
47    /// This determines whether we detect user's emotion while they speak and send it as an additional info to model.  Default `false` because the model is usually are good at understanding the user's emotion from text.  @default false
48    #[serde(
49        rename = "emotionRecognitionEnabled",
50        skip_serializing_if = "Option::is_none"
51    )]
52    pub emotion_recognition_enabled: Option<bool>,
53    /// This sets how many turns at the start of the conversation to use a smaller, faster model from the same provider before switching to the primary model. Example, gpt-3.5-turbo if provider is openai.  Default is 0.  @default 0
54    #[serde(rename = "numFastTurns", skip_serializing_if = "Option::is_none")]
55    pub num_fast_turns: Option<f64>,
56}
57
58impl GoogleModel {
59    pub fn new(model: Model, provider: Provider) -> GoogleModel {
60        GoogleModel {
61            messages: None,
62            tools: None,
63            tool_ids: None,
64            knowledge_base: None,
65            knowledge_base_id: None,
66            model,
67            provider,
68            realtime_config: None,
69            temperature: None,
70            max_tokens: None,
71            emotion_recognition_enabled: None,
72            num_fast_turns: None,
73        }
74    }
75}
76/// This is the Google model that will be used.
77#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize, ToSchema)]
78pub enum Model {
79    #[serde(rename = "gemini-2.0-flash-thinking-exp")]
80    Gemini2Period0FlashThinkingExp,
81    #[serde(rename = "gemini-2.0-pro-exp-02-05")]
82    Gemini2Period0ProExp0205,
83    #[serde(rename = "gemini-2.0-flash")]
84    Gemini2Period0Flash,
85    #[serde(rename = "gemini-2.0-flash-lite-preview-02-05")]
86    Gemini2Period0FlashLitePreview0205,
87    #[serde(rename = "gemini-2.0-flash-exp")]
88    Gemini2Period0FlashExp,
89    #[serde(rename = "gemini-2.0-flash-realtime-exp")]
90    Gemini2Period0FlashRealtimeExp,
91    #[serde(rename = "gemini-1.5-flash")]
92    Gemini1Period5Flash,
93    #[serde(rename = "gemini-1.5-flash-002")]
94    Gemini1Period5Flash002,
95    #[serde(rename = "gemini-1.5-pro")]
96    Gemini1Period5Pro,
97    #[serde(rename = "gemini-1.5-pro-002")]
98    Gemini1Period5Pro002,
99    #[serde(rename = "gemini-1.0-pro")]
100    Gemini1Period0Pro,
101}
102
103impl Default for Model {
104    fn default() -> Model {
105        Self::Gemini2Period0FlashThinkingExp
106    }
107}
108///
109#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize, ToSchema)]
110pub enum Provider {
111    #[serde(rename = "google")]
112    Google,
113}
114
115impl Default for Provider {
116    fn default() -> Provider {
117        Self::Google
118    }
119}