okooooooo/EasyCLIProxyAPI
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(agents): sync Kimi and Grok context windows with CPA core

Browse Source
This commit is contained in:
lzt404 committed 2026-08-15 15:51:56 +08:00
1 parent b312b904a2
commit 51ea7fd69f
11 files changed
+218 -37
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src-tauri/src/tests/agent_configuration.rs
+73 -2
View File
@@ -22,6 +22,10 @@ fn claude_agent_config_preserves_existing_fields() {
assert_eq!(value["env"]["ANTHROPIC_BASE_URL"], "http://127.0.0.1:8317");
assert_eq!(value["env"]["ANTHROPIC_AUTH_TOKEN"], DEFAULT_API_KEY);
assert_eq!(value["env"]["ANTHROPIC_MODEL"], "claude-test");
assert_eq!(
value["env"][CLAUDE_CODE_ENABLE_GATEWAY_MODEL_DISCOVERY_ENV],
"1"
);
assert_eq!(value["env"][CLAUDE_CODE_MAX_CONTEXT_TOKENS_ENV], "200000");
assert_eq!(value["env"][CLAUDE_AUTOCOMPACT_PCT_OVERRIDE_ENV], "90");
assert!(value["env"].get(DISABLE_AUTO_COMPACT_ENV).is_none());
@@ -44,6 +48,7 @@ fn claude_code_inspection_normalizes_legacy_1m_suffix() {
"env": {
"ANTHROPIC_BASE_URL": "http://127.0.0.1:8317",
"ANTHROPIC_AUTH_TOKEN": "test-key",
"CLAUDE_CODE_ENABLE_GATEWAY_MODEL_DISCOVERY": "1",
"ANTHROPIC_MODEL": "deepseek-v4-pro[1m]",
"ANTHROPIC_DEFAULT_OPUS_MODEL": "deepseek-v4-pro[1m]",
"ANTHROPIC_DEFAULT_SONNET_MODEL": "deepseek-v4-pro[1m]",
@@ -71,6 +76,30 @@ fn claude_code_inspection_normalizes_legacy_1m_suffix() {
fs::remove_dir_all(directory).unwrap();
}
#[test]
fn claude_code_requires_gateway_model_discovery_to_be_enabled() {
let directory = agent_test_home("claude-code-gateway-model-discovery");
let path = directory.join("settings.json");
fs::write(
&path,
r#"{
"env": {
"ANTHROPIC_BASE_URL": "http://127.0.0.1:8317",
"ANTHROPIC_AUTH_TOKEN": "test-key",
"ANTHROPIC_MODEL": "gpt-test",
"CLAUDE_CODE_ENABLE_GATEWAY_MODEL_DISCOVERY": "0"
}
}"#,
)
.unwrap();
let (configured, model) = inspect_claude_agent_config(&path, 8317, "test-key").unwrap();
assert!(!configured);
assert_eq!(model.as_deref(), Some("gpt-test"));
fs::remove_dir_all(directory).unwrap();
}
#[test]
fn claude_mapping_legacy_json_defaults_1m_preferences_off() {
let mappings: ClaudeDesktopModelMappings =
@@ -762,7 +791,9 @@ fn kimi_code_config_preserves_existing_tables_and_uses_openai_provider() {
let home = agent_test_home("kimi-code-agent-config");
let path = home.join(".kimi-code/config.toml");
fs::create_dir_all(path.parent().unwrap()).unwrap();
let models = test_agent_models(&["gpt-test", "deepseek-test"]);
let mut models = test_agent_models(&["gpt-test", "deepseek-test"]);
models[0].context_window = Some(1_048_576);
models[1].context_window = Some(131_072);
let rendered = build_kimi_code_agent_config(
Some(
r#"theme = "dark"
@@ -806,6 +837,14 @@ model = "model"
.as_str(),
Some("tool_use")
);
assert_eq!(
value["models"]["cpa-gui/gpt-test"]["max_context_size"].as_integer(),
Some(1_048_576)
);
assert_eq!(
value["models"]["cpa-gui/deepseek-test"]["max_context_size"].as_integer(),
Some(131_072)
);
assert_eq!(
inspect_kimi_code_agent_config(&path, 8317, DEFAULT_API_KEY).unwrap(),
(true, Some("gpt-test".to_string()))
@@ -818,7 +857,9 @@ fn grok_build_config_preserves_existing_models_and_uses_chat_completions() {
let home = agent_test_home("grok-build-agent-config");
let path = home.join(".grok/config.toml");
fs::create_dir_all(path.parent().unwrap()).unwrap();
let models = test_agent_models(&["gpt-test", "deepseek-test"]);
let mut models = test_agent_models(&["gpt-test", "deepseek-test"]);
models[0].context_window = Some(1_048_576);
models[1].context_window = Some(131_072);
let rendered = build_grok_build_agent_config(
Some(
r#"theme = "dark"
@@ -858,6 +899,14 @@ api_key = "other-key"
value["model"]["cpa-gui/gpt-test"]["api_backend"].as_str(),
Some("chat_completions")
);
assert_eq!(
value["model"]["cpa-gui/gpt-test"]["context_window"].as_integer(),
Some(1_048_576)
);
assert_eq!(
value["model"]["cpa-gui/deepseek-test"]["context_window"].as_integer(),
Some(131_072)
);
assert_eq!(
inspect_grok_build_agent_config(&path, 8317, DEFAULT_API_KEY).unwrap(),
(true, Some("gpt-test".to_string()))
@@ -1077,6 +1126,28 @@ fn agent_model_list_parser_exposes_aliases_as_selectable_model_ids() {
);
}
#[test]
fn kimi_grok_and_zcode_use_cpa_runtime_context_windows() {
for client in [
AgentClient::KimiCode,
AgentClient::GrokBuild,
AgentClient::ZCode,
] {
assert!(agent_uses_cpa_runtime_context_windows(client));
}
for client in [
AgentClient::ClaudeCode,
AgentClient::ClaudeDesktop,
AgentClient::Codex,
AgentClient::OpenCode,
AgentClient::OpenClaw,
AgentClient::Hermes,
AgentClient::DeepSeekHarness,
] {
assert!(!agent_uses_cpa_runtime_context_windows(client));
}
}
#[test]
fn claude_model_lists_mark_yaml_aliases_when_api_returns_only_model_ids() {
let mut models = test_agent_models(&[
src-tauri/src/tests/agent_state.rs
+43 -1
View File
@@ -453,6 +453,7 @@ fn zcode_legacy_single_file_state_upgrades_and_restores_cli_config() {
version: AGENT_APPLIED_STATE_VERSION,
client: AgentClient::ZCode.id().to_string(),
model: "gpt-one".to_string(),
configuration_revision: 0,
claude_desktop_model_mappings: None,
backup_files: vec![AgentAppliedBackupFile {
path: paths[0].clone(),
@@ -489,9 +490,37 @@ fn zcode_legacy_single_file_state_upgrades_and_restores_cli_config() {
fs::remove_dir_all(home).unwrap();
}
#[test]
fn kimi_and_grok_legacy_states_require_context_catalog_resynchronization() {
for client in [AgentClient::KimiCode, AgentClient::GrokBuild] {
let home = agent_test_home(&format!("{}-context-revision", client.id()));
let paths = agent_config_paths(client, &home);
let state_path = agent_state_path(&paths).unwrap();
fs::create_dir_all(state_path.parent().unwrap()).unwrap();
let mut state = AgentAppliedState {
version: AGENT_APPLIED_STATE_VERSION,
client: client.id().to_string(),
model: "gpt-test".to_string(),
configuration_revision: 0,
claude_desktop_model_mappings: None,
backup_files: Vec::new(),
updated_at_unix: 1,
};
write_agent_applied_state(&state_path, &state).unwrap();
assert!(!agent_configuration_is_synchronized(client, &home, true));
state.configuration_revision = AGENT_CONFIGURATION_REVISION;
write_agent_applied_state(&state_path, &state).unwrap();
assert!(agent_configuration_is_synchronized(client, &home, true));
assert!(!agent_configuration_is_synchronized(client, &home, false));
fs::remove_dir_all(home).unwrap();
}
}
#[test]
fn session_merge_preserves_runtime_fields_for_other_agent_formats() {
let claude_original = r#"{"env":{"KEEP_ENV":"original"},"keep":"claude"}"#;
let claude_original = r#"{"env":{"KEEP_ENV":"original","CLAUDE_CODE_ENABLE_GATEWAY_MODEL_DISCOVERY":"0"},"keep":"claude"}"#;
let claude_managed = build_claude_agent_config(
Some(claude_original),
"http://127.0.0.1:8317",
@@ -516,6 +545,10 @@ fn session_merge_preserves_runtime_fields_for_other_agent_formats() {
assert_eq!(claude["runtimeAdded"], true);
assert_eq!(claude["env"]["RUNTIME_ENV"], "keep");
assert_eq!(claude["env"]["KEEP_ENV"], "original");
assert_eq!(
claude["env"][CLAUDE_CODE_ENABLE_GATEWAY_MODEL_DISCOVERY_ENV],
"0"
);
assert!(claude["env"].get("ANTHROPIC_BASE_URL").is_none());
assert!(claude.get("model").is_none());
@@ -735,6 +768,7 @@ fn legacy_codex_single_file_state_remains_restorable() {
version: AGENT_APPLIED_STATE_VERSION,
client: AgentClient::Codex.id().to_string(),
model: "gpt-test".to_string(),
configuration_revision: AGENT_CONFIGURATION_REVISION,
claude_desktop_model_mappings: None,
backup_files: vec![AgentAppliedBackupFile {
path: path.clone(),
@@ -799,6 +833,7 @@ fn claude_desktop_version_three_state_can_be_closed_safely() {
version: 3,
client: AgentClient::ClaudeDesktop.id().to_string(),
model: "gpt-test".to_string(),
configuration_revision: 0,
claude_desktop_model_mappings: None,
backup_files: Vec::new(),
updated_at_unix: 1,
@@ -930,6 +965,7 @@ fn applied_state_rejects_backup_paths_outside_the_managed_config_set() {
version: AGENT_APPLIED_STATE_VERSION,
client: AgentClient::OpenCode.id().to_string(),
model: "gpt-test".to_string(),
configuration_revision: AGENT_CONFIGURATION_REVISION,
claude_desktop_model_mappings: None,
backup_files: vec![AgentAppliedBackupFile {
path: path.clone(),
@@ -966,6 +1002,7 @@ fn missing_session_backup_is_detected_before_any_file_is_restored() {
version: AGENT_APPLIED_STATE_VERSION,
client: AgentClient::Codex.id().to_string(),
model: "gpt-test".to_string(),
configuration_revision: AGENT_CONFIGURATION_REVISION,
claude_desktop_model_mappings: None,
backup_files: vec![
AgentAppliedBackupFile {
@@ -1003,6 +1040,7 @@ fn version_three_cleanup_covers_every_non_desktop_agent_format() {
version: 3,
client: client.id().to_string(),
model: "gpt-test".to_string(),
configuration_revision: 0,
claude_desktop_model_mappings: None,
backup_files: Vec::new(),
updated_at_unix: 1,
@@ -1017,6 +1055,7 @@ fn version_three_cleanup_covers_every_non_desktop_agent_format() {
"ANTHROPIC_BASE_URL": "http://127.0.0.1:8317",
"ANTHROPIC_API_KEY": "secret",
"ANTHROPIC_AUTH_TOKEN": "secret",
"CLAUDE_CODE_ENABLE_GATEWAY_MODEL_DISCOVERY": "1",
"ANTHROPIC_MODEL": "gpt-test",
"ANTHROPIC_DEFAULT_HAIKU_MODEL": "gpt-test",
"KEEP_ENV": "yes"
@@ -1039,6 +1078,9 @@ fn version_three_cleanup_covers_every_non_desktop_agent_format() {
serde_json::from_str(&fs::read_to_string(&claude_path).unwrap()).unwrap();
assert!(claude.get("model").is_none());
assert!(claude["env"].get("ANTHROPIC_BASE_URL").is_none());
assert!(claude["env"]
.get(CLAUDE_CODE_ENABLE_GATEWAY_MODEL_DISCOVERY_ENV)
.is_none());
assert_eq!(claude["env"]["KEEP_ENV"], "yes");
assert_eq!(claude["keep"], "claude");