claude_code_switcher/templates/
kimi.rs

1//! Kimi AI provider template implementation
2
3use crate::{
4    settings::{
5        ClaudeSettings, EndpointConfig, HTTPConfig, ModelConfig, Permissions, ProviderConfig,
6    },
7    snapshots::SnapshotScope,
8    templates::Template,
9};
10use std::collections::HashMap;
11
12/// Kimi AI provider template
13#[derive(Debug, Clone)]
14pub struct KimiTemplate;
15
16impl Template for KimiTemplate {
17    fn template_type(&self) -> crate::templates::TemplateType {
18        crate::templates::TemplateType::Kimi
19    }
20
21    fn env_var_name(&self) -> &'static str {
22        "KIMI_API_KEY"
23    }
24
25    fn display_name(&self) -> &'static str {
26        "Kimi"
27    }
28
29    fn description(&self) -> &'static str {
30        "Kimi For Coding API - Specialized for coding tasks"
31    }
32
33    fn create_settings(&self, api_key: &str, scope: &SnapshotScope) -> ClaudeSettings {
34        let mut settings = ClaudeSettings::new();
35
36        if matches!(scope, SnapshotScope::Common | SnapshotScope::All) {
37            settings.provider = Some(ProviderConfig {
38                id: "kimi".to_string(),
39                metadata: None,
40            });
41
42            settings.model = Some(ModelConfig {
43                name: "kimi-for-coding".to_string(),
44                metadata: None,
45            });
46
47            settings.endpoint = Some(EndpointConfig {
48                id: "kimi".to_string(),
49                api_base: "https://api.kimi.com/coding/".to_string(),
50                api_key: None,
51                endpoint_id: None,
52                metadata: None,
53            });
54
55            settings.http = Some(HTTPConfig {
56                timeout_ms: Some(30000),
57                max_retries: Some(3),
58                retry_backoff_factor: Some(2.0),
59            });
60
61            settings.permissions = Some(Permissions {
62                allow_network_access: Some(true),
63                allow_filesystem_access: Some(true),
64                allow_command_execution: Some(false),
65            });
66        }
67
68        if matches!(scope, SnapshotScope::Env | SnapshotScope::All) {
69            let mut env = HashMap::new();
70            env.insert("ANTHROPIC_AUTH_TOKEN".to_string(), api_key.to_string());
71            env.insert(
72                "ANTHROPIC_BASE_URL".to_string(),
73                "https://api.kimi.com/coding/".to_string(),
74            );
75            env.insert("ANTHROPIC_MODEL".to_string(), "kimi-for-coding".to_string());
76            env.insert(
77                "ANTHROPIC_SMALL_FAST_MODEL".to_string(),
78                "kimi-for-coding".to_string(),
79            );
80            env.insert("API_TIMEOUT_MS".to_string(), "600000".to_string());
81            env.insert(
82                "CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC".to_string(),
83                "1".to_string(),
84            );
85            settings.environment = Some(env);
86        }
87
88        settings
89    }
90}
91
92/// Create Kimi template settings (legacy compatibility function)
93pub fn create_kimi_template(api_key: &str, scope: &SnapshotScope) -> ClaudeSettings {
94    let template = KimiTemplate;
95    template.create_settings(api_key, scope)
96}
claude_code_switcher/templates/kimi.rs

claude_code_switcher/templates/
kimi.rs
