va-ai-api-bridge 0.1.7

Protocol translation primitives for AI API request and response shapes.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49

use serde_json::Value;

pub(super) fn repair_anthropic_thinking_tool_use_order(request: &mut Value) {
    let Some(messages) = request.get_mut("messages").and_then(Value::as_array_mut) else {
        return;
    };

    for message in messages {
        if message.get("role").and_then(Value::as_str) != Some("assistant") {
            continue;
        }
        let Some(blocks) = message.get_mut("content").and_then(Value::as_array_mut) else {
            continue;
        };
        move_first_thinking_before_leading_tool_uses(blocks);
    }
}

fn move_first_thinking_before_leading_tool_uses(blocks: &mut Vec<Value>) {
    let Some(thinking_index) = blocks.iter().position(is_anthropic_thinking_block) else {
        return;
    };
    if thinking_index == 0
        || !blocks[..thinking_index]
            .iter()
            .all(is_anthropic_tool_use_block)
    {
        return;
    }

    let thinking = blocks.remove(thinking_index);
    blocks.insert(0, thinking);
}

fn is_anthropic_thinking_block(block: &Value) -> bool {
    anthropic_block_type(block) == Some("thinking")
        && block
            .get("thinking")
            .and_then(Value::as_str)
            .is_some_and(|thinking| !thinking.is_empty())
}

fn is_anthropic_tool_use_block(block: &Value) -> bool {
    anthropic_block_type(block) == Some("tool_use")
}

fn anthropic_block_type(block: &Value) -> Option<&str> {
    block.get("type").and_then(Value::as_str)
}