1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
//! # AgentAI Image
//!
//! Image generation and editing for [AgentAI](https://crates.io/crates/agentai): from your
//! code with [`ImageGenerator`], or by an agent with the toolboxes (feature `toolbox`).
//!
//! ## Generating images
//!
//! ```rust,no_run
//! use agentai_image::ImageGenerator;
//!
//! # async fn run() -> Result<(), agentai_image::ImageError> {
//! // The provider comes from the model name, the API key from `OPENAI_API_KEY`
//! let generator = ImageGenerator::new("openai::gpt-image-1")?.with_size("1024x1024");
//! let response = generator.generate("A lighthouse at dawn").await?;
//! // response.image is an agentai `Media`, base64 data or a URL
//! # Ok(())
//! # }
//! ```
//!
//! With [`with_storage`](ImageGenerator::with_storage) every image is saved in a
//! [`MediaSave`](storage::MediaSave) and gets an ID. The [`storage`] module (from
//! `agentai-core`) has the storage traits and [`MemoryStorage`](storage::MemoryStorage).
//!
//! ## Tools for agents
//!
//! - [`GenerateImageToolBox`]: `generate_image` and `edit_image` (the latter needs a storage). The model
//! gets the image, or only its metadata as JSON ([`ImageToolOutput`]).
//! - [`ReadMediaToolBox`]: `read_media` gives the model a stored file by ID.
//!
//! ```rust,no_run
//! # #[cfg(feature = "toolbox")]
//! # async fn run() -> anyhow::Result<()> {
//! use agentai::Agent;
//! use agentai::tool::ToolBoxSet;
//! use agentai_image::storage::MemoryStorage;
//! use agentai_image::{GenerateImageToolBox, ImageGenerator, ReadMediaToolBox};
//!
//! let storage = MemoryStorage::new();
//! let mut tools = ToolBoxSet::new();
//! tools.add_tool(
//! GenerateImageToolBox::new(ImageGenerator::new("openai::gpt-image-1")?)
//! .with_storage(storage.clone()),
//! );
//! tools.add_tool(ReadMediaToolBox::new(storage.clone()));
//! let agent = Agent::new("claude-haiku-5-5", "You are an illustrator.")?.with_toolbox(tools);
//! let result = agent.run::<String>("Draw a lighthouse at dawn").await?;
//! # Ok(())
//! # }
//! ```
//!
//! ## Providers
//!
//! The image APIs live in [`provider`]: [`openai`](provider::openai) (OpenAI and compatible
//! servers) and [`openrouter`](provider::openrouter). [`ImageGenerator::new`] picks one from the
//! model name (`openai::gpt-image-1`, `openrouter::bytedance-seed/seedream-4.5`) with its
//! settings from environment variables. [`provider::list_models`] lists the image models of a
//! provider, with names ready for `ImageGenerator::new`:
//!
//! ```rust,no_run
//! # async fn run() -> Result<(), agentai_image::ImageError> {
//! for model in agentai_image::provider::list_models("openrouter").await? {
//! println!("{}", model.model);
//! }
//! # Ok(())
//! # }
//! ```
//! To set the API key
//! or base URL in code, build the provider and pass it to [`ImageGenerator::from_provider`]:
//!
//! ```rust,no_run
//! use agentai_image::ImageGenerator;
//! use agentai_image::provider::openai::OpenAiImageProvider;
//!
//! let provider = OpenAiImageProvider::new("gpt-image-1", "sk-...", "https://api.openai.com/v1");
//! let generator = ImageGenerator::from_provider(provider);
//! ```
//!
//! Implement [`provider::ImageProvider`] to use another API.
//!
//! ## Features
pub use storage;
pub use ImageGenerator;
pub use ;
pub use ;