mirror of
https://github.com/farion1231/cc-switch.git
synced 2026-04-28 21:53:06 +08:00
Compare commits
4 Commits
codex/upst
...
codex/issu
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
5bccb4f65b | ||
|
|
b4f0f7cec3 | ||
|
|
ad89c4719a | ||
|
|
b11e36963f |
@@ -206,6 +206,10 @@ fn normalize_openai_system_messages(messages: &mut Vec<Value>) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
let mut parts = Vec::new();
|
let mut parts = Vec::new();
|
||||||
|
let mut inherited_cache_control: Option<Value> = None;
|
||||||
|
let mut cache_control_conflict = false;
|
||||||
|
let mut saw_cache_control = false;
|
||||||
|
let mut saw_missing_cache_control = false;
|
||||||
messages.retain(|message| {
|
messages.retain(|message| {
|
||||||
if message.get("role").and_then(|value| value.as_str()) != Some("system") {
|
if message.get("role").and_then(|value| value.as_str()) != Some("system") {
|
||||||
return true;
|
return true;
|
||||||
@@ -226,11 +230,28 @@ fn normalize_openai_system_messages(messages: &mut Vec<Value>) {
|
|||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if let Some(cache_control) = message.get("cache_control") {
|
||||||
|
saw_cache_control = true;
|
||||||
|
match &inherited_cache_control {
|
||||||
|
None => inherited_cache_control = Some(cache_control.clone()),
|
||||||
|
Some(existing) if existing == cache_control => {}
|
||||||
|
Some(_) => cache_control_conflict = true,
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
saw_missing_cache_control = true;
|
||||||
|
}
|
||||||
|
|
||||||
false
|
false
|
||||||
});
|
});
|
||||||
|
|
||||||
if !parts.is_empty() {
|
if !parts.is_empty() {
|
||||||
messages.insert(0, json!({"role": "system", "content": parts.join("\n")}));
|
let mut merged = json!({"role": "system", "content": parts.join("\n")});
|
||||||
|
if !(cache_control_conflict || (saw_cache_control && saw_missing_cache_control)) {
|
||||||
|
if let Some(cache_control) = inherited_cache_control {
|
||||||
|
merged["cache_control"] = cache_control;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
messages.insert(0, merged);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -613,18 +634,15 @@ mod tests {
|
|||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_anthropic_to_openai_normalizes_fragmented_system_messages() {
|
fn test_anthropic_to_openai_preserves_matching_system_cache_control_when_merging() {
|
||||||
let input = json!({
|
let input = json!({
|
||||||
"model": "claude-3-sonnet",
|
"model": "claude-3-sonnet",
|
||||||
"max_tokens": 1024,
|
"max_tokens": 1024,
|
||||||
"system": [
|
"system": [
|
||||||
{"type": "text", "text": "You are Claude Code."},
|
{"type": "text", "text": "You are Claude Code.", "cache_control": {"type": "ephemeral"}},
|
||||||
{"type": "text", "text": "Be concise."}
|
{"type": "text", "text": "Be concise.", "cache_control": {"type": "ephemeral"}}
|
||||||
],
|
],
|
||||||
"messages": [
|
"messages": [{"role": "user", "content": "Hello"}]
|
||||||
{"role": "system", "content": "Follow repo conventions."},
|
|
||||||
{"role": "user", "content": "Hello"}
|
|
||||||
]
|
|
||||||
});
|
});
|
||||||
|
|
||||||
let result = anthropic_to_openai(input, None).unwrap();
|
let result = anthropic_to_openai(input, None).unwrap();
|
||||||
@@ -632,11 +650,54 @@ mod tests {
|
|||||||
assert_eq!(result["messages"][0]["role"], "system");
|
assert_eq!(result["messages"][0]["role"], "system");
|
||||||
assert_eq!(
|
assert_eq!(
|
||||||
result["messages"][0]["content"],
|
result["messages"][0]["content"],
|
||||||
"You are Claude Code.\nBe concise.\nFollow repo conventions."
|
"You are Claude Code.\nBe concise."
|
||||||
);
|
);
|
||||||
|
assert_eq!(result["messages"][0]["cache_control"]["type"], "ephemeral");
|
||||||
assert_eq!(result["messages"][1]["role"], "user");
|
assert_eq!(result["messages"][1]["role"], "user");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_anthropic_to_openai_drops_mixed_present_absent_system_cache_control_when_merging() {
|
||||||
|
let input = json!({
|
||||||
|
"model": "claude-3-sonnet",
|
||||||
|
"max_tokens": 1024,
|
||||||
|
"system": [
|
||||||
|
{"type": "text", "text": "You are Claude Code.", "cache_control": {"type": "ephemeral"}},
|
||||||
|
{"type": "text", "text": "Be concise."}
|
||||||
|
],
|
||||||
|
"messages": [{"role": "user", "content": "Hello"}]
|
||||||
|
});
|
||||||
|
|
||||||
|
let result = anthropic_to_openai(input, None).unwrap();
|
||||||
|
assert_eq!(result["messages"][0]["role"], "system");
|
||||||
|
assert_eq!(
|
||||||
|
result["messages"][0]["content"],
|
||||||
|
"You are Claude Code.\nBe concise."
|
||||||
|
);
|
||||||
|
assert!(result["messages"][0].get("cache_control").is_none());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_anthropic_to_openai_drops_conflicting_system_cache_control_when_merging() {
|
||||||
|
let input = json!({
|
||||||
|
"model": "claude-3-sonnet",
|
||||||
|
"max_tokens": 1024,
|
||||||
|
"system": [
|
||||||
|
{"type": "text", "text": "You are Claude Code.", "cache_control": {"type": "ephemeral"}},
|
||||||
|
{"type": "text", "text": "Be concise.", "cache_control": {"type": "ephemeral", "ttl": "5m"}}
|
||||||
|
],
|
||||||
|
"messages": [{"role": "user", "content": "Hello"}]
|
||||||
|
});
|
||||||
|
|
||||||
|
let result = anthropic_to_openai(input, None).unwrap();
|
||||||
|
assert_eq!(result["messages"][0]["role"], "system");
|
||||||
|
assert_eq!(
|
||||||
|
result["messages"][0]["content"],
|
||||||
|
"You are Claude Code.\nBe concise."
|
||||||
|
);
|
||||||
|
assert!(result["messages"][0].get("cache_control").is_none());
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_anthropic_to_openai_tool_use() {
|
fn test_anthropic_to_openai_tool_use() {
|
||||||
let input = json!({
|
let input = json!({
|
||||||
|
|||||||
Reference in New Issue
Block a user