feat(harness): media feature with GenerateImageTool / GenerateVideoTool

crates/tinyagents-harness/Cargo.toml

@@ -20,6 +20,7 @@ dirs = { version = "6", optional = true }
 flate2 = { version = "1", optional = true }
 futures = { workspace = true }
 regex = "1"
+rustix = { version = "1", features = ["fs"] }
 reqwest = { workspace = true, features = ["stream", "http2"], optional = true }
 rusqlite = { workspace = true, optional = true }
 serde = { workspace = true }
@@ -31,6 +32,8 @@ tinyagents-definition = { path = "../tinyagents-definition", version = "2.1.2" }
 tinytools-agent = { path = "../../vendor/tinytools/crates/tinytools-agent", version = "0.4.1", default-features = false }
 tinyinference-llm = { path = "../../vendor/tinyinference/crates/tinyinference-llm", version = "0.3.0" }
 tinyinference-embeddings = { path = "../../vendor/tinyinference/crates/tinyinference-embeddings", version = "0.3.0" }
+tinyinference-image = { path = "../../vendor/tinyinference/crates/tinyinference-image", version = "0.3.0", optional = true }
+tinyinference-video = { path = "../../vendor/tinyinference/crates/tinyinference-video", version = "0.3.0", optional = true }
 tinytools = { path = "../../vendor/tinytools/crates/tinytools", version = "0.4.1" }
 tokio = { workspace = true, features = ["sync", "time", "macros", "rt", "rt-multi-thread", "fs", "io-util", "process"] }
 tempfile = { workspace = true }
@@ -45,6 +48,8 @@ sqlite = ["dep:rusqlite"]
 tools = ["builtin-tools"]
 builtin-tools = ["dep:chrono-tz"]
 multimodal = ["dep:flate2", "dep:reqwest"]
+# Image and video generation tools over tinyinference-image / -video.
+media = ["dep:tinyinference-image", "dep:tinyinference-video"]
 # Gates the Claude Code CLI and Claude Agent SDK provider adapters, which are
 # the only consumers of `uuid`, `tempfile` beyond the artifact tests, `dirs`,
 # and `wait-timeout` in this crate.

crates/tinyagents-harness/src/error.rs

@@ -354,6 +354,31 @@ impl From<tinyinference_embeddings::Error> for TinyAgentsError {
     }
 }
 
+#[cfg(feature = "media")]
+impl From<tinyinference_image::Error> for TinyAgentsError {
+    fn from(error: tinyinference_image::Error) -> Self {
+        match error {
+            tinyinference_image::Error::Serialization(error) => Self::Serialization(error),
+            error @ (tinyinference_image::Error::Validation(_)
+            | tinyinference_image::Error::Unsupported { .. }) => {
+                Self::Validation(error.to_string())
+            }
+            other => Self::Model(other.to_string()),
+        }
+    }
+}
+
+#[cfg(feature = "media")]
+impl From<tinyinference_video::Error> for TinyAgentsError {
+    fn from(error: tinyinference_video::Error) -> Self {
+        match error {
+            tinyinference_video::Error::Media(error) => error.into(),
+            error @ tinyinference_video::Error::Timeout { .. } => Self::Timeout(error.to_string()),
+            other => Self::Model(other.to_string()),
+        }
+    }
+}
+
 impl TinyAgentsError {
     /// Builds the right error for a structured provider failure, promoting a
     /// recognised context overflow to [`TinyAgentsError::ContextOverflow`].

crates/tinyagents-harness/src/lib.rs

@@ -73,6 +73,8 @@ pub mod handoff;
 pub mod host;
 pub mod ids;
 pub mod limits;
+#[cfg(feature = "media")]
+pub mod media;
 pub mod middleware;
 pub mod model_registry;
 #[cfg(feature = "multimodal")]
@@ -98,6 +100,8 @@ pub mod tool;
 pub mod tools;
 pub mod workspace;
 
+#[cfg(feature = "media")]
+pub use tinyinference_image;
 /// Re-exported vendor crates. Downstream consumers should reach these
 /// dependencies' types through these re-exports (e.g.
 /// `tinyagents_harness::tinyinference_llm::ChatMessage`) rather than adding
@@ -106,6 +110,8 @@ pub mod workspace;
 /// independent dependency would produce a duplicate, incompatible copy of
 /// the same types.
 pub use tinyinference_llm;
+#[cfg(feature = "media")]
+pub use tinyinference_video;
 pub use tinytools;
 pub use tinytools_agent;