mirror of
https://github.com/fabro-sh/fabro.git
synced 2026-08-28 05:27:41 +00:00
5d8befa6ac
Add the missing with_replacement for ModelControls so progenitor reuses fabro_model::ModelControls instead of generating a dead parallel DTO, re-export it from fabro_api::types, and assert type identity in the round-trip test. Drop #[serde(default)] from Model.controls and ModelControls.reasoning_effort: the OpenAPI spec marks both required, matching the strict deserialization of the sibling features/costs fields. Update CLI stub payloads to include the now-required field. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
310 lines
11 KiB
Rust
310 lines
11 KiB
Rust
#![expect(
|
|
clippy::disallowed_methods,
|
|
reason = "integration tests stage fixtures with sync std::fs; test infrastructure, not Tokio-hot path"
|
|
)]
|
|
|
|
use fabro_test::{fabro_snapshot, test_context};
|
|
use httpmock::MockServer;
|
|
|
|
fn successful_stdout(mut cmd: assert_cmd::Command) -> String {
|
|
let assert = cmd.assert().success().stderr("");
|
|
String::from_utf8(assert.get_output().stdout.clone())
|
|
.expect("model list stdout should be UTF-8")
|
|
}
|
|
|
|
fn assert_model_list_table(stdout: &str) {
|
|
let mut lines = stdout.lines();
|
|
let header = lines.next().expect("model list should render a header");
|
|
assert!(header.contains("MODEL"), "missing MODEL column: {header}");
|
|
assert!(
|
|
header.contains("PROVIDER"),
|
|
"missing PROVIDER column: {header}"
|
|
);
|
|
assert!(
|
|
header.contains("CONTEXT"),
|
|
"missing CONTEXT column: {header}"
|
|
);
|
|
assert!(
|
|
lines.any(|line| !line.trim().is_empty()),
|
|
"model list should render at least one model row"
|
|
);
|
|
}
|
|
|
|
#[test]
|
|
fn help() {
|
|
let context = test_context!();
|
|
let mut cmd = context.model();
|
|
cmd.arg("--help");
|
|
fabro_snapshot!(context.filters(), cmd, @"
|
|
success: true
|
|
exit_code: 0
|
|
----- stdout -----
|
|
List and test LLM models
|
|
|
|
Usage: fabro model [OPTIONS] [COMMAND]
|
|
|
|
Commands:
|
|
list List available models
|
|
test Test model availability by sending a simple prompt
|
|
help Print this message or the help of the given subcommand(s)
|
|
|
|
Options:
|
|
--json Output as JSON [env: FABRO_JSON=]
|
|
--debug Enable DEBUG-level logging (default is INFO) [env: FABRO_DEBUG=]
|
|
--no-upgrade-check Disable automatic upgrade check [env: FABRO_NO_UPGRADE_CHECK=true]
|
|
--quiet Suppress non-essential output [env: FABRO_QUIET=]
|
|
--verbose Enable verbose output [env: FABRO_VERBOSE=]
|
|
-h, --help Print help
|
|
----- stderr -----
|
|
");
|
|
}
|
|
|
|
#[test]
|
|
fn bare() {
|
|
let context = test_context!();
|
|
let stdout = successful_stdout(context.model());
|
|
assert_model_list_table(&stdout);
|
|
}
|
|
|
|
#[test]
|
|
fn list() {
|
|
let context = test_context!();
|
|
let mut cmd = context.model();
|
|
cmd.arg("list");
|
|
let stdout = successful_stdout(cmd);
|
|
assert_model_list_table(&stdout);
|
|
}
|
|
|
|
#[test]
|
|
fn list_with_filters_renders_server_models_table() {
|
|
let context = test_context!();
|
|
let server = MockServer::start();
|
|
let mock = server.mock(|when, then| {
|
|
when.method("GET")
|
|
.path("/api/v1/models")
|
|
.query_param("page[limit]", "100")
|
|
.query_param("page[offset]", "0")
|
|
.query_param("provider", "fixture")
|
|
.query_param("query", "stable");
|
|
then.status(200)
|
|
.header("Content-Type", "application/json")
|
|
.body(
|
|
serde_json::json!({
|
|
"data": [
|
|
{
|
|
"id": "fixture-128k",
|
|
"display_name": "Fixture 128k",
|
|
"provider": "fixture",
|
|
"family": "test",
|
|
"aliases": ["stable", "fixture-alias"],
|
|
"limits": {
|
|
"context_window": 128_000,
|
|
"max_output": 4096
|
|
},
|
|
"training": null,
|
|
"knowledge_cutoff": null,
|
|
"features": {
|
|
"tools": true,
|
|
"vision": false,
|
|
"reasoning": false
|
|
},
|
|
"controls": {
|
|
"reasoning_effort": []
|
|
},
|
|
"costs": {
|
|
"input_cost_per_mtok": 1.2,
|
|
"output_cost_per_mtok": 3.4,
|
|
"cache_input_cost_per_mtok": null
|
|
},
|
|
"estimated_output_tps": 88.0,
|
|
"default": false,
|
|
"configured": false
|
|
},
|
|
{
|
|
"id": "fixture-1m",
|
|
"display_name": "Fixture 1m",
|
|
"provider": "fixture",
|
|
"family": "test",
|
|
"aliases": [],
|
|
"limits": {
|
|
"context_window": 1_000_000,
|
|
"max_output": 4096
|
|
},
|
|
"training": null,
|
|
"knowledge_cutoff": null,
|
|
"features": {
|
|
"tools": false,
|
|
"vision": true,
|
|
"reasoning": true
|
|
},
|
|
"controls": {
|
|
"reasoning_effort": []
|
|
},
|
|
"costs": {
|
|
"input_cost_per_mtok": null,
|
|
"output_cost_per_mtok": null,
|
|
"cache_input_cost_per_mtok": null
|
|
},
|
|
"estimated_output_tps": null,
|
|
"default": false,
|
|
"configured": false
|
|
}
|
|
],
|
|
"meta": { "has_more": false }
|
|
})
|
|
.to_string(),
|
|
);
|
|
});
|
|
context.set_http_target(&server.base_url());
|
|
|
|
let mut cmd = context.model();
|
|
cmd.args(["list", "--provider", "fixture", "--query", "stable"]);
|
|
fabro_snapshot!(context.filters(), cmd, @"
|
|
success: true
|
|
exit_code: 0
|
|
----- stdout -----
|
|
MODEL PROVIDER ALIASES CONTEXT COST SPEED
|
|
fixture-128k fixture stable, fixture-alias 128k $1.2 / $3.4 88 tok/s
|
|
fixture-1m fixture 1m - / - -
|
|
----- stderr -----
|
|
");
|
|
mock.assert();
|
|
}
|
|
|
|
#[test]
|
|
fn list_unknown_provider_returns_empty() {
|
|
let context = test_context!();
|
|
let mut cmd = context.model();
|
|
cmd.args(["list", "--provider", "not-a-provider"]);
|
|
fabro_snapshot!(context.filters(), cmd, @"
|
|
success: true
|
|
exit_code: 0
|
|
----- stdout -----
|
|
MODEL PROVIDER ALIASES CONTEXT COST SPEED
|
|
----- stderr -----
|
|
");
|
|
}
|
|
|
|
#[test]
|
|
fn list_uses_configured_server_target_without_server_flag() {
|
|
let context = test_context!();
|
|
let server = MockServer::start();
|
|
let mock = server.mock(|when, then| {
|
|
when.method("GET");
|
|
then.status(200)
|
|
.header("Content-Type", "application/json")
|
|
.body(
|
|
serde_json::json!({
|
|
"data": [{
|
|
"id": "remote-model",
|
|
"display_name": "Remote Model",
|
|
"provider": "openai",
|
|
"family": "test",
|
|
"aliases": ["remote"],
|
|
"limits": {
|
|
"context_window": 131_072,
|
|
"max_output": 4096
|
|
},
|
|
"training": null,
|
|
"knowledge_cutoff": null,
|
|
"features": {
|
|
"tools": true,
|
|
"vision": false,
|
|
"reasoning": false
|
|
},
|
|
"controls": {
|
|
"reasoning_effort": []
|
|
},
|
|
"costs": {
|
|
"input_cost_per_mtok": 1.0,
|
|
"output_cost_per_mtok": 2.0,
|
|
"cache_input_cost_per_mtok": null
|
|
},
|
|
"estimated_output_tps": 42.0,
|
|
"default": false,
|
|
"configured": false
|
|
}],
|
|
"meta": { "has_more": false }
|
|
})
|
|
.to_string(),
|
|
);
|
|
});
|
|
context.set_http_target(&server.base_url());
|
|
|
|
let mut cmd = context.model();
|
|
cmd.args(["list", "--json"]);
|
|
let output = cmd.assert().success().get_output().stdout.clone();
|
|
let models: serde_json::Value =
|
|
serde_json::from_slice(&output).expect("model list json should parse");
|
|
|
|
mock.assert();
|
|
assert_eq!(models.as_array().map(Vec::len), Some(1));
|
|
assert_eq!(models[0]["id"].as_str(), Some("remote-model"));
|
|
}
|
|
|
|
#[test]
|
|
fn list_uses_fabro_config_for_machine_settings() {
|
|
let context = test_context!();
|
|
let server = MockServer::start();
|
|
let mock = server.mock(|when, then| {
|
|
when.method("GET");
|
|
then.status(200)
|
|
.header("Content-Type", "application/json")
|
|
.body(
|
|
serde_json::json!({
|
|
"data": [{
|
|
"id": "remote-model",
|
|
"display_name": "Remote Model",
|
|
"provider": "openai",
|
|
"family": "test",
|
|
"aliases": ["remote"],
|
|
"limits": {
|
|
"context_window": 131_072,
|
|
"max_output": 4096
|
|
},
|
|
"training": null,
|
|
"knowledge_cutoff": null,
|
|
"features": {
|
|
"tools": true,
|
|
"vision": false,
|
|
"reasoning": false
|
|
},
|
|
"controls": {
|
|
"reasoning_effort": []
|
|
},
|
|
"costs": {
|
|
"input_cost_per_mtok": 1.0,
|
|
"output_cost_per_mtok": 2.0,
|
|
"cache_input_cost_per_mtok": null
|
|
},
|
|
"estimated_output_tps": 42.0,
|
|
"default": false,
|
|
"configured": false
|
|
}],
|
|
"meta": { "has_more": false }
|
|
})
|
|
.to_string(),
|
|
);
|
|
});
|
|
let config_dir = tempfile::tempdir().unwrap();
|
|
let config_path = config_dir.path().join("custom-settings.toml");
|
|
std::fs::write(
|
|
&config_path,
|
|
format!(
|
|
"_version = 1\n\n[cli.target]\ntype = \"http\"\nurl = \"{}/api/v1\"\n",
|
|
server.base_url()
|
|
),
|
|
)
|
|
.unwrap();
|
|
|
|
let mut cmd = context.model();
|
|
cmd.args(["list", "--json"]);
|
|
cmd.env("FABRO_CONFIG", &config_path);
|
|
let output = cmd.assert().success().get_output().stdout.clone();
|
|
let models: serde_json::Value =
|
|
serde_json::from_slice(&output).expect("model list json should parse");
|
|
|
|
mock.assert();
|
|
assert_eq!(models.as_array().map(Vec::len), Some(1));
|
|
assert_eq!(models[0]["id"].as_str(), Some("remote-model"));
|
|
}
|