mirror of
https://github.com/fabro-sh/fabro.git
synced 2026-10-07 03:00:29 +00:00
This PR adds support for mapping the unified `reasoning_effort` field to
Anthropic's `output_config.effort` API parameter, which is the
recommended way to control thinking depth for Claude Opus 4.6 and Sonnet
4.6 models. Previously, the Anthropic provider silently dropped
`reasoning_effort` from requests, while the OpenAI provider already
correctly mapped it to `reasoning: { effort }`. This inconsistency meant
that workflow nodes setting `reasoning_effort: "high"` (the default) had
no effect when routing through Anthropic.
The change adds an `output_config: Option<serde_json::Value>` field to
the `ApiRequest` struct with `skip_serializing_if = "Option::is_none"`
to ensure it's omitted when not set, then populates it in
`build_api_request` by transforming `request.reasoning_effort` into
`{"effort": <value>}` — mirroring the pattern used in the OpenAI
provider. All existing `ApiRequest` constructions in tests are updated
to include `output_config: None`.
The PR also takes the opportunity to refactor the test module by
extracting a `make_base_request()` helper, which reduces boilerplate in
`build_api_request_omits_whitespace_only_system_prompt` and
`make_request_with_format` and makes the two new tests
(`build_api_request_maps_reasoning_effort_to_output_config` and
`build_api_request_omits_output_config_when_no_reasoning_effort`) easy
to read. All existing provider tests continue to pass and no new Clippy
warnings are introduced.
### Fabro Details
<details>
<summary>Ran 9 stages in 10m 28s for $1.48</summary>
| Stage | Duration | Cost | Retries |
|---|---|---|---|
| start | 0s | – | 0 |
| toolchain | 0s | – | 0 |
| preflight_compile | 1m 8s | – | 0 |
| preflight_lint | 12s | – | 0 |
| implement | 2m 13s | $0.49 | 0 |
| simplify_opus | 4m 45s | $0.99 | 0 |
| simplify_gpt | 0s | – | 0 |
| verify | 1m 40s | – | 0 |
| fmt | 1s | – | 0 |
| **Total** | **10m 28s** | **$1.48** | **0** |
</details>
<details>
<summary>Ran <code>ImplementAndSimplify.fabro</code> (12 nodes and 15
edges)</summary>
```dot
digraph ImplementAndSimplify {
graph [
goal="Implement and simplify",
model_stylesheet="
* { backend: api; model: claude-opus-4-6;}
"
]
rankdir=LR
start [shape=Mdiamond, label="Start"]
exit [shape=Msquare, label="Exit"]
toolchain [label="Toolchain", shape=parallelogram, script="command -v cargo >/dev/null || { curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y && sudo ln -sf $HOME/.cargo/bin/* /usr/local/bin/; }; cargo --version 2>&1", max_retries=0]
preflight_compile [label="Preflight Compile", shape=parallelogram, script="cargo check -q --workspace 2>&1", max_retries=0]
preflight_lint [label="Preflight Lint", shape=parallelogram, script="cargo clippy -q --workspace -- -D warnings 2>&1", max_retries=0]
fix_lints [label="Fix Lints", prompt="The preflight lint step failed. Read the build output from context and fix all clippy lint warnings.", max_visits=3]
implement [label="Implement", prompt="Read the plan file referenced in the goal and implement every step. Make all the code changes described in the plan. Use red/green TDD."]
simplify_opus [label="Simplify (Opus)", prompt="@prompts/simplify.md"]
simplify_gpt [label="Simplify (GPT-54)", prompt="@prompts/simplify.md", model="gpt-54"]
verify [label="Verify", shape=parallelogram, script="cargo clippy -q --workspace -- -D warnings 2>&1 && cargo nextest run --cargo-quiet --workspace --status-level fail 2>&1", goal_gate=true, retry_target="fixup"]
fixup [label="Fixup", prompt="The verify step failed. Read the build output from context and fix all clippy lint warnings and test failures.", max_visits=3]
fmt [label="Format", shape=parallelogram, script="cargo fmt --all 2>&1", max_retries=0]
start -> toolchain
toolchain -> preflight_compile [condition="outcome=success"]
toolchain -> exit
preflight_compile -> preflight_lint [condition="outcome=success"]
preflight_compile -> exit
preflight_lint -> implement [condition="outcome=success"]
preflight_lint -> fix_lints
fix_lints -> preflight_lint
implement -> simplify_opus -> simplify_gpt -> verify
verify -> fmt [condition="outcome=success"]
verify -> fixup
fixup -> verify
fmt -> exit
}
```
</details>
⚒️ Generated with [Fabro](https://fabro.sh)
---------
Co-authored-by: Fabro <noreply@fabro.sh>
|
||
|---|---|---|
| .. | ||
| src | ||
| tests | ||
| Cargo.toml | ||
| README.md | ||
unified-llm
A unified async Rust client library for multiple LLM providers. Write your LLM integration code once and switch between Anthropic, OpenAI, and Google Gemini without changing your application logic.
Key concepts
- Client -- Routes requests to registered provider adapters. Can be created explicitly or auto-configured from environment variables.
- ProviderAdapter -- The trait every provider implements (
completeandstream). Built-in adapters:AnthropicAdapter,OpenAiAdapter,GeminiAdapter,OpenAiCompatibleAdapter. - Middleware -- Intercepts requests/responses for logging, caching, or transformation. Supports both blocking and streaming paths.
- generate() -- High-level function that wraps
Client.complete()with automatic tool execution loops, retries, timeouts, and cancellation. - Tool -- Active tools (with an execute handler) run automatically in the tool loop. Passive tools (no handler) surface tool calls back to the caller.
- Model catalog -- Built-in metadata for common models. Advisory only; unknown model strings pass through.
Providers
| Provider | Adapter | API | Env var |
|---|---|---|---|
| Anthropic | AnthropicAdapter |
Messages API | ANTHROPIC_API_KEY |
| OpenAI | OpenAiAdapter |
Responses API | OPENAI_API_KEY |
| Google Gemini | GeminiAdapter |
generateContent | GEMINI_API_KEY or GOOGLE_API_KEY |
| OpenAI-compatible | OpenAiCompatibleAdapter |
Chat Completions | (custom) |
All adapters support streaming, tool calling, structured output (response_format), and provider-specific options via provider_options.
Usage
Auto-configure from environment
use unified_llm::client::Client;
use unified_llm::types::{Message, Request};
let client = Client::from_env().await?;
let request = Request {
model: "claude-sonnet-4-5".to_string(),
messages: vec![Message::user("What is the capital of France?")],
provider: None,
tools: None,
tool_choice: None,
response_format: None,
temperature: Some(0.0),
top_p: None,
max_tokens: Some(100),
stop_sequences: None,
reasoning_effort: None,
metadata: None,
provider_options: None,
};
let response = client.complete(&request).await?;
println!("{}", response.text());
High-level generate()
use unified_llm::generate::{generate, GenerateParams};
let result = generate(
GenerateParams::new("claude-sonnet-4-5")
.prompt("Explain monads in one sentence")
.system("You are a concise programming tutor.")
.max_tokens(200)
).await?;
println!("{}", result.text());
Tool calling
use unified_llm::generate::{generate, GenerateParams};
use unified_llm::tools::Tool;
use std::sync::Arc;
let weather_tool = Tool::active(
"get_weather",
"Get the current weather for a city",
serde_json::json!({
"type": "object",
"properties": {
"city": {"type": "string", "description": "City name"}
},
"required": ["city"]
}),
|args, _ctx| async move {
let city = args["city"].as_str().unwrap_or("unknown");
Ok(serde_json::json!({"temp": "72F", "city": city}))
},
);
let result = generate(
GenerateParams::new("claude-sonnet-4-5")
.prompt("What's the weather in San Francisco?")
.tools(vec![weather_tool])
.max_tool_rounds(3)
).await?;
Streaming
use unified_llm::client::Client;
use unified_llm::types::{Message, Request, StreamEvent};
use futures::StreamExt;
let client = Client::from_env().await?;
let request = Request {
model: "claude-sonnet-4-5".to_string(),
messages: vec![Message::user("Tell me a joke")],
// ...other fields set to None/defaults
# provider: None, tools: None, tool_choice: None,
# response_format: None, temperature: None, top_p: None,
# max_tokens: None, stop_sequences: None, reasoning_effort: None,
# metadata: None, provider_options: None,
};
let mut stream = client.stream(&request).await?;
while let Some(event) = stream.next().await {
match event? {
StreamEvent::TextDelta { delta, .. } => print!("{delta}"),
StreamEvent::Finish { response, .. } => {
println!("\nTokens used: {}", response.usage.total_tokens);
}
_ => {}
}
}
Middleware
use unified_llm::middleware::{Middleware, NextFn, NextStreamFn};
use unified_llm::types::{Request, Response};
use unified_llm::provider::StreamEventStream;
use unified_llm::error::SdkError;
struct LoggingMiddleware;
#[async_trait::async_trait]
impl Middleware for LoggingMiddleware {
async fn handle_complete(
&self,
request: Request,
next: NextFn,
) -> Result<Response, SdkError> {
eprintln!("Request to model: {}", request.model);
let response = next(request).await?;
eprintln!("Response tokens: {}", response.usage.total_tokens);
Ok(response)
}
async fn handle_stream(
&self,
request: Request,
next: NextStreamFn,
) -> Result<StreamEventStream, SdkError> {
next(request).await
}
}
OpenAI-compatible providers
use unified_llm::providers::OpenAiCompatibleAdapter;
use std::sync::Arc;
let adapter = OpenAiCompatibleAdapter::new("your-api-key", "https://api.groq.com/openai/v1")
.with_name("groq");
Model catalog
use unified_llm::catalog::{get_model_info, list_models, get_latest_model};
let info = get_model_info("claude-opus-4-6");
let anthropic_models = list_models(Some("anthropic"));
let best_reasoner = get_latest_model("anthropic", Some("reasoning"));
Key types
| Type | Description |
|---|---|
Request |
Unified request with model, messages, tools, temperature, etc. |
Response |
Unified response with message, finish reason, usage, rate limit info |
Message |
A message with role, content parts, and optional tool call ID |
ContentPart |
Text, Image, Audio, Document, ToolCall, ToolResult, Thinking |
StreamEvent |
Events for streaming: TextDelta, ToolCallStart/Delta/End, Finish, etc. |
SdkError |
Typed errors with retryability, status codes, and provider error kinds |
GenerateParams |
Builder for the high-level generate() function |
GenerateResult |
Result containing response, tool results, total usage, and step history |
ToolDefinition |
Tool name, description, and JSON Schema parameters |
ToolChoice |
Auto, None, Required, or Named tool selection |
Usage |
Token counts including input, output, reasoning, and cache tokens |
RetryPolicy |
Configurable retry with exponential backoff, jitter, and max delay |
ModelInfo |
Metadata about a model (context window, capabilities, costs) |
Error handling
SdkError provides structured error variants with built-in retryability classification:
- Retryable:
RateLimit,Server,Network,Stream,RequestTimeout - Non-retryable:
Authentication,AccessDenied,InvalidRequest,ContextLength,Configuration
The retry() function and generate() respect Retry-After headers and use exponential backoff with jitter.
Provider-specific options
Pass provider-specific parameters via provider_options without losing portability:
use unified_llm::types::Request;
let request = Request {
provider_options: Some(serde_json::json!({
"anthropic": {
"thinking": {"type": "enabled", "budget_tokens": 10000},
"auto_cache": true
},
"openai": {
"store": true,
"previous_response_id": "resp_abc123"
},
"gemini": {
"safetySettings": [
{"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"}
]
}
})),
// ...other fields
# model: String::new(), messages: vec![], provider: None, tools: None,
# tool_choice: None, response_format: None, temperature: None,
# top_p: None, max_tokens: None, stop_sequences: None,
# reasoning_effort: None, metadata: None,
};