refactor: consume focused inference crates

Co-authored-by: Medulla <medulla@tinyhumans.ai>

Author: senamakel

Cargo.lock

@@ -61,15 +61,15 @@ tinyagents-registry = { git = "https://github.com/tinyhumansai/tinyagents", pack
 # The code samples below build `Message` and provider types directly from
 # TinyInference, the message/model crate TinyAgents is built on. It is a
 # separate git dependency, not re-exported by the crates above.
-tinyinference-core = { git = "https://github.com/tinyhumansai/tinyinference", package = "tinyinference-core" }
+tinyinference-llm = { git = "https://github.com/tinyhumansai/tinyinference", package = "tinyinference-llm" }
 ```
 
 A minimal typed graph — a whole-state agent/tool loop (trimmed from
 [`examples/basic_graph.rs`](crates/tinyagents-integration-tests/examples/basic_graph.rs)):
 
 ```rust
 use tinyagents_graph::*;
-use tinyinference_core::message::Message;
+use tinyinference_llm::message::Message;
 
 #[derive(Clone, Debug)]
 struct AgentState {
@@ -113,8 +113,8 @@ A one-shot model call through the harness (`export OPENAI_API_KEY=...` then
 ```rust
 use std::sync::Arc;
 use tinyagents_harness::runtime::AgentHarness;
-use tinyinference_core::message::Message;
-use tinyinference_core::providers::openai::OpenAiModel;
+use tinyinference_llm::message::Message;
+use tinyinference_llm::providers::openai::OpenAiModel;
 
 let model = OpenAiModel::from_env()?;
 let mut harness: AgentHarness<()> = AgentHarness::new();

README.md

@@ -19,7 +19,7 @@ sha2 = "0.11"
 tinyagents-harness = { path = "../tinyagents-harness", version = "2.1.2", default-features = false }
 tinyagents-language = { path = "../tinyagents-language", version = "2.1.2" }
 tinyagents-tracing = { path = "../tinyagents-tracing", version = "2.1.2", default-features = false }
-tinyinference-core = { path = "../../vendor/tinyinference/crates/tinyinference-core", version = "0.2.1" }
+tinyinference-llm = { path = "../../vendor/tinyinference/crates/tinyinference-llm", version = "0.2.1" }
 tokio = { version = "1", default-features = false, features = ["sync", "time", "macros", "rt", "rt-multi-thread", "fs"] }
 
 [features]

crates/tinyagents-graph/Cargo.toml

@@ -262,7 +262,7 @@ mod tool_tests {
     use tinyagents_harness::ids::{RunId, ThreadId};
     use tinyagents_harness::store::{InMemoryStore, Store};
     use tinyagents_harness::tool::{Tool, ToolExecutionContext};
-    use tinyinference_core::tool::ToolCall;
+    use tinyinference_llm::tool::ToolCall;
 
     fn store() -> Arc<dyn Store> {
         Arc::new(InMemoryStore::default())

crates/tinyagents-graph/src/goals/test.rs

@@ -26,7 +26,7 @@ use tinyagents_harness::store::Store;
 use tinyagents_harness::tool::{
     Tool, ToolExecutionContext, ToolPolicy, ToolRegistry, ToolResult, ToolSideEffects,
 };
-use tinyinference_core::tool::{ToolCall, ToolSchema};
+use tinyinference_llm::tool::{ToolCall, ToolSchema};
 
 /// Which thread-goal control a [`GoalTool`] implements.
 #[derive(Clone, Copy, Debug, PartialEq, Eq, Hash)]

crates/tinyagents-graph/src/goals/tool.rs

@@ -7,7 +7,7 @@ use tinyagents_harness::cancel::CancellationToken;
 use tinyagents_harness::ids::TaskId;
 use tinyagents_harness::steering::SteeringHandle;
 use tinyagents_harness::tool::{Tool, ToolRegistry};
-use tinyinference_core::tool::ToolCall;
+use tinyinference_llm::tool::ToolCall;
 
 fn graph_spec(id: &str) -> OrchestrationTaskSpec {
     OrchestrationTaskSpec::new(

crates/tinyagents-graph/src/orchestration/test.rs

@@ -10,8 +10,8 @@ use crate::{Result, TinyAgentsError};
 use tinyagents_harness::ids::{GraphId, TaskId, new_call_id, next_seq};
 use tinyagents_harness::steering::{SteeringCommand, SteeringHandle};
 use tinyagents_harness::tool::{Tool, ToolRegistry, ToolResult};
-use tinyinference_core::message::Message;
-use tinyinference_core::tool::{ToolCall, ToolSchema};
+use tinyinference_llm::message::Message;
+use tinyinference_llm::tool::{ToolCall, ToolSchema};
 
 use super::store::{TaskStore, orchestration_not_found};
 use super::types::*;

crates/tinyagents-graph/src/orchestration/tool.rs

@@ -200,7 +200,7 @@ pub struct ChildRun {
     /// folds the delegated harness agent's [`UsageTotals`] here so it is visible
     /// on the parent [`GraphExecution`](crate::GraphExecution) rollup.
     #[serde(default)]
-    pub usage: tinyinference_core::usage::UsageTotals,
+    pub usage: tinyinference_llm::usage::UsageTotals,
 }
 
 /// A thread-safe collector the executor hands to node contexts so that a

crates/tinyagents-graph/src/recursion/types.rs

@@ -19,11 +19,11 @@ use crate::builder::GraphBuilder;
 use tinyagents_harness::runtime::AgentHarness;
 use tinyagents_harness::subagent::SubAgent;
 use tinyagents_harness::testkit::{EventRecorder, FakeTool, ScriptedModel};
-use tinyinference_core::message::AssistantMessage;
-use tinyinference_core::model::ModelResponse;
-use tinyinference_core::providers::MockModel;
-use tinyinference_core::tool::ToolCall;
-use tinyinference_core::usage::Usage;
+use tinyinference_llm::message::AssistantMessage;
+use tinyinference_llm::model::ModelResponse;
+use tinyinference_llm::providers::MockModel;
+use tinyinference_llm::tool::ToolCall;
+use tinyinference_llm::usage::Usage;
 
 #[derive(Default)]
 struct TestAgentRegistry {

crates/tinyagents-graph/src/subagent_node/test.rs

@@ -18,7 +18,7 @@ use serde_json::Value;
 use crate::Result;
 use tinyagents_harness::events::EventSink;
 use tinyagents_harness::retry::RetryPolicy;
-use tinyinference_core::usage::UsageTotals;
+use tinyinference_llm::usage::UsageTotals;
 
 /// A harness agent invocable by name from a graph node.
 ///