Skip to main content

atomr_infer_runtime_gemini/
lib.rs

1//! # inference-runtime-gemini
2//!
3//! Google Gemini runtime for both AI Studio (`api_key`) and Vertex AI
4//! (OAuth2 over `project + region`). Doc §10.3.
5//!
6//! OAuth2 token refresh is exposed as a pluggable
7//! [`CredentialProvider`](atomr_infer_remote_core::session::CredentialProvider)
8//! trait so we don't pull a full `oauth2` stack into the workspace
9//! root. Operators wire `StaticApiKey` for AI Studio or supply their
10//! own provider for Vertex (typically `gcloud auth print-access-token`
11//! refreshed on a timer).
12
13#![forbid(unsafe_code)]
14#![deny(rust_2018_idioms)]
15
16pub mod config;
17pub mod cost;
18pub mod error;
19pub mod runner;
20pub mod wire;
21
22pub use config::{GeminiConfig, GeminiVariant, SafetySetting};
23pub use cost::GeminiPricing;
24pub use error::classify_gemini_error;
25pub use runner::GeminiRunner;