mod inference;
mod middleware;
mod pipeline;
mod spec;
mod r#trait;
pub use inference::{InferenceClient, LocalRunnerClient, TurnResult};
pub use middleware::{
AgentMiddleware, LoggingMiddleware, MiddlewareStack, RateLimitMiddleware, RedactionMiddleware,
ToolErrorRecoveryMiddleware,
};
pub use pipeline::Agent;
pub use r#trait::AgentRef;
pub use spec::AgentSpec;
pub use atomr_agents_tool::Provider;
pub mod providers {
#[cfg(feature = "provider-anthropic")]
pub mod anthropic {
pub use atomr_infer_runtime_anthropic::{
classify_anthropic_error, AnthropicConfig, AnthropicPricing, AnthropicRunner,
};
}
#[cfg(feature = "provider-openai")]
pub mod openai {
pub use atomr_infer_runtime_openai::{
classify_openai_error, OpenAiConfig, OpenAiPricing, OpenAiRunner, OpenAiVariant,
};
}
#[cfg(feature = "provider-gemini")]
pub mod gemini {
pub use atomr_infer_runtime_gemini::{
classify_gemini_error, GeminiConfig, GeminiPricing, GeminiRunner, GeminiVariant,
SafetySetting,
};
}
}