use reqwest::{Client, RequestBuilder};
use crate::http::model::{ModelReference, ModelType};
#[derive(Clone, PartialEq, serde::Deserialize, serde::Serialize, Debug, Default)]
#[serde(rename_all = "camelCase")]
pub struct ListModelsRequest {
pub max_results: Option<i64>,
}
#[derive(Clone, PartialEq, serde::Deserialize, serde::Serialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct ModelOverview {
pub model_reference: ModelReference,
#[serde(deserialize_with = "crate::http::from_str")]
pub creation_time: i64,
#[serde(deserialize_with = "crate::http::from_str")]
pub last_modified_time: u64,
pub model_type: Option<ModelType>,
}
#[derive(Clone, PartialEq, serde::Deserialize, serde::Serialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct ListModelsResponse {
pub models: Vec<ModelOverview>,
pub next_page_token: Option<String>,
}
pub fn build(
base_url: &str,
client: &Client,
project_id: &str,
dataset_id: &str,
req: &ListModelsRequest,
page_token: Option<String>,
) -> RequestBuilder {
let url = format!("{}/projects/{}/datasets/{}/models", base_url, project_id, dataset_id);
let builder = client.get(url).query(req).query(req);
if let Some(page_token) = page_token {
builder.query(&[("pageToken", page_token.as_str())])
} else {
builder
}
}