auth-cloudflare 0.0.1

auth-cloudflare: Cloudflare Workers AI auth provider core - account/token resolution, model catalog types, cache paths.
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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112

//! Capabilities - marker-based capability inference from a model id.
//!
//! The three-state contract (feedback 01/05) forbids encoding incomplete
//! metadata as `Unsupported`: only a positive safety marker yields
//! `Unsupported`, and only a positive tool-family marker yields `Confirmed`.
//! Every other verdict stays `Unknown` - absence of evidence is never
//! evidence of absence.

use crate::catalog::{CapabilityState, ModelCapabilities};

/// Tool-capable model families, per Cloudflare docs/schema (feedback 01/02).
/// Family-level markers: each matches every released variant of the family
/// (e.g. `glm-5` covers glm-5.2, glm-5.3, glm-5.3-flash).
const TOOL_CAPABLE_FAMILIES: &[&str] = &[
	"deepseek-v4",
	"kimi",
	"gpt-oss",
	"qwen3",
	"nemotron",
	"mistral-small",
	"llama-3.3",
	"llama-4-scout",
	"granite",
	"glm-5",
];

/// Non-chat variants whose id also matches a tool-capable family marker
/// (image/video models must never be inferred tool-capable).
const NON_CHAT_OVERRIDES: &[&str] = &["kimi-k3", "qwen3-vl", "qwen-image"];

/// Safety/classification models - never chat-capable, never tool-capable.
const SAFETY_MARKERS: &[&str] = &["llama-guard"];

/// Infer the capability matrix from a model id (marker-based).
///
/// - Safety models: `chat` and `tools` are `Unsupported` (positive marker).
/// - Tool-capable families: `tools` is `Confirmed`.
/// - Anything else: `Unknown` - never `Unsupported` from absence alone.
pub fn infer_from_id(id: &str) -> ModelCapabilities {
	let lower = id.to_lowercase();
	let mut capabilities = ModelCapabilities::default();

	if SAFETY_MARKERS.iter().any(|marker| lower.contains(marker)) {
		capabilities.chat = CapabilityState::Unsupported;
		capabilities.tools = CapabilityState::Unsupported;
		return capabilities;
	}

	if NON_CHAT_OVERRIDES.iter().any(|marker| lower.contains(marker)) {
		return capabilities;
	}

	if TOOL_CAPABLE_FAMILIES.iter().any(|marker| lower.contains(marker)) {
		capabilities.tools = CapabilityState::Confirmed;
	}

	capabilities
}

#[cfg(test)]
mod tests {
	use super::*;

	#[test]
	fn deepseek_is_tool_capable() {
		let caps = infer_from_id("@cf/deepseek-ai/deepseek-v4-flash-0731");
		assert_eq!(caps.tools, CapabilityState::Confirmed);
		assert_eq!(caps.chat, CapabilityState::Confirmed);
	}

	#[test]
	fn premium_models_are_tool_capable() {
		assert_eq!(
			infer_from_id("@cf/deepseek-ai/deepseek-v4-pro-0813").tools,
			CapabilityState::Confirmed
		);
		assert_eq!(infer_from_id("@cf/moonshotai/kimi-k2.7-code").tools, CapabilityState::Confirmed);
		assert_eq!(infer_from_id("@cf/zai-org/glm-5.3-flash").tools, CapabilityState::Confirmed);
		assert_eq!(infer_from_id("@cf/openai/gpt-oss-120b").tools, CapabilityState::Confirmed);
	}

	#[test]
	fn safety_model_is_unsupported_not_unknown() {
		let caps = infer_from_id("@cf/meta/llama-guard-3-8b");
		assert_eq!(caps.chat, CapabilityState::Unsupported);
		assert_eq!(caps.tools, CapabilityState::Unsupported);
	}

	#[test]
	fn unknown_model_stays_unknown_not_unsupported() {
		let caps = infer_from_id("@cf/acme/mystery-model");
		assert_eq!(caps.chat, CapabilityState::Confirmed);
		assert_eq!(caps.tools, CapabilityState::Unknown);
		assert_ne!(caps.tools, CapabilityState::Unsupported);
	}

	#[test]
	fn unlisted_chat_family_stays_unknown() {
		// llama-3.2 has no tool marker - absence must yield Unknown, not Unsupported.
		let caps = infer_from_id("@cf/meta/llama-3.2-1b-instruct");
		assert_eq!(caps.tools, CapabilityState::Unknown);
		assert_ne!(caps.tools, CapabilityState::Unsupported);
	}

	#[test]
	fn non_chat_override_never_tool_capable() {
		// kimi-k3 matches the "kimi" family marker but is an image model.
		let caps = infer_from_id("@cf/moonshotai/kimi-k3");
		assert_eq!(caps.tools, CapabilityState::Unknown);
		assert_ne!(caps.tools, CapabilityState::Confirmed);
	}
}