#![allow(unused_imports)]
#![allow(clippy::too_many_arguments)]
#![allow(clippy::needless_return)]
#![allow(clippy::derivable_impls)]
#![allow(clippy::empty_docs)]
extern crate reqwest;
extern crate serde;
extern crate serde_json;
extern crate serde_repr;
extern crate url;
pub mod apis;
#[cfg(feature = "arrow")]
pub mod arrow;
pub mod auth;
pub mod client;
pub mod field;
pub(crate) mod http;
pub mod http_log;
pub mod models;
pub mod query;
pub mod resources;
pub mod status;
pub mod uploads;
#[cfg(all(test, unix))]
mod test_support;
pub use apis::configuration::{ApiKey, BasicAuth, Configuration};
pub use apis::Error;
#[cfg(feature = "arrow")]
pub use arrow::{
get_result_arrow, stream_result_arrow, ArrowBatchStream, ArrowError, ArrowResult,
ARROW_STREAM_MEDIA_TYPE,
};
pub use auth::{
BearerTokenProvider, PersistCallback, TokenExchangeError, TokenManager, TokenManagerOptions,
};
#[cfg(feature = "arrow")]
pub use client::QueryToArrowError;
pub use client::{AwaitResultError, Client, ClientBuilder, ClientError, PollConfig, QueryOutcome};
pub use query::{
PollPolicy, QueryConfig, QueryError, ResultError, RetryPolicy, TooLargeKind,
DEFAULT_MAX_AUTO_BYTES, DEFAULT_MAX_AUTO_ROWS, OVERLOADED_ERROR_CODE,
};
pub use resources::{
ConnectionTypesApi, ConnectionsApi, DatabaseContextApi, DatabasesApi, EmbeddingProvidersApi,
IndexesApi, InformationSchemaApi, JobsApi, QueryApi, QueryRunsApi, RefreshApi, ResultsApi,
SavedQueriesApi, SecretsApi, UploadsApi, WorkspacesApi,
};
pub use status::{QueryRunStatus, QueryRunStatusExt, ResultStatus, ResultStatusExt};
pub use uploads::{
auto_part_size_hint, effective_in_flight, UploadError, UploadOptions, UploadProgress,
DEFAULT_MAX_CONCURRENCY, DEFAULT_PART_SIZE, MAX_PART_SIZE, MIN_PART_SIZE, TARGET_MAX_PARTS,
UPLOAD_MEMORY_BUDGET,
};
#[cfg(test)]
pub(crate) static ENV_LOCK: std::sync::Mutex<()> = std::sync::Mutex::new(());
pub mod prelude {
pub use crate::apis::configuration::Configuration;
#[cfg(feature = "arrow")]
pub use crate::arrow::{ArrowError, ArrowResult};
pub use crate::client::{Client, ClientBuilder, PollConfig, QueryOutcome};
pub use crate::field;
pub use crate::models::*;
pub use crate::query::{
PollPolicy, QueryConfig, QueryError, ResultError, RetryPolicy, TooLargeKind,
};
pub use crate::resources::*;
pub use crate::status::{QueryRunStatus, QueryRunStatusExt, ResultStatus, ResultStatusExt};
pub use crate::uploads::{UploadError, UploadOptions, UploadProgress};
}