claude_code_switcher/templates/
longcat.rs

1//! Longcat 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/// Longcat AI provider template
13#[derive(Debug, Clone)]
14pub struct LongcatTemplate;
15
16impl Template for LongcatTemplate {
17    fn template_type(&self) -> crate::templates::TemplateType {
18        crate::templates::TemplateType::Longcat
19    }
20
21    fn env_var_name(&self) -> &'static str {
22        "LONGCAT_API_KEY"
23    }
24
25    fn display_name(&self) -> &'static str {
26        "Longcat"
27    }
28
29    fn description(&self) -> &'static str {
30        "Longcat Flash Chat API - Fast and efficient conversational AI"
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: "longcat".to_string(),
39                metadata: None,
40            });
41
42            settings.model = Some(ModelConfig {
43                name: "LongCat-Flash-Chat".to_string(),
44                metadata: None,
45            });
46
47            settings.endpoint = Some(EndpointConfig {
48                id: "longcat".to_string(),
49                api_base: "https://api.longcat.chat/anthropic".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.longcat.chat/anthropic".to_string(),
74            );
75            env.insert(
76                "ANTHROPIC_MODEL".to_string(),
77                "LongCat-Flash-Chat".to_string(),
78            );
79            env.insert(
80                "ANTHROPIC_SMALL_FAST_MODEL".to_string(),
81                "LongCat-Flash-Chat".to_string(),
82            );
83            env.insert(
84                "ANTHROPIC_DEFAULT_SONNET_MODEL".to_string(),
85                "LongCat-Flash-Chat".to_string(),
86            );
87            env.insert(
88                "CLAUDE_CODE_MAX_OUTPUT_TOKENS".to_string(),
89                "8192".to_string(),
90            );
91            env.insert(
92                "CLAUDE_CODE_DISABLE_NONESSENTIAL_TRAFFIC".to_string(),
93                "1".to_string(),
94            );
95            settings.environment = Some(env);
96        }
97
98        settings
99    }
100}
101
102/// Create Longcat template settings (legacy compatibility function)
103pub fn create_longcat_template(api_key: &str, scope: &SnapshotScope) -> ClaudeSettings {
104    let template = LongcatTemplate;
105    template.create_settings(api_key, scope)
106}