crabtalk-core 0.0.17

Core types and traits for the Crabtalk agent runtime
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
50
51
52
53
54
55
56
57
58
59
60

//! Tests for default_context_limit prefix matching.

use crabtalk_core::model::default_context_limit;

#[test]
fn claude_family() {
    assert_eq!(default_context_limit("claude-3-sonnet"), 200_000);
    assert_eq!(default_context_limit("claude-3-opus"), 200_000);
}

#[test]
fn gpt4o_family() {
    assert_eq!(default_context_limit("gpt-4o"), 128_000);
    assert_eq!(default_context_limit("gpt-4o-mini"), 128_000);
}

#[test]
fn gpt4_turbo() {
    assert_eq!(default_context_limit("gpt-4-turbo"), 128_000);
}

#[test]
fn gpt4_non_turbo() {
    assert_eq!(default_context_limit("gpt-4"), 8_192);
}

#[test]
fn gpt35() {
    assert_eq!(default_context_limit("gpt-3.5-turbo"), 16_385);
}

#[test]
fn o_series() {
    assert_eq!(default_context_limit("o1-preview"), 200_000);
    assert_eq!(default_context_limit("o3-mini"), 200_000);
    assert_eq!(default_context_limit("o4-mini"), 200_000);
}

#[test]
fn grok() {
    assert_eq!(default_context_limit("grok-2"), 131_072);
}

#[test]
fn qwen() {
    assert_eq!(default_context_limit("qwen-72b"), 32_768);
    assert_eq!(default_context_limit("qwq-32b"), 32_768);
}

#[test]
fn kimi_moonshot() {
    assert_eq!(default_context_limit("kimi-k1"), 128_000);
    assert_eq!(default_context_limit("moonshot-v1"), 128_000);
}

#[test]
fn unknown_model_default() {
    assert_eq!(default_context_limit("llama-3"), 8_192);
    assert_eq!(default_context_limit("unknown"), 8_192);
}