Add support for the GoogleSearchRetrieval tool

This commit is contained in:
2024-06-14 13:35:21 +01:00
parent 3a97f75a5c
commit 81c7dccbdf
2 changed files with 48 additions and 1 deletions

View File

@@ -27,9 +27,17 @@ impl GenerateContentRequest {
}
}
#[derive(Clone, Serialize, Deserialize)]
#[derive(Clone, Default, Serialize, Deserialize)]
pub struct Tools {
pub function_declarations: Option<Vec<FunctionDeclaration>>,
#[serde(rename = "googleSearchRetrieval")]
pub google_search_retrieval: Option<GoogleSearchRetrieval>,
}
#[derive(Clone, Default, Serialize, Deserialize)]
#[serde(rename_all = "camelCase")]
pub struct GoogleSearchRetrieval {
pub disable_attribution: bool,
}
#[derive(Clone, Debug, Serialize, Deserialize, Default)]