hack/tools/scripts/llm-key-hunter/results/_cache/content/christoph-d_japanese-tools.json

{"repo": "christoph-d/japanese-tools", "files": {"ai/src/model.rs": {"6572516a1786d0c8e65b6c5961ac78e2aa679b9e": {"t": 1785745830.6747158, "n": 41547, "h": "efbb2bf50770d7de", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{\n    CONFIG_FILE_NAME, DEFAULT_MAX_TOKENS, DEFAULT_MAX_TOKENS_WITH_REASONING, DEFAULT_TIMEOUT,\n    DEFAULT_TIMEOUT_REASONING,\n};\n\n#[derive(Debug, Clone, PartialEq)]\nstruct Provider {\n    name: String,\n    raw_name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n    max_tokens: Option<u32>,\n    max_tokens_with_reasoning: Option<u32>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelModelConfig {\n    pub temperature: Option<f64>,\n    pub timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub system_prompt: Option<String>,\n    pub models: HashMap<String, HashMap<String, ChannelModelConfig>>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Model {\n    pub id: String,\n    pub provider: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n    pub max_tokens: Option<u32>,\n    pub timeout: Option<u64>,\n    pub temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(default)]\n    max_tokens_with_reasoning: Option<u32>,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlChannelModel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    models: Option<HashMap<String, HashMap<String, TomlChannelModel>>>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    raw_name: provider_name.clone(),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                let models = channel\n                    .models\n                    .unwrap_or_default()\n                    .into_iter()\n                    .map(|(provider_name, provider_models)| {\n                        let model_configs = provider_models\n                            .into_iter()\n                            .map(|(model_id, model_config)| {\n                                (\n                                    model_id,\n                                    ChannelModelConfig {\n                                        temperature: model_config.temperature,\n                                        timeout: model_config.timeout,\n                                    },\n                                )\n                            })\n                            .collect();\n                        (provider_name, model_configs)\n                    })\n                    .collect();\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        system_prompt: channel.system_prompt,\n                        models,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n            max_tokens: toml_config.general.max_tokens,\n            max_tokens_with_reasoning: toml_config.general.max_tokens_with_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_model_temperature(\n        &self,\n        channel_name: &str,\n        provider: &str,\n        model_id: &str,\n    ) -> Option<f64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(provider))\n            .and_then(|p| p.get(model_id))\n            .and_then(|m| m.temperature)\n    }\n\n    pub fn get_channel_model_timeout(\n        &self,\n        channel_name: &str,\n        provider: &str,\n        model_id: &str,\n    ) -> Option<u64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(provider))\n            .and_then(|p| p.get(model_id))\n            .and_then(|m| m.timeout)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, model: &Model, channel_name: &str) -> u64 {\n        self.get_channel_model_timeout(channel_name, &model.provider, &model.id)\n            .or(model.timeout)\n            .unwrap_or(if model.reasoning {\n                self.timeout_reasoning\n            } else {\n                self.timeout\n            })\n    }\n\n    pub fn get_max_tokens(&self, model: &Model) -> u32 {\n        model.max_tokens.unwrap_or_else(|| {\n            if model.reasoning {\n                self.max_tokens_with_reasoning\n                    .unwrap_or(DEFAULT_MAX_TOKENS_WITH_REASONING)\n            } else {\n                self.max_tokens.unwrap_or(DEFAULT_MAX_TOKENS)\n            }\n        })\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    provider: provider.raw_name.clone(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                    max_tokens: toml_model.max_tokens,\n                    timeout: toml_model.timeout,\n                    temperature: toml_model.temperature,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                provider: \"deepseek\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                provider: \"openrouter\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n            max_tokens: None,\n            max_tokens_with_reasoning: None,\n            timeout: DEFAULT_TIMEOUT,\n            timeout_reasoning: DEFAULT_TIMEOUT_REASONING,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    provider: \"deepseek\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    provider: \"deepseek\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            provider: \"deepseek\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n            max_tokens: None,\n            timeout: None,\n            temperature: None,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_model_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels.\"#test\"]\nmodels = { deepseek = { default = { temperature = 0.7 } } }\n\n[channels.\"#no-temp\"]\ndefault_model = \"default\"\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"deepseek\", \"default\"),\n            Some(0.7)\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"deepseek\", \"unknown-model\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#no-temp\", \"deepseek\", \"default\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#unknown\", \"deepseek\", \"default\"),\n            None\n        );\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_default_fallback() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"flash-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"flash-model\", short_name = \"d\", name = \"flash-model\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"flash-model\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"flash-model\")\n            .unwrap();\n\n        assert_eq!(\n            config.get_max_tokens(non_reasoning_model),\n            DEFAULT_MAX_TOKENS\n        );\n        assert_eq!(\n            config.get_max_tokens(reasoning_model),\n            DEFAULT_MAX_TOKENS_WITH_REASONING\n        );\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_global_config() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"flash-model\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"flash-model\", short_name = \"f\", name = \"Flash Model\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"f\".to_string()], \"flash-model\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"flash-model\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(non_reasoning_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_model), 8192);\n    }\n\n    #[test]\n    fn test_get_max_tokens_per_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"flash-model\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"flash-model\", short_name = \"f\", name = \"flash-model\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true, max_tokens = 16384 },\n  { id = \"custom-model\", short_name = \"c\", name = \"Custom Model\", reasoning = false, max_tokens = 500 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let normal_model = model_list\n            .select_model_for_channel(&[\"f\".to_string()], \"flash-model\")\n            .unwrap();\n        let reasoning_with_override = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"flash-model\")\n            .unwrap();\n        let non_reasoning_with_override = model_list\n            .select_model_for_channel(&[\"c\".to_string()], \"flash-model\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(normal_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_with_override), 16384);\n        assert_eq!(config.get_max_tokens(non_reasoning_with_override), 500);\n    }\n\n    #[test]\n    fn test_get_timeout_with_defaults() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_custom_values() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 30\ntimeout_reasoning = 60\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 30);\n    }\n\n    #[test]\n    fn test_get_timeout_only_non_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 25\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 25);\n    }\n\n    #[test]\n    fn test_get_timeout_only_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout_reasoning = 50\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\ntimeout_reasoning = 40\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"custom-timeout\", short_name = \"c\", name = \"Custom Timeout\", timeout = 100 },\n  { id = \"reasoning-default\", short_name = \"r\", name = \"Reasoning Default\", reasoning = true },\n  { id = \"reasoning-custom\", short_name = \"rc\", name = \"Reasoning Custom\", reasoning = true, timeout = 200 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let custom_timeout_model = models.models.get(1).unwrap();\n        let reasoning_default_model = models.models.get(2).unwrap();\n        let reasoning_custom_model = models.models.get(3).unwrap();\n\n        assert_eq!(config.get_timeout(default_model, \"\"), 20);\n        assert_eq!(config.get_timeout(custom_timeout_model, \"\"), 100);\n        assert_eq!(config.get_timeout(reasoning_default_model, \"\"), 40);\n        assert_eq!(config.get_timeout(reasoning_custom_model, \"\"), 200);\n    }\n\n    #[test]\n    fn test_get_timeout_channel_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"model-timeout\", short_name = \"m\", name = \"Model Timeout\", timeout = 50 }\n]\n\n[channels.\"#test\"]\nmodels = { deepseek = { default = { timeout = 100 } } }\n\n[channels.\"#other\"]\nmodels = { deepseek = { \"model-timeout\" = { timeout = 200 } } }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let model_timeout_model = models.models.get(1).unwrap();\n\n        // Channel-model override takes precedence\n        assert_eq!(config.get_timeout(default_model, \"#test\"), 100);\n        // Model-level timeout when no channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#test\"), 50);\n        // Different channel, different model with channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#other\"), 200);\n        // No channel override, uses model timeout\n        assert_eq!(config.get_timeout(model_timeout_model, \"#unknown\"), 50);\n        // No channel override, no model timeout, uses global\n        assert_eq!(config.get_timeout(default_model, \"#unknown\"), 20);\n    }\n\n    #[test]\n    fn test_model_temperature_parsed() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"with-temp\", short_name = \"t\", name = \"With Temperature\", temperature = 0.3 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(config_dir).unwrap();\n        let config = Config::new(config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.first().unwrap();\n        let temp_model = models.models.get(1).unwrap();\n\n        assert_eq!(default_model.temperature, None);\n        assert_eq!(temp_model.temperature, Some(0.3));\n    }\n}\n"}, "b8741b1eebb9aba2106ef3b90845ec90fc681146": {"t": 1785756915.0324492, "n": 41547, "h": "efbb2bf50770d7de", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{\n    CONFIG_FILE_NAME, DEFAULT_MAX_TOKENS, DEFAULT_MAX_TOKENS_WITH_REASONING, DEFAULT_TIMEOUT,\n    DEFAULT_TIMEOUT_REASONING,\n};\n\n#[derive(Debug, Clone, PartialEq)]\nstruct Provider {\n    name: String,\n    raw_name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n    max_tokens: Option<u32>,\n    max_tokens_with_reasoning: Option<u32>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelModelConfig {\n    pub temperature: Option<f64>,\n    pub timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub system_prompt: Option<String>,\n    pub models: HashMap<String, HashMap<String, ChannelModelConfig>>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Model {\n    pub id: String,\n    pub provider: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n    pub max_tokens: Option<u32>,\n    pub timeout: Option<u64>,\n    pub temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(default)]\n    max_tokens_with_reasoning: Option<u32>,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlChannelModel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    models: Option<HashMap<String, HashMap<String, TomlChannelModel>>>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    raw_name: provider_name.clone(),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                let models = channel\n                    .models\n                    .unwrap_or_default()\n                    .into_iter()\n                    .map(|(provider_name, provider_models)| {\n                        let model_configs = provider_models\n                            .into_iter()\n                            .map(|(model_id, model_config)| {\n                                (\n                                    model_id,\n                                    ChannelModelConfig {\n                                        temperature: model_config.temperature,\n                                        timeout: model_config.timeout,\n                                    },\n                                )\n                            })\n                            .collect();\n                        (provider_name, model_configs)\n                    })\n                    .collect();\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        system_prompt: channel.system_prompt,\n                        models,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n            max_tokens: toml_config.general.max_tokens,\n            max_tokens_with_reasoning: toml_config.general.max_tokens_with_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_model_temperature(\n        &self,\n        channel_name: &str,\n        provider: &str,\n        model_id: &str,\n    ) -> Option<f64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(provider))\n            .and_then(|p| p.get(model_id))\n            .and_then(|m| m.temperature)\n    }\n\n    pub fn get_channel_model_timeout(\n        &self,\n        channel_name: &str,\n        provider: &str,\n        model_id: &str,\n    ) -> Option<u64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(provider))\n            .and_then(|p| p.get(model_id))\n            .and_then(|m| m.timeout)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, model: &Model, channel_name: &str) -> u64 {\n        self.get_channel_model_timeout(channel_name, &model.provider, &model.id)\n            .or(model.timeout)\n            .unwrap_or(if model.reasoning {\n                self.timeout_reasoning\n            } else {\n                self.timeout\n            })\n    }\n\n    pub fn get_max_tokens(&self, model: &Model) -> u32 {\n        model.max_tokens.unwrap_or_else(|| {\n            if model.reasoning {\n                self.max_tokens_with_reasoning\n                    .unwrap_or(DEFAULT_MAX_TOKENS_WITH_REASONING)\n            } else {\n                self.max_tokens.unwrap_or(DEFAULT_MAX_TOKENS)\n            }\n        })\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    provider: provider.raw_name.clone(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                    max_tokens: toml_model.max_tokens,\n                    timeout: toml_model.timeout,\n                    temperature: toml_model.temperature,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                provider: \"deepseek\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                provider: \"openrouter\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n            max_tokens: None,\n            max_tokens_with_reasoning: None,\n            timeout: DEFAULT_TIMEOUT,\n            timeout_reasoning: DEFAULT_TIMEOUT_REASONING,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    provider: \"deepseek\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    provider: \"deepseek\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            provider: \"deepseek\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n            max_tokens: None,\n            timeout: None,\n            temperature: None,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_model_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels.\"#test\"]\nmodels = { deepseek = { default = { temperature = 0.7 } } }\n\n[channels.\"#no-temp\"]\ndefault_model = \"default\"\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"deepseek\", \"default\"),\n            Some(0.7)\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"deepseek\", \"unknown-model\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#no-temp\", \"deepseek\", \"default\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#unknown\", \"deepseek\", \"default\"),\n            None\n        );\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_default_fallback() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"flash-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"flash-model\", short_name = \"d\", name = \"flash-model\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"flash-model\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"flash-model\")\n            .unwrap();\n\n        assert_eq!(\n            config.get_max_tokens(non_reasoning_model),\n            DEFAULT_MAX_TOKENS\n        );\n        assert_eq!(\n            config.get_max_tokens(reasoning_model),\n            DEFAULT_MAX_TOKENS_WITH_REASONING\n        );\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_global_config() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"flash-model\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"flash-model\", short_name = \"f\", name = \"Flash Model\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"f\".to_string()], \"flash-model\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"flash-model\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(non_reasoning_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_model), 8192);\n    }\n\n    #[test]\n    fn test_get_max_tokens_per_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"flash-model\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"flash-model\", short_name = \"f\", name = \"flash-model\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true, max_tokens = 16384 },\n  { id = \"custom-model\", short_name = \"c\", name = \"Custom Model\", reasoning = false, max_tokens = 500 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let normal_model = model_list\n            .select_model_for_channel(&[\"f\".to_string()], \"flash-model\")\n            .unwrap();\n        let reasoning_with_override = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"flash-model\")\n            .unwrap();\n        let non_reasoning_with_override = model_list\n            .select_model_for_channel(&[\"c\".to_string()], \"flash-model\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(normal_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_with_override), 16384);\n        assert_eq!(config.get_max_tokens(non_reasoning_with_override), 500);\n    }\n\n    #[test]\n    fn test_get_timeout_with_defaults() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_custom_values() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 30\ntimeout_reasoning = 60\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 30);\n    }\n\n    #[test]\n    fn test_get_timeout_only_non_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 25\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 25);\n    }\n\n    #[test]\n    fn test_get_timeout_only_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout_reasoning = 50\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\ntimeout_reasoning = 40\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"custom-timeout\", short_name = \"c\", name = \"Custom Timeout\", timeout = 100 },\n  { id = \"reasoning-default\", short_name = \"r\", name = \"Reasoning Default\", reasoning = true },\n  { id = \"reasoning-custom\", short_name = \"rc\", name = \"Reasoning Custom\", reasoning = true, timeout = 200 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let custom_timeout_model = models.models.get(1).unwrap();\n        let reasoning_default_model = models.models.get(2).unwrap();\n        let reasoning_custom_model = models.models.get(3).unwrap();\n\n        assert_eq!(config.get_timeout(default_model, \"\"), 20);\n        assert_eq!(config.get_timeout(custom_timeout_model, \"\"), 100);\n        assert_eq!(config.get_timeout(reasoning_default_model, \"\"), 40);\n        assert_eq!(config.get_timeout(reasoning_custom_model, \"\"), 200);\n    }\n\n    #[test]\n    fn test_get_timeout_channel_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"model-timeout\", short_name = \"m\", name = \"Model Timeout\", timeout = 50 }\n]\n\n[channels.\"#test\"]\nmodels = { deepseek = { default = { timeout = 100 } } }\n\n[channels.\"#other\"]\nmodels = { deepseek = { \"model-timeout\" = { timeout = 200 } } }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let model_timeout_model = models.models.get(1).unwrap();\n\n        // Channel-model override takes precedence\n        assert_eq!(config.get_timeout(default_model, \"#test\"), 100);\n        // Model-level timeout when no channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#test\"), 50);\n        // Different channel, different model with channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#other\"), 200);\n        // No channel override, uses model timeout\n        assert_eq!(config.get_timeout(model_timeout_model, \"#unknown\"), 50);\n        // No channel override, no model timeout, uses global\n        assert_eq!(config.get_timeout(default_model, \"#unknown\"), 20);\n    }\n\n    #[test]\n    fn test_model_temperature_parsed() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"with-temp\", short_name = \"t\", name = \"With Temperature\", temperature = 0.3 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(config_dir).unwrap();\n        let config = Config::new(config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.first().unwrap();\n        let temp_model = models.models.get(1).unwrap();\n\n        assert_eq!(default_model.temperature, None);\n        assert_eq!(temp_model.temperature, Some(0.3));\n    }\n}\n"}, "ed0f7a7bace554eededb0901383945f0d2d66448": {"t": 1785756917.9273503, "n": 41564, "h": "2254d19dcd9f7e14", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{\n    CONFIG_FILE_NAME, DEFAULT_MAX_TOKENS, DEFAULT_MAX_TOKENS_WITH_REASONING, DEFAULT_TIMEOUT,\n    DEFAULT_TIMEOUT_REASONING,\n};\n\n#[derive(Debug, Clone, PartialEq)]\nstruct Provider {\n    name: String,\n    raw_name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n    max_tokens: Option<u32>,\n    max_tokens_with_reasoning: Option<u32>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelModelConfig {\n    pub temperature: Option<f64>,\n    pub timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub system_prompt: Option<String>,\n    pub models: HashMap<String, HashMap<String, ChannelModelConfig>>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Model {\n    pub id: String,\n    pub provider: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n    pub max_tokens: Option<u32>,\n    pub timeout: Option<u64>,\n    pub temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(default)]\n    max_tokens_with_reasoning: Option<u32>,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlChannelModel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    models: Option<HashMap<String, HashMap<String, TomlChannelModel>>>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    raw_name: provider_name.clone(),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                let models = channel\n                    .models\n                    .unwrap_or_default()\n                    .into_iter()\n                    .map(|(provider_name, provider_models)| {\n                        let model_configs = provider_models\n                            .into_iter()\n                            .map(|(model_id, model_config)| {\n                                (\n                                    model_id,\n                                    ChannelModelConfig {\n                                        temperature: model_config.temperature,\n                                        timeout: model_config.timeout,\n                                    },\n                                )\n                            })\n                            .collect();\n                        (provider_name, model_configs)\n                    })\n                    .collect();\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        system_prompt: channel.system_prompt,\n                        models,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n            max_tokens: toml_config.general.max_tokens,\n            max_tokens_with_reasoning: toml_config.general.max_tokens_with_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_model_temperature(\n        &self,\n        channel_name: &str,\n        provider: &str,\n        model_id: &str,\n    ) -> Option<f64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(provider))\n            .and_then(|p| p.get(model_id))\n            .and_then(|m| m.temperature)\n    }\n\n    pub fn get_channel_model_timeout(\n        &self,\n        channel_name: &str,\n        provider: &str,\n        model_id: &str,\n    ) -> Option<u64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(provider))\n            .and_then(|p| p.get(model_id))\n            .and_then(|m| m.timeout)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, model: &Model, channel_name: &str) -> u64 {\n        self.get_channel_model_timeout(channel_name, &model.provider, &model.id)\n            .or(model.timeout)\n            .unwrap_or(if model.reasoning {\n                self.timeout_reasoning\n            } else {\n                self.timeout\n            })\n    }\n\n    pub fn get_max_tokens(&self, model: &Model) -> u32 {\n        model.max_tokens.unwrap_or_else(|| {\n            if model.reasoning {\n                self.max_tokens_with_reasoning\n                    .unwrap_or(DEFAULT_MAX_TOKENS_WITH_REASONING)\n            } else {\n                self.max_tokens.unwrap_or(DEFAULT_MAX_TOKENS)\n            }\n        })\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    provider: provider.raw_name.clone(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                    max_tokens: toml_model.max_tokens,\n                    timeout: toml_model.timeout,\n                    temperature: toml_model.temperature,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                provider: \"deepseek\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                provider: \"openrouter\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n            max_tokens: None,\n            max_tokens_with_reasoning: None,\n            timeout: DEFAULT_TIMEOUT,\n            timeout_reasoning: DEFAULT_TIMEOUT_REASONING,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    provider: \"deepseek\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    provider: \"deepseek\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            provider: \"deepseek\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n            max_tokens: None,\n            timeout: None,\n            temperature: None,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_model_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels.\"#test\"]\nmodels = { deepseek = { default = { temperature = 0.7 } } }\n\n[channels.\"#no-temp\"]\ndefault_model = \"default\"\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"deepseek\", \"default\"),\n            Some(0.7)\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"deepseek\", \"unknown-model\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#no-temp\", \"deepseek\", \"default\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#unknown\", \"deepseek\", \"default\"),\n            None\n        );\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_default_fallback() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(\n            config.get_max_tokens(non_reasoning_model),\n            DEFAULT_MAX_TOKENS\n        );\n        assert_eq!(\n            config.get_max_tokens(reasoning_model),\n            DEFAULT_MAX_TOKENS_WITH_REASONING\n        );\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_global_config() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(non_reasoning_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_model), 8192);\n    }\n\n    #[test]\n    fn test_get_max_tokens_per_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true, max_tokens = 16384 },\n  { id = \"custom-model\", short_name = \"c\", name = \"Custom Model\", reasoning = false, max_tokens = 500 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let normal_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_with_override = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let non_reasoning_with_override = model_list\n            .select_model_for_channel(&[\"c\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(normal_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_with_override), 16384);\n        assert_eq!(config.get_max_tokens(non_reasoning_with_override), 500);\n    }\n\n    #[test]\n    fn test_get_timeout_with_defaults() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_custom_values() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 30\ntimeout_reasoning = 60\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 30);\n    }\n\n    #[test]\n    fn test_get_timeout_only_non_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 25\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 25);\n    }\n\n    #[test]\n    fn test_get_timeout_only_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout_reasoning = 50\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\ntimeout_reasoning = 40\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"custom-timeout\", short_name = \"c\", name = \"Custom Timeout\", timeout = 100 },\n  { id = \"reasoning-default\", short_name = \"r\", name = \"Reasoning Default\", reasoning = true },\n  { id = \"reasoning-custom\", short_name = \"rc\", name = \"Reasoning Custom\", reasoning = true, timeout = 200 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let custom_timeout_model = models.models.get(1).unwrap();\n        let reasoning_default_model = models.models.get(2).unwrap();\n        let reasoning_custom_model = models.models.get(3).unwrap();\n\n        assert_eq!(config.get_timeout(default_model, \"\"), 20);\n        assert_eq!(config.get_timeout(custom_timeout_model, \"\"), 100);\n        assert_eq!(config.get_timeout(reasoning_default_model, \"\"), 40);\n        assert_eq!(config.get_timeout(reasoning_custom_model, \"\"), 200);\n    }\n\n    #[test]\n    fn test_get_timeout_channel_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"model-timeout\", short_name = \"m\", name = \"Model Timeout\", timeout = 50 }\n]\n\n[channels.\"#test\"]\nmodels = { deepseek = { default = { timeout = 100 } } }\n\n[channels.\"#other\"]\nmodels = { deepseek = { \"model-timeout\" = { timeout = 200 } } }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let model_timeout_model = models.models.get(1).unwrap();\n\n        // Channel-model override takes precedence\n        assert_eq!(config.get_timeout(default_model, \"#test\"), 100);\n        // Model-level timeout when no channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#test\"), 50);\n        // Different channel, different model with channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#other\"), 200);\n        // No channel override, uses model timeout\n        assert_eq!(config.get_timeout(model_timeout_model, \"#unknown\"), 50);\n        // No channel override, no model timeout, uses global\n        assert_eq!(config.get_timeout(default_model, \"#unknown\"), 20);\n    }\n\n    #[test]\n    fn test_model_temperature_parsed() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"with-temp\", short_name = \"t\", name = \"With Temperature\", temperature = 0.3 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(config_dir).unwrap();\n        let config = Config::new(config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.first().unwrap();\n        let temp_model = models.models.get(1).unwrap();\n\n        assert_eq!(default_model.temperature, None);\n        assert_eq!(temp_model.temperature, Some(0.3));\n    }\n}\n"}, "ccf8258e065f3398dc5474b3b09559e497ca12e7": {"t": 1785756919.649877, "n": 40461, "h": "c58886ca6a4d7395", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{\n    CONFIG_FILE_NAME, DEFAULT_MAX_TOKENS, DEFAULT_MAX_TOKENS_WITH_REASONING, DEFAULT_TIMEOUT,\n    DEFAULT_TIMEOUT_REASONING,\n};\n\n#[derive(Debug, Clone, PartialEq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n    max_tokens: Option<u32>,\n    max_tokens_with_reasoning: Option<u32>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelModelConfig {\n    pub temperature: Option<f64>,\n    pub timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub system_prompt: Option<String>,\n    pub models: HashMap<String, ChannelModelConfig>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n    pub max_tokens: Option<u32>,\n    pub timeout: Option<u64>,\n    pub temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(default)]\n    max_tokens_with_reasoning: Option<u32>,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlChannelModel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\n#[serde(deny_unknown_fields)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    models: Option<HashMap<String, TomlChannelModel>>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                let models = channel\n                    .models\n                    .unwrap_or_default()\n                    .into_iter()\n                    .map(|(model_id, model_config)| {\n                        (\n                            model_id,\n                            ChannelModelConfig {\n                                temperature: model_config.temperature,\n                                timeout: model_config.timeout,\n                            },\n                        )\n                    })\n                    .collect();\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        system_prompt: channel.system_prompt,\n                        models,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n            max_tokens: toml_config.general.max_tokens,\n            max_tokens_with_reasoning: toml_config.general.max_tokens_with_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_model_temperature(&self, channel_name: &str, model_id: &str) -> Option<f64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(model_id))\n            .and_then(|m| m.temperature)\n    }\n\n    pub fn get_channel_model_timeout(&self, channel_name: &str, model_id: &str) -> Option<u64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(model_id))\n            .and_then(|m| m.timeout)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, model: &Model, channel_name: &str) -> u64 {\n        self.get_channel_model_timeout(channel_name, &model.id)\n            .or(model.timeout)\n            .unwrap_or(if model.reasoning {\n                self.timeout_reasoning\n            } else {\n                self.timeout\n            })\n    }\n\n    pub fn get_max_tokens(&self, model: &Model) -> u32 {\n        model.max_tokens.unwrap_or_else(|| {\n            if model.reasoning {\n                self.max_tokens_with_reasoning\n                    .unwrap_or(DEFAULT_MAX_TOKENS_WITH_REASONING)\n            } else {\n                self.max_tokens.unwrap_or(DEFAULT_MAX_TOKENS)\n            }\n        })\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                    max_tokens: toml_model.max_tokens,\n                    timeout: toml_model.timeout,\n                    temperature: toml_model.temperature,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n            max_tokens: None,\n            max_tokens_with_reasoning: None,\n            timeout: DEFAULT_TIMEOUT,\n            timeout_reasoning: DEFAULT_TIMEOUT_REASONING,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n            max_tokens: None,\n            timeout: None,\n            temperature: None,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_model_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels.\"#test\"]\nmodels = { \"default\" = { temperature = 0.7 } }\n\n[channels.\"#no-temp\"]\ndefault_model = \"default\"\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"default\"),\n            Some(0.7)\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"unknown-model\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#no-temp\", \"default\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#unknown\", \"default\"),\n            None\n        );\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_default_fallback() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(\n            config.get_max_tokens(non_reasoning_model),\n            DEFAULT_MAX_TOKENS\n        );\n        assert_eq!(\n            config.get_max_tokens(reasoning_model),\n            DEFAULT_MAX_TOKENS_WITH_REASONING\n        );\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_global_config() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(non_reasoning_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_model), 8192);\n    }\n\n    #[test]\n    fn test_get_max_tokens_per_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true, max_tokens = 16384 },\n  { id = \"custom-model\", short_name = \"c\", name = \"Custom Model\", reasoning = false, max_tokens = 500 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let normal_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_with_override = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let non_reasoning_with_override = model_list\n            .select_model_for_channel(&[\"c\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(normal_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_with_override), 16384);\n        assert_eq!(config.get_max_tokens(non_reasoning_with_override), 500);\n    }\n\n    #[test]\n    fn test_get_timeout_with_defaults() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_custom_values() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 30\ntimeout_reasoning = 60\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 30);\n    }\n\n    #[test]\n    fn test_get_timeout_only_non_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 25\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 25);\n    }\n\n    #[test]\n    fn test_get_timeout_only_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout_reasoning = 50\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\ntimeout_reasoning = 40\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"custom-timeout\", short_name = \"c\", name = \"Custom Timeout\", timeout = 100 },\n  { id = \"reasoning-default\", short_name = \"r\", name = \"Reasoning Default\", reasoning = true },\n  { id = \"reasoning-custom\", short_name = \"rc\", name = \"Reasoning Custom\", reasoning = true, timeout = 200 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let custom_timeout_model = models.models.get(1).unwrap();\n        let reasoning_default_model = models.models.get(2).unwrap();\n        let reasoning_custom_model = models.models.get(3).unwrap();\n\n        assert_eq!(config.get_timeout(default_model, \"\"), 20);\n        assert_eq!(config.get_timeout(custom_timeout_model, \"\"), 100);\n        assert_eq!(config.get_timeout(reasoning_default_model, \"\"), 40);\n        assert_eq!(config.get_timeout(reasoning_custom_model, \"\"), 200);\n    }\n\n    #[test]\n    fn test_get_timeout_channel_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"model-timeout\", short_name = \"m\", name = \"Model Timeout\", timeout = 50 }\n]\n\n[channels.\"#test\"]\nmodels = { \"default\" = { timeout = 100 } }\n\n[channels.\"#other\"]\nmodels = { \"model-timeout\" = { timeout = 200 } }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let model_timeout_model = models.models.get(1).unwrap();\n\n        // Channel-model override takes precedence\n        assert_eq!(config.get_timeout(default_model, \"#test\"), 100);\n        // Model-level timeout when no channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#test\"), 50);\n        // Different channel, different model with channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#other\"), 200);\n        // No channel override, uses model timeout\n        assert_eq!(config.get_timeout(model_timeout_model, \"#unknown\"), 50);\n        // No channel override, no model timeout, uses global\n        assert_eq!(config.get_timeout(default_model, \"#unknown\"), 20);\n    }\n\n    #[test]\n    fn test_model_temperature_parsed() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"with-temp\", short_name = \"t\", name = \"With Temperature\", temperature = 0.3 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(config_dir).unwrap();\n        let config = Config::new(config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.first().unwrap();\n        let temp_model = models.models.get(1).unwrap();\n\n        assert_eq!(default_model.temperature, None);\n        assert_eq!(temp_model.temperature, Some(0.3));\n    }\n}\n"}, "deb1e1c932e5c063942af53b2f692e7dd7658e13": {"t": 1785756921.5069544, "n": 40281, "h": "0235cf5b398ec6e2", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{\n    CONFIG_FILE_NAME, DEFAULT_MAX_TOKENS, DEFAULT_MAX_TOKENS_WITH_REASONING, DEFAULT_TIMEOUT,\n    DEFAULT_TIMEOUT_REASONING,\n};\n\n#[derive(Debug, Clone, PartialEq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n    max_tokens: Option<u32>,\n    max_tokens_with_reasoning: Option<u32>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelModelConfig {\n    pub temperature: Option<f64>,\n    pub timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub system_prompt: Option<String>,\n    pub models: HashMap<String, ChannelModelConfig>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n    pub max_tokens: Option<u32>,\n    pub timeout: Option<u64>,\n    pub temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(default)]\n    max_tokens_with_reasoning: Option<u32>,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannelModel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    models: Option<HashMap<String, TomlChannelModel>>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                let models = channel\n                    .models\n                    .unwrap_or_default()\n                    .into_iter()\n                    .map(|(model_id, model_config)| {\n                        (\n                            model_id,\n                            ChannelModelConfig {\n                                temperature: model_config.temperature,\n                                timeout: model_config.timeout,\n                            },\n                        )\n                    })\n                    .collect();\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        system_prompt: channel.system_prompt,\n                        models,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n            max_tokens: toml_config.general.max_tokens,\n            max_tokens_with_reasoning: toml_config.general.max_tokens_with_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_model_temperature(&self, channel_name: &str, model_id: &str) -> Option<f64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(model_id))\n            .and_then(|m| m.temperature)\n    }\n\n    pub fn get_channel_model_timeout(&self, channel_name: &str, model_id: &str) -> Option<u64> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.models.get(model_id))\n            .and_then(|m| m.timeout)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, model: &Model, channel_name: &str) -> u64 {\n        self.get_channel_model_timeout(channel_name, &model.id)\n            .or(model.timeout)\n            .unwrap_or(if model.reasoning {\n                self.timeout_reasoning\n            } else {\n                self.timeout\n            })\n    }\n\n    pub fn get_max_tokens(&self, model: &Model) -> u32 {\n        model.max_tokens.unwrap_or_else(|| {\n            if model.reasoning {\n                self.max_tokens_with_reasoning\n                    .unwrap_or(DEFAULT_MAX_TOKENS_WITH_REASONING)\n            } else {\n                self.max_tokens.unwrap_or(DEFAULT_MAX_TOKENS)\n            }\n        })\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                    max_tokens: toml_model.max_tokens,\n                    timeout: toml_model.timeout,\n                    temperature: toml_model.temperature,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n            max_tokens: None,\n            max_tokens_with_reasoning: None,\n            timeout: DEFAULT_TIMEOUT,\n            timeout_reasoning: DEFAULT_TIMEOUT_REASONING,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n            max_tokens: None,\n            timeout: None,\n            temperature: None,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_model_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels.\"#test\"]\nmodels = { \"default\" = { temperature = 0.7 } }\n\n[channels.\"#no-temp\"]\ndefault_model = \"default\"\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"default\"),\n            Some(0.7)\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#test\", \"unknown-model\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#no-temp\", \"default\"),\n            None\n        );\n        assert_eq!(\n            config.get_channel_model_temperature(\"#unknown\", \"default\"),\n            None\n        );\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_default_fallback() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(\n            config.get_max_tokens(non_reasoning_model),\n            DEFAULT_MAX_TOKENS\n        );\n        assert_eq!(\n            config.get_max_tokens(reasoning_model),\n            DEFAULT_MAX_TOKENS_WITH_REASONING\n        );\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_global_config() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(non_reasoning_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_model), 8192);\n    }\n\n    #[test]\n    fn test_get_max_tokens_per_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true, max_tokens = 16384 },\n  { id = \"custom-model\", short_name = \"c\", name = \"Custom Model\", reasoning = false, max_tokens = 500 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let normal_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_with_override = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let non_reasoning_with_override = model_list\n            .select_model_for_channel(&[\"c\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(normal_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_with_override), 16384);\n        assert_eq!(config.get_max_tokens(non_reasoning_with_override), 500);\n    }\n\n    #[test]\n    fn test_get_timeout_with_defaults() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_custom_values() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 30\ntimeout_reasoning = 60\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 30);\n    }\n\n    #[test]\n    fn test_get_timeout_only_non_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 25\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), 25);\n    }\n\n    #[test]\n    fn test_get_timeout_only_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout_reasoning = 50\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model, \"\"), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\ntimeout_reasoning = 40\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"custom-timeout\", short_name = \"c\", name = \"Custom Timeout\", timeout = 100 },\n  { id = \"reasoning-default\", short_name = \"r\", name = \"Reasoning Default\", reasoning = true },\n  { id = \"reasoning-custom\", short_name = \"rc\", name = \"Reasoning Custom\", reasoning = true, timeout = 200 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let custom_timeout_model = models.models.get(1).unwrap();\n        let reasoning_default_model = models.models.get(2).unwrap();\n        let reasoning_custom_model = models.models.get(3).unwrap();\n\n        assert_eq!(config.get_timeout(default_model, \"\"), 20);\n        assert_eq!(config.get_timeout(custom_timeout_model, \"\"), 100);\n        assert_eq!(config.get_timeout(reasoning_default_model, \"\"), 40);\n        assert_eq!(config.get_timeout(reasoning_custom_model, \"\"), 200);\n    }\n\n    #[test]\n    fn test_get_timeout_channel_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"model-timeout\", short_name = \"m\", name = \"Model Timeout\", timeout = 50 }\n]\n\n[channels.\"#test\"]\nmodels = { \"default\" = { timeout = 100 } }\n\n[channels.\"#other\"]\nmodels = { \"model-timeout\" = { timeout = 200 } }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let model_timeout_model = models.models.get(1).unwrap();\n\n        // Channel-model override takes precedence\n        assert_eq!(config.get_timeout(default_model, \"#test\"), 100);\n        // Model-level timeout when no channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#test\"), 50);\n        // Different channel, different model with channel override\n        assert_eq!(config.get_timeout(model_timeout_model, \"#other\"), 200);\n        // No channel override, uses model timeout\n        assert_eq!(config.get_timeout(model_timeout_model, \"#unknown\"), 50);\n        // No channel override, no model timeout, uses global\n        assert_eq!(config.get_timeout(default_model, \"#unknown\"), 20);\n    }\n\n    #[test]\n    fn test_model_temperature_parsed() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"with-temp\", short_name = \"t\", name = \"With Temperature\", temperature = 0.3 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(config_dir).unwrap();\n        let config = Config::new(config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.first().unwrap();\n        let temp_model = models.models.get(1).unwrap();\n\n        assert_eq!(default_model.temperature, None);\n        assert_eq!(temp_model.temperature, Some(0.3));\n    }\n}\n"}, "b89d714b530ae0d9d4ed73ce1e24f73ad29df59f": {"t": 1785756922.999373, "n": 36721, "h": "8bdcff279ea59f8d", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{\n    CONFIG_FILE_NAME, DEFAULT_MAX_TOKENS, DEFAULT_MAX_TOKENS_WITH_REASONING, DEFAULT_TIMEOUT,\n    DEFAULT_TIMEOUT_REASONING,\n};\n\n#[derive(Debug, Clone, PartialEq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n    max_tokens: Option<u32>,\n    max_tokens_with_reasoning: Option<u32>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n    pub max_tokens: Option<u32>,\n    pub timeout: Option<u64>,\n    pub temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(default)]\n    max_tokens_with_reasoning: Option<u32>,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n            max_tokens: toml_config.general.max_tokens,\n            max_tokens_with_reasoning: toml_config.general.max_tokens_with_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, model: &Model) -> u64 {\n        model.timeout.unwrap_or(if model.reasoning {\n            self.timeout_reasoning\n        } else {\n            self.timeout\n        })\n    }\n\n    pub fn get_max_tokens(&self, model: &Model) -> u32 {\n        model.max_tokens.unwrap_or_else(|| {\n            if model.reasoning {\n                self.max_tokens_with_reasoning\n                    .unwrap_or(DEFAULT_MAX_TOKENS_WITH_REASONING)\n            } else {\n                self.max_tokens.unwrap_or(DEFAULT_MAX_TOKENS)\n            }\n        })\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                    max_tokens: toml_model.max_tokens,\n                    timeout: toml_model.timeout,\n                    temperature: toml_model.temperature,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n                temperature: None,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n            max_tokens: None,\n            max_tokens_with_reasoning: None,\n            timeout: DEFAULT_TIMEOUT,\n            timeout_reasoning: DEFAULT_TIMEOUT_REASONING,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                    temperature: None,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n            max_tokens: None,\n            timeout: None,\n            temperature: None,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_default_fallback() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(\n            config.get_max_tokens(non_reasoning_model),\n            DEFAULT_MAX_TOKENS\n        );\n        assert_eq!(\n            config.get_max_tokens(reasoning_model),\n            DEFAULT_MAX_TOKENS_WITH_REASONING\n        );\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_global_config() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(non_reasoning_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_model), 8192);\n    }\n\n    #[test]\n    fn test_get_max_tokens_per_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true, max_tokens = 16384 },\n  { id = \"custom-model\", short_name = \"c\", name = \"Custom Model\", reasoning = false, max_tokens = 500 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let normal_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_with_override = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let non_reasoning_with_override = model_list\n            .select_model_for_channel(&[\"c\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(normal_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_with_override), 16384);\n        assert_eq!(config.get_max_tokens(non_reasoning_with_override), 500);\n    }\n\n    #[test]\n    fn test_get_timeout_with_defaults() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_custom_values() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 30\ntimeout_reasoning = 60\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), 30);\n    }\n\n    #[test]\n    fn test_get_timeout_only_non_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 25\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), 25);\n    }\n\n    #[test]\n    fn test_get_timeout_only_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout_reasoning = 50\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\ntimeout_reasoning = 40\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"custom-timeout\", short_name = \"c\", name = \"Custom Timeout\", timeout = 100 },\n  { id = \"reasoning-default\", short_name = \"r\", name = \"Reasoning Default\", reasoning = true },\n  { id = \"reasoning-custom\", short_name = \"rc\", name = \"Reasoning Custom\", reasoning = true, timeout = 200 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let custom_timeout_model = models.models.get(1).unwrap();\n        let reasoning_default_model = models.models.get(2).unwrap();\n        let reasoning_custom_model = models.models.get(3).unwrap();\n\n        assert_eq!(config.get_timeout(default_model), 20);\n        assert_eq!(config.get_timeout(custom_timeout_model), 100);\n        assert_eq!(config.get_timeout(reasoning_default_model), 40);\n        assert_eq!(config.get_timeout(reasoning_custom_model), 200);\n    }\n\n    #[test]\n    fn test_model_temperature_parsed() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"with-temp\", short_name = \"t\", name = \"With Temperature\", temperature = 0.3 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(config_dir).unwrap();\n        let config = Config::new(config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.first().unwrap();\n        let temp_model = models.models.get(1).unwrap();\n\n        assert_eq!(default_model.temperature, None);\n        assert_eq!(temp_model.temperature, Some(0.3));\n    }\n}\n"}, "603c554c11a47f985d307c7506da67b2eb29b9b2": {"t": 1785756924.9549394, "n": 35329, "h": "13bd0368d56ef170", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{\n    CONFIG_FILE_NAME, DEFAULT_MAX_TOKENS, DEFAULT_MAX_TOKENS_WITH_REASONING, DEFAULT_TIMEOUT,\n    DEFAULT_TIMEOUT_REASONING,\n};\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n    max_tokens: Option<u32>,\n    max_tokens_with_reasoning: Option<u32>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n    pub max_tokens: Option<u32>,\n    pub timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(default)]\n    max_tokens_with_reasoning: Option<u32>,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n            max_tokens: toml_config.general.max_tokens,\n            max_tokens_with_reasoning: toml_config.general.max_tokens_with_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, model: &Model) -> u64 {\n        model.timeout.unwrap_or_else(|| {\n            if model.reasoning {\n                self.timeout_reasoning\n            } else {\n                self.timeout\n            }\n        })\n    }\n\n    pub fn get_max_tokens(&self, model: &Model) -> u32 {\n        model.max_tokens.unwrap_or_else(|| {\n            if model.reasoning {\n                self.max_tokens_with_reasoning\n                    .unwrap_or(DEFAULT_MAX_TOKENS_WITH_REASONING)\n            } else {\n                self.max_tokens.unwrap_or(DEFAULT_MAX_TOKENS)\n            }\n        })\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                    max_tokens: toml_model.max_tokens,\n                    timeout: toml_model.timeout,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n            max_tokens: None,\n            max_tokens_with_reasoning: None,\n            timeout: DEFAULT_TIMEOUT,\n            timeout_reasoning: DEFAULT_TIMEOUT_REASONING,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n            max_tokens: None,\n            timeout: None,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_default_fallback() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(\n            config.get_max_tokens(non_reasoning_model),\n            DEFAULT_MAX_TOKENS\n        );\n        assert_eq!(\n            config.get_max_tokens(reasoning_model),\n            DEFAULT_MAX_TOKENS_WITH_REASONING\n        );\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_global_config() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(non_reasoning_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_model), 8192);\n    }\n\n    #[test]\n    fn test_get_max_tokens_per_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true, max_tokens = 16384 },\n  { id = \"custom-model\", short_name = \"c\", name = \"Custom Model\", reasoning = false, max_tokens = 500 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let normal_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_with_override = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let non_reasoning_with_override = model_list\n            .select_model_for_channel(&[\"c\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(normal_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_with_override), 16384);\n        assert_eq!(config.get_max_tokens(non_reasoning_with_override), 500);\n    }\n\n    #[test]\n    fn test_get_timeout_with_defaults() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_custom_values() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 30\ntimeout_reasoning = 60\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), 30);\n    }\n\n    #[test]\n    fn test_get_timeout_only_non_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 25\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), 25);\n    }\n\n    #[test]\n    fn test_get_timeout_only_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout_reasoning = 50\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\ntimeout_reasoning = 40\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"custom-timeout\", short_name = \"c\", name = \"Custom Timeout\", timeout = 100 },\n  { id = \"reasoning-default\", short_name = \"r\", name = \"Reasoning Default\", reasoning = true },\n  { id = \"reasoning-custom\", short_name = \"rc\", name = \"Reasoning Custom\", reasoning = true, timeout = 200 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let custom_timeout_model = models.models.get(1).unwrap();\n        let reasoning_default_model = models.models.get(2).unwrap();\n        let reasoning_custom_model = models.models.get(3).unwrap();\n\n        assert_eq!(config.get_timeout(default_model), 20);\n        assert_eq!(config.get_timeout(custom_timeout_model), 100);\n        assert_eq!(config.get_timeout(reasoning_default_model), 40);\n        assert_eq!(config.get_timeout(reasoning_custom_model), 200);\n    }\n}\n"}, "28e2a96b35926edb405ff5dddc6833300dcbb988": {"t": 1785756926.3073702, "n": 35394, "h": "32d0c1360de63612", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{\n    CONFIG_FILE_NAME, DEFAULT_MAX_TOKENS, DEFAULT_MAX_TOKENS_WITH_REASONING, DEFAULT_TIMEOUT,\n    DEFAULT_TIMEOUT_REASONING,\n};\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n    max_tokens: Option<u32>,\n    max_tokens_with_reasoning: Option<u32>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n    pub max_tokens: Option<u32>,\n    pub timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(default)]\n    max_tokens_with_reasoning: Option<u32>,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n    #[serde(default)]\n    max_tokens: Option<u32>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n            max_tokens: toml_config.general.max_tokens,\n            max_tokens_with_reasoning: toml_config.general.max_tokens_with_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, model: &Model) -> u64 {\n        model.timeout.unwrap_or_else(|| {\n            if model.reasoning {\n                self.timeout_reasoning\n            } else {\n                self.timeout\n            }\n        })\n    }\n\n    pub fn get_max_tokens(&self, model: &Model) -> u32 {\n        model.max_tokens.unwrap_or_else(|| {\n            if model.reasoning {\n                self.max_tokens_with_reasoning\n                    .unwrap_or(DEFAULT_MAX_TOKENS_WITH_REASONING)\n            } else {\n                self.max_tokens.unwrap_or(DEFAULT_MAX_TOKENS)\n            }\n        })\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                    max_tokens: toml_model.max_tokens,\n                    timeout: toml_model.timeout,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n                max_tokens: None,\n                timeout: None,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n            max_tokens: None,\n            max_tokens_with_reasoning: None,\n            timeout: DEFAULT_TIMEOUT,\n            timeout_reasoning: DEFAULT_TIMEOUT_REASONING,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    max_tokens: None,\n                    timeout: None,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n            max_tokens: None,\n            timeout: None,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_default_fallback() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(\n            config.get_max_tokens(non_reasoning_model),\n            DEFAULT_MAX_TOKENS\n        );\n        assert_eq!(\n            config.get_max_tokens(reasoning_model),\n            DEFAULT_MAX_TOKENS_WITH_REASONING\n        );\n    }\n\n    #[test]\n    fn test_get_max_tokens_uses_global_config() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_model = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(non_reasoning_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_model), 8192);\n    }\n\n    #[test]\n    fn test_get_max_tokens_per_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-chat\"\nmax_tokens = 1000\nmax_tokens_with_reasoning = 8192\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-chat\", short_name = \"d\", name = \"DeepSeek\" },\n  { id = \"reasoning-model\", short_name = \"r\", name = \"Reasoning Model\", reasoning = true, max_tokens = 16384 },\n  { id = \"custom-model\", short_name = \"c\", name = \"Custom Model\", reasoning = false, max_tokens = 500 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let normal_model = model_list\n            .select_model_for_channel(&[\"d\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let reasoning_with_override = model_list\n            .select_model_for_channel(&[\"r\".to_string()], \"deepseek-chat\")\n            .unwrap();\n        let non_reasoning_with_override = model_list\n            .select_model_for_channel(&[\"c\".to_string()], \"deepseek-chat\")\n            .unwrap();\n\n        assert_eq!(config.get_max_tokens(normal_model), 1000);\n        assert_eq!(config.get_max_tokens(reasoning_with_override), 16384);\n        assert_eq!(config.get_max_tokens(non_reasoning_with_override), 500);\n    }\n\n    #[test]\n    fn test_get_timeout_with_defaults() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_custom_values() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 30\ntimeout_reasoning = 60\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), 30);\n    }\n\n    #[test]\n    fn test_get_timeout_only_non_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 25\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), 25);\n    }\n\n    #[test]\n    fn test_get_timeout_only_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout_reasoning = 50\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\ntimeout_reasoning = 40\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"custom-timeout\", short_name = \"c\", name = \"Custom Timeout\", timeout = 100 },\n  { id = \"reasoning-default\", short_name = \"r\", name = \"Reasoning Default\", reasoning = true },\n  { id = \"reasoning-custom\", short_name = \"rc\", name = \"Reasoning Custom\", reasoning = true, timeout = 200 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let custom_timeout_model = models.models.get(1).unwrap();\n        let reasoning_default_model = models.models.get(2).unwrap();\n        let reasoning_custom_model = models.models.get(3).unwrap();\n\n        assert_eq!(config.get_timeout(default_model), 20);\n        assert_eq!(config.get_timeout(custom_timeout_model), 100);\n        assert_eq!(config.get_timeout(reasoning_default_model), 40);\n        assert_eq!(config.get_timeout(reasoning_custom_model), 200);\n    }\n}\n"}, "8b877592ad4abe70c1f0ff117c79631e4a56ceaa": {"t": 1785756927.9411688, "n": 29689, "h": "6e5a82093fc14a4e", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{CONFIG_FILE_NAME, DEFAULT_TIMEOUT, DEFAULT_TIMEOUT_REASONING};\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n    pub timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    timeout: Option<u64>,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, model: &Model) -> u64 {\n        model.timeout.unwrap_or_else(|| {\n            if model.reasoning {\n                self.timeout_reasoning\n            } else {\n                self.timeout\n            }\n        })\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                    timeout: toml_model.timeout,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n                timeout: None,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n                timeout: None,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n            timeout: DEFAULT_TIMEOUT,\n            timeout_reasoning: DEFAULT_TIMEOUT_REASONING,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    timeout: None,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                    timeout: None,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n            timeout: None,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_timeout_with_defaults() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_custom_values() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 30\ntimeout_reasoning = 60\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), 30);\n    }\n\n    #[test]\n    fn test_get_timeout_only_non_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 25\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), 25);\n    }\n\n    #[test]\n    fn test_get_timeout_only_reasoning_custom() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout_reasoning = 50\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let non_reasoning_model = models.models.get(0).unwrap();\n        assert_eq!(config.get_timeout(non_reasoning_model), DEFAULT_TIMEOUT);\n    }\n\n    #[test]\n    fn test_get_timeout_with_model_override() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\ntimeout = 20\ntimeout_reasoning = 40\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" },\n  { id = \"custom-timeout\", short_name = \"c\", name = \"Custom Timeout\", timeout = 100 },\n  { id = \"reasoning-default\", short_name = \"r\", name = \"Reasoning Default\", reasoning = true },\n  { id = \"reasoning-custom\", short_name = \"rc\", name = \"Reasoning Custom\", reasoning = true, timeout = 200 }\n]\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let default_model = models.models.get(0).unwrap();\n        let custom_timeout_model = models.models.get(1).unwrap();\n        let reasoning_default_model = models.models.get(2).unwrap();\n        let reasoning_custom_model = models.models.get(3).unwrap();\n\n        assert_eq!(config.get_timeout(default_model), 20);\n        assert_eq!(config.get_timeout(custom_timeout_model), 100);\n        assert_eq!(config.get_timeout(reasoning_default_model), 40);\n        assert_eq!(config.get_timeout(reasoning_custom_model), 200);\n    }\n}\n"}, "b31f3d72eb8375b38a08a3255f26fe995059332a": {"t": 1785756929.3505707, "n": 23668, "h": "599f7e210b894c41", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::{CONFIG_FILE_NAME, DEFAULT_TIMEOUT, DEFAULT_TIMEOUT_REASONING};\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n    timeout: u64,\n    timeout_reasoning: u64,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n    #[serde(default = \"default_timeout\")]\n    timeout: u64,\n    #[serde(default = \"default_timeout_reasoning\")]\n    timeout_reasoning: u64,\n}\n\nfn default_timeout() -> u64 {\n    DEFAULT_TIMEOUT\n}\n\nfn default_timeout_reasoning() -> u64 {\n    DEFAULT_TIMEOUT_REASONING\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n            timeout: toml_config.general.timeout,\n            timeout_reasoning: toml_config.general.timeout_reasoning,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n\n    pub fn get_timeout(&self, reasoning: bool) -> u64 {\n        if reasoning {\n            self.timeout_reasoning\n        } else {\n            self.timeout\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n}\n"}, "ee9fc0906f8877f826b98c84dcd6fa39dd487183": {"t": 1785756930.6812477, "n": 23017, "h": "7f270bed591261a7", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::CONFIG_FILE_NAME;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n    pub reasoning: bool,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n    #[serde(default)]\n    reasoning: bool,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                    reasoning: toml_model.reasoning,\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                reasoning: false,\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                reasoning: false,\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                    reasoning: false,\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            reasoning: false,\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n}\n"}, "37e351a854ba1a07d6eeaef45a5664ef3fea2b4b": {"t": 1785756932.6361814, "n": 22722, "h": "c152a2c8c4538ffb", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::CONFIG_FILE_NAME;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\nconst Z_AI_CODE_API_ENDPOINT: &str = \"https://api.z.ai/api/coding/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" | \"z-ai-code\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"z-ai-code\" => Z_AI_CODE_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            \"z-ai-code\" => \"Z.AI Code\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n}\n"}, "43d9ebd18c768f75987942f9987198ec8997e2a7": {"t": 1785756934.0685678, "n": 22492, "h": "c46f9f7a25be4554", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::CONFIG_FILE_NAME;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\nconst Z_AI_API_ENDPOINT: &str = \"https://api.z.ai/api/paas/v4/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase().replace(\"-\", \"_\");\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" | \"z-ai\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"z-ai\" => Z_AI_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            \"z-ai\" => \"Z.AI\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n}\n"}, "aa46e80ad19a4d5f192b48192e4de158e88e501a": {"t": 1785756935.554833, "n": 22281, "h": "1d3eeaf462131cfa", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::CONFIG_FILE_NAME;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase();\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix))\n                && !api_key.is_empty()\n            {\n                let endpoint = match provider_name.as_str() {\n                    \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                    \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                    \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                    \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                    \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                    _ => unreachable!(),                          // validated above\n                };\n                providers.push(Provider {\n                    name: Self::provider_display_name(&provider_name),\n                    api_key: api_key.clone(),\n                    endpoint,\n                    models: toml_provider.models,\n                });\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n}\n"}, "333e17c4296d29f4f96c3b76c82cdcb1b4a27832": {"t": 1785756937.0177777, "n": 22345, "h": "bbc25ea824d68731", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::CONFIG_FILE_NAME;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase();\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix)) {\n                if !api_key.is_empty() {\n                    let endpoint = match provider_name.as_str() {\n                        \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                        \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                        \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                        \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                        \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                        _ => unreachable!(),                          // validated above\n                    };\n                    providers.push(Provider {\n                        name: Self::provider_display_name(&provider_name),\n                        api_key: api_key.clone(),\n                        endpoint,\n                        models: toml_provider.models,\n                    });\n                }\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self, channel_default_id: &str) -> Vec<String> {\n        self.models\n            .iter()\n            .filter(|m| m.id != channel_default_id)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n            enable_compiler_explorer: false,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(\"deepseek-1\"),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_for_channel_different_default() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" },\n  { id = \"deepseek-2\", short_name = \"d2\", name = \"Deepseek 2\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"deepseek-2\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"deepseek-2\");\n\n        assert_eq!(\n            model_list.list_model_flags_human_readable(channel_default),\n            vec![\"[d]Deepseek\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_without_default_for_channel() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(\n            &env_file,\n            \"DEEPSEEK_API_KEY=test-key\\nOPENROUTER_API_KEY=test-key2\\n\",\n        )\n        .unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"d\", name = \"Deepseek\" }\n]\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"openrouter-1\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&config).expect(\"ModelList::new()\");\n\n        let channel_default = config.get_channel_default_model(\"#test\");\n        assert_eq!(channel_default, \"openrouter-1\");\n\n        assert_eq!(\n            model_list.list_model_flags_without_default(channel_default),\n            vec![\"d\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n}\n"}, "b70495a3f139bca6d1d9012444661676a5b9605e": {"t": 1785756938.409652, "n": 19769, "h": "19a5dd89c4cea08b", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::CONFIG_FILE_NAME;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n    #[serde(default)]\n    enable_compiler_explorer: bool,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase();\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix)) {\n                if !api_key.is_empty() {\n                    let endpoint = match provider_name.as_str() {\n                        \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                        \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                        \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                        \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                        \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                        _ => unreachable!(),                          // validated above\n                    };\n                    providers.push(Provider {\n                        name: Self::provider_display_name(&provider_name),\n                        api_key: api_key.clone(),\n                        endpoint,\n                        models: toml_provider.models,\n                    });\n                }\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n            enable_compiler_explorer: toml_config.general.enable_compiler_explorer,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n\n    pub fn is_compiler_explorer_enabled(&self) -> bool {\n        self.enable_compiler_explorer\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n}\n"}, "0b468d8db9cdb1eb26354dd3f94e11bf6872c2ec": {"t": 1785756940.1668062, "n": 19489, "h": "bbb0086c8506a847", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::EnvVars;\nuse crate::constants::CONFIG_FILE_NAME;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path, env_vars: &EnvVars) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase();\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Some(api_key) = env_vars.get(&format!(\"{}_API_KEY\", env_prefix)) {\n                if !api_key.is_empty() {\n                    let endpoint = match provider_name.as_str() {\n                        \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                        \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                        \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                        \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                        \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                        _ => unreachable!(),                          // validated above\n                    };\n                    providers.push(Provider {\n                        name: Self::provider_display_name(&provider_name),\n                        api_key: api_key.clone(),\n                        endpoint,\n                        models: toml_provider.models,\n                    });\n                }\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n        let cfg = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let env_vars = EnvVars {\n            vars: HashMap::new(),\n        };\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n}\n"}, "406158a184a45cb49731c9cc9b2f70db9d1a4ea4": {"t": 1785756941.7390404, "n": 19223, "h": "4f0b722b4d5636e9", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::constants::CONFIG_FILE_NAME;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n    channels: HashMap<String, ChannelConfig>,\n}\n\n#[derive(Debug, Clone, PartialEq)]\npub struct ChannelConfig {\n    pub default_model: Option<String>,\n    pub temperature: Option<f64>,\n    pub system_prompt: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    channels: Option<HashMap<String, TomlChannel>>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n}\n\n#[derive(Debug, Clone, serde::Deserialize)]\nstruct TomlChannel {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    default_model: Option<String>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    temperature: Option<f64>,\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    system_prompt: Option<String>,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase();\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                if !api_key.is_empty() {\n                    let endpoint = match provider_name.as_str() {\n                        \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                        \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                        \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                        \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                        \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                        _ => unreachable!(),                          // validated above\n                    };\n                    providers.push(Provider {\n                        name: Self::provider_display_name(&provider_name),\n                        api_key,\n                        endpoint,\n                        models: toml_provider.models,\n                    });\n                }\n            }\n        }\n        let channels = toml_config\n            .channels\n            .unwrap_or_default()\n            .into_iter()\n            .map(|(name, channel)| {\n                (\n                    name,\n                    ChannelConfig {\n                        default_model: channel.default_model,\n                        temperature: channel.temperature,\n                        system_prompt: channel.system_prompt,\n                    },\n                )\n            })\n            .collect();\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n            channels,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n\n    pub fn get_channel_system_prompt(&self, channel_name: &str) -> Option<&str> {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.system_prompt.as_deref())\n    }\n\n    pub fn get_channel_default_model(&self, channel_name: &str) -> &str {\n        self.channels\n            .get(channel_name)\n            .and_then(|c| c.default_model.as_deref())\n            .unwrap_or(&self.default_model_id)\n    }\n\n    pub fn get_channel_temperature(&self, channel_name: &str) -> Option<f64> {\n        self.channels.get(channel_name).and_then(|c| c.temperature)\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.to_string(),\n                    short_name: toml_model.short_name.to_string(),\n                    name: toml_model.name.to_string(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    pub fn select_model_for_channel(\n        &self,\n        flags: &[String],\n        channel_default_model_id: &str,\n    ) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n\n        // Use channel default model if available\n        if let Some(model) = self\n            .models\n            .iter()\n            .find(|m| m.id == channel_default_model_id)\n        {\n            return Ok(model);\n        }\n\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n            channels: HashMap::new(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        // Clear any existing DEEPSEEK_API_KEY to avoid pollution\n        unsafe {\n            std::env::remove_var(\"DEEPSEEK_API_KEY\");\n        }\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=key1\\n\").unwrap();\n        dotenvy::from_path(&env_file).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"deepseek-1\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"deepseek-1\", short_name = \"short1\", name = \"Deepseek 1\" },\n  { id = \"deepseek-2\", short_name = \"short2\", name = \"Deepseek 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let cfg = Config::new(&config_dir).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n        dotenvy::from_path(&env_file).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"openrouter-1\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter 1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter 2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let cfg = Config::new(&config_dir).expect(\"Config::new()\");\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"OPENROUTER_API_KEY=key2\\n\").unwrap();\n        dotenvy::from_path(&env_file).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r#\"\n[general]\ndefault_model = \"unknown_model\"\n\n[providers.openrouter]\nmodels = [\n  { id = \"openrouter-1\", short_name = \"o\", name = \"OpenRouter-1\" },\n  { id = \"openrouter-2\", short_name = \"p\", name = \"OpenRouter-2\" }\n]\n\"#,\n        )\n        .unwrap();\n        let cfg = Config::new(&config_dir).expect(\"Config::new()\");\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_select_model_for_channel_with_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_flags_override_channel_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![\"d\".to_string()], \"openrouter-2\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_for_channel_fallback_to_global_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model_for_channel(&vec![], \"unknown-model\");\n        let model = result.expect(\"select_model_for_channel()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_get_channel_default_model() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        dotenvy::from_path(&env_file).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"global-default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"global-default\", short_name = \"g\", name = \"Global Default\" },\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_default_model(\"#test\"), \"test-model\");\n        assert_eq!(\n            config.get_channel_default_model(\"#unknown\"),\n            \"global-default\"\n        );\n    }\n\n    #[test]\n    fn test_get_channel_temperature() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        dotenvy::from_path(&env_file).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.7 }\n\"#no-temp\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir).expect(\"Config::new()\");\n\n        assert_eq!(config.get_channel_temperature(\"#test\"), Some(0.7));\n        assert_eq!(config.get_channel_temperature(\"#no-temp\"), None);\n        assert_eq!(config.get_channel_temperature(\"#unknown\"), None);\n    }\n\n    #[test]\n    fn test_get_channel_system_prompt() {\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        dotenvy::from_path(&env_file).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default\", short_name = \"d\", name = \"Default\" }\n]\n\n[channels]\n\"#test\" = { system_prompt = \"Test prompt\" }\n\"#no-prompt\" = { default_model = \"default\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir).expect(\"Config::new()\");\n\n        assert_eq!(\n            config.get_channel_system_prompt(\"#test\"),\n            Some(\"Test prompt\")\n        );\n        assert_eq!(config.get_channel_system_prompt(\"#no-prompt\"), None);\n        assert_eq!(config.get_channel_system_prompt(\"#unknown\"), None);\n    }\n}\n"}, "4fa058ceb35b6432d9dd7bd9b8c10f92344548c7": {"t": 1785756943.3026989, "n": 15112, "h": "0de0353996836884", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\nuse crate::constants::CONFIG_FILE_NAME;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path) -> Result<Self, String> {\n        let toml_path = config_path.join(CONFIG_FILE_NAME);\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read {}: {}\", CONFIG_FILE_NAME, e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse {}: {}\", CONFIG_FILE_NAME, e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase();\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(format!(\n                            \"{}: LiteLLM provider requires an endpoint.\",\n                            CONFIG_FILE_NAME\n                        ));\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"{}: Provider '{}' endpoint is not configurable.\",\n                            CONFIG_FILE_NAME, provider_name\n                        ));\n                    }\n                }\n                _ => {\n                    return Err(format!(\n                        \"{}: Unknown provider: {}\",\n                        CONFIG_FILE_NAME, provider_name\n                    ));\n                }\n            }\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                if !api_key.is_empty() {\n                    let endpoint = match provider_name.as_str() {\n                        \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                        \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                        \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                        \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                        \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                        _ => unreachable!(),                          // validated above\n                    };\n                    providers.push(Provider {\n                        name: Self::provider_display_name(&provider_name),\n                        api_key,\n                        endpoint,\n                        models: toml_provider.models,\n                    });\n                }\n            }\n        }\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.clone(),\n                    short_name: toml_model.short_name.clone(),\n                    name: toml_model.name.clone(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"Default model not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                models: vec![\n                    TomlModel {\n                        id: \"deepseek-1\".to_string(),\n                        short_name: \"short1\".to_string(),\n                        name: \"Deepseek 1\".to_string(),\n                    },\n                    TomlModel {\n                        id: \"deepseek-2\".to_string(),\n                        short_name: \"short2\".to_string(),\n                        name: \"Deepseek 2\".to_string(),\n                    },\n                ],\n            }],\n            default_model_id: \"deepseek-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                models: vec![\n                    TomlModel {\n                        id: \"openrouter-1\".to_string(),\n                        short_name: \"o\".to_string(),\n                        name: \"OpenRouter 1\".to_string(),\n                    },\n                    TomlModel {\n                        id: \"openrouter-2\".to_string(),\n                        short_name: \"p\".to_string(),\n                        name: \"OpenRouter 2\".to_string(),\n                    },\n                ],\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                models: vec![\n                    TomlModel {\n                        id: \"openrouter-1\".to_string(),\n                        short_name: \"o\".to_string(),\n                        name: \"OpenRouter-1\".to_string(),\n                    },\n                    TomlModel {\n                        id: \"openrouter-2\".to_string(),\n                        short_name: \"p\".to_string(),\n                        name: \"OpenRouter-2\".to_string(),\n                    },\n                ],\n            }],\n            default_model_id: \"unknown_model\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"Default model\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"d\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_with_empty_flag() {\n        let model_list = setup_model_list();\n        let flags = vec![\"\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"d\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n}\n"}, "3e8fe7f8804b40a20a3637ae3dfebb3da49630ca": {"t": 1785756945.6798081, "n": 14855, "h": "e62bd0063f00ed32", "c": "use gettextrs::gettext;\nuse std::collections::HashMap;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlConfig {\n    general: TomlGeneral,\n    providers: HashMap<String, TomlProvider>,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlGeneral {\n    default_model: String,\n}\n\n#[derive(Debug, serde::Deserialize)]\nstruct TomlProvider {\n    #[serde(skip_serializing_if = \"Option::is_none\")]\n    endpoint: Option<String>,\n    models: Vec<TomlModel>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq, serde::Deserialize)]\nstruct TomlModel {\n    id: String,\n    short_name: String,\n    name: String,\n}\n\nimpl Config {\n    pub fn new(config_path: &std::path::Path) -> Result<Self, String> {\n        let toml_path = config_path.join(\"config.toml\");\n        let toml_content = std::fs::read_to_string(toml_path)\n            .map_err(|e| format!(\"Failed to read config.toml: {}\", e))?;\n        let toml_config: TomlConfig = toml::from_str(&toml_content)\n            .map_err(|e| format!(\"Failed to parse config.toml: {}\", e))?;\n\n        let mut providers = Vec::new();\n        for (provider_name, toml_provider) in toml_config.providers {\n            let env_prefix = provider_name.to_uppercase();\n            match provider_name.as_str() {\n                \"litellm\" => {\n                    if toml_provider\n                        .endpoint\n                        .as_deref()\n                        .unwrap_or_default()\n                        .is_empty()\n                    {\n                        return Err(\"LiteLLM provider requires an endpoint.\".to_string());\n                    }\n                }\n                \"anthropic\" | \"deepseek\" | \"mistral\" | \"openrouter\" => {\n                    if toml_provider.endpoint.is_some() {\n                        return Err(format!(\n                            \"Provider '{}' endpoint is not configurable.\",\n                            provider_name\n                        ));\n                    }\n                }\n                _ => return Err(format!(\"Unknown provider: {}\", provider_name)),\n            }\n\n            // API key comes from an environment variable\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                if !api_key.is_empty() {\n                    let endpoint = match provider_name.as_str() {\n                        \"anthropic\" => ANTHROPIC_API_ENDPOINT.to_string(),\n                        \"deepseek\" => DEEPSEEK_API_ENDPOINT.to_string(),\n                        \"mistral\" => MISTRAL_API_ENDPOINT.to_string(),\n                        \"openrouter\" => OPENROUTER_API_ENDPOINT.to_string(),\n                        \"litellm\" => toml_provider.endpoint.unwrap(), // we validated above\n                        _ => unreachable!(),                          // validated above\n                    };\n                    providers.push(Provider {\n                        name: Self::provider_display_name(&provider_name),\n                        api_key,\n                        endpoint,\n                        models: toml_provider.models,\n                    });\n                }\n            }\n        }\n\n        Ok(Config {\n            providers,\n            default_model_id: toml_config.general.default_model,\n        })\n    }\n\n    fn provider_display_name(provider_name: &str) -> String {\n        match provider_name {\n            \"anthropic\" => \"Anthropic\".to_string(),\n            \"deepseek\" => \"Deepseek\".to_string(),\n            \"litellm\" => \"LiteLLM\".to_string(),\n            \"mistral\" => \"Mistral\".to_string(),\n            \"openrouter\" => \"OpenRouter\".to_string(),\n            _ => provider_name.to_string(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            for toml_model in &provider.models {\n                models.push(Model {\n                    id: toml_model.id.clone(),\n                    short_name: toml_model.short_name.clone(),\n                    name: toml_model.name.clone(),\n                    api_key: provider.api_key.clone(),\n                    endpoint: provider.endpoint.clone(),\n                });\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"DEFAULT_MODEL not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                models: vec![\n                    TomlModel {\n                        id: \"deepseek-1\".to_string(),\n                        short_name: \"short1\".to_string(),\n                        name: \"Deepseek 1\".to_string(),\n                    },\n                    TomlModel {\n                        id: \"deepseek-2\".to_string(),\n                        short_name: \"short2\".to_string(),\n                        name: \"Deepseek 2\".to_string(),\n                    },\n                ],\n            }],\n            default_model_id: \"deepseek-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                models: vec![\n                    TomlModel {\n                        id: \"openrouter-1\".to_string(),\n                        short_name: \"o\".to_string(),\n                        name: \"OpenRouter 1\".to_string(),\n                    },\n                    TomlModel {\n                        id: \"openrouter-2\".to_string(),\n                        short_name: \"p\".to_string(),\n                        name: \"OpenRouter 2\".to_string(),\n                    },\n                ],\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                models: vec![\n                    TomlModel {\n                        id: \"openrouter-1\".to_string(),\n                        short_name: \"o\".to_string(),\n                        name: \"OpenRouter-1\".to_string(),\n                    },\n                    TomlModel {\n                        id: \"openrouter-2\".to_string(),\n                        short_name: \"p\".to_string(),\n                        name: \"OpenRouter-2\".to_string(),\n                    },\n                ],\n            }],\n            default_model_id: \"unknown_model\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"DEFAULT_MODEL\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"d\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_with_empty_flag() {\n        let model_list = setup_model_list();\n        let flags = vec![\"\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"d\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n}\n"}, "0414dc7d24b430c39ff62df4b4f1200c1786a676": {"t": 1785756947.0054426, "n": 15697, "h": "d8cd6d662bc86de0", "c": "use gettextrs::gettext;\nuse nom::{\n    IResult, Parser,\n    bytes::complete::{tag, take_while1},\n    character::complete::multispace1,\n    combinator::{all_consuming, map},\n    multi::separated_list0,\n    sequence::delimited,\n};\n\nuse crate::formatget;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    model_string: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\nconst ANTHROPIC_API_ENDPOINT: &str = \"https://api.anthropic.com/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        let mut providers = Vec::new();\n        let provider_configs = [\n            (\"ANTHROPIC\", \"Anthropic\", ANTHROPIC_API_ENDPOINT),\n            (\"DEEPSEEK\", \"Deepseek\", DEEPSEEK_API_ENDPOINT),\n            (\n                \"LITELLM\",\n                \"LiteLLM\",\n                &std::env::var(\"LITELLM_API_ENDPOINT\").unwrap_or_default(),\n            ),\n            (\"MISTRAL\", \"Mistral\", MISTRAL_API_ENDPOINT),\n            (\"OPENROUTER\", \"OpenRouter\", OPENROUTER_API_ENDPOINT),\n        ];\n        for (env_prefix, name, endpoint) in provider_configs.iter() {\n            if endpoint.is_empty() {\n                continue;\n            }\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                if !api_key.is_empty() {\n                    let model_string =\n                        std::env::var(format!(\"{}_MODELS\", env_prefix)).unwrap_or_default();\n                    providers.push(Provider {\n                        name: name.to_string(),\n                        api_key,\n                        endpoint: endpoint.to_string(),\n                        model_string,\n                    });\n                }\n            }\n        }\n        Config {\n            providers,\n            default_model_id: std::env::var(\"DEFAULT_MODEL\").unwrap_or_default(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            if !provider.model_string.is_empty() {\n                models.extend(parse_model_config(\n                    &provider.model_string,\n                    &provider.api_key,\n                    &provider.endpoint,\n                )?);\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"DEFAULT_MODEL not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\nfn parse_model_id(i: &str) -> IResult<&str, &str> {\n    take_while1(|c: char| c != '[' && !c.is_whitespace())(i)\n}\n\nfn is_short_name_char(c: char) -> bool {\n    c.is_alphanumeric() || c == '_' || c == '-'\n}\n\nfn is_human_name_char(c: char) -> bool {\n    c != ')'\n}\n\n// Parses a single model from this syntax: <model ID>[<short name>](<human-readable name>)\nfn parse_model<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Model> {\n    map(\n        (\n            parse_model_id,\n            delimited(tag(\"[\"), take_while1(is_short_name_char), tag(\"]\")),\n            delimited(tag(\"(\"), take_while1(is_human_name_char), tag(\")\")),\n        ),\n        |(id, short_name, name)| Model {\n            id: id.to_string(),\n            short_name: short_name.to_string(),\n            name: name.to_string(),\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        },\n    )\n    .parse(i)\n}\n\n// Parses a white-space separated list of models.\nfn parse_models<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Vec<Model>> {\n    all_consuming(separated_list0(multispace1, |i| {\n        parse_model(i, api_key, endpoint)\n    }))\n    .parse(i)\n}\n\n#[cfg(test)]\nmod parser_tests {\n    use super::*;\n\n    #[test]\n    fn test_parse_model_nom() {\n        let input = \"deepseek-1[short1](Deepseek 1)\";\n        let (rest, parsed) = parse_model(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(rest, \"\");\n        assert_eq!(\n            parsed,\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"short1\".to_string(),\n                name: \"Deepseek 1\".to_string(),\n                api_key: \"key\".to_string(),\n                endpoint: \"endpoint\".to_string()\n            }\n        );\n\n        let input2 = \"openrouter-2[o](OpenRouter 2)\";\n        let (_, parsed2) = parse_model(input2, \"\", \"\").unwrap();\n        assert_eq!(parsed2.id, \"openrouter-2\");\n        assert_eq!(parsed2.short_name, \"o\");\n        assert_eq!(parsed2.name, \"OpenRouter 2\");\n    }\n\n    #[test]\n    fn test_parse_model_nom_invalid() {\n        assert!(parse_model(\"invalidmodel\", \"\", \"\").is_err());\n        assert!(parse_model(\"id[short](name\", \"\", \"\").is_err());\n        assert!(parse_model(\"idshort](name)\", \"\", \"\").is_err());\n    }\n\n    #[test]\n    fn test_parse_models() {\n        let input = \"deepseek-1[short1](Deepseek 1) openrouter-2[o](OpenRouter 2)\";\n        let (rest, parsed) = parse_models(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(\n            parsed,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                },\n                Model {\n                    id: \"openrouter-2\".to_string(),\n                    short_name: \"o\".to_string(),\n                    name: \"OpenRouter 2\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                }\n            ]\n        );\n        assert_eq!(rest, \"\");\n    }\n}\n\nfn parse_model_config(\n    models_list: &str,\n    api_key: &str,\n    endpoint: &str,\n) -> Result<Vec<Model>, String> {\n    match parse_models(models_list, api_key, endpoint) {\n        Ok((_, models)) => Ok(models),\n        Err(e) => Err(formatget!(\"Invalid model syntax: {}\", e)),\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                model_string: \"deepseek-1[short1](Deepseek 1) deepseek-2[short2](Deepseek 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"deepseek-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter 1) openrouter-2[p](OpenRouter 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_parses_missing_short_name() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2\".to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"openrouter-2\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2[p](OpenRouter-2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"unknown_model\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"DEFAULT_MODEL\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"d\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_with_empty_flag() {\n        let model_list = setup_model_list();\n        let flags = vec![\"\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"d\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n}\n"}, "fee34386eaf34ea308140b779aba29451399d459": {"t": 1785756948.2139544, "n": 15547, "h": "83b2b86c53486e80", "c": "use gettextrs::gettext;\nuse nom::{\n    IResult, Parser,\n    bytes::complete::{tag, take_while1},\n    character::complete::multispace1,\n    combinator::{all_consuming, map},\n    multi::separated_list0,\n    sequence::delimited,\n};\n\nuse crate::formatget;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    model_string: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        let mut providers = Vec::new();\n        let provider_configs = [\n            (\"DEEPSEEK\", \"Deepseek\", DEEPSEEK_API_ENDPOINT),\n            (\"MISTRAL\", \"Mistral\", MISTRAL_API_ENDPOINT),\n            (\"OPENROUTER\", \"OpenRouter\", OPENROUTER_API_ENDPOINT),\n            (\n                \"LITELLM\",\n                \"LiteLLM\",\n                &std::env::var(\"LITELLM_API_ENDPOINT\").unwrap_or_default(),\n            ),\n        ];\n        for (env_prefix, name, endpoint) in provider_configs.iter() {\n            if endpoint.is_empty() {\n                continue;\n            }\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                if !api_key.is_empty() {\n                    let model_string =\n                        std::env::var(format!(\"{}_MODELS\", env_prefix)).unwrap_or_default();\n                    providers.push(Provider {\n                        name: name.to_string(),\n                        api_key,\n                        endpoint: endpoint.to_string(),\n                        model_string,\n                    });\n                }\n            }\n        }\n        Config {\n            providers,\n            default_model_id: std::env::var(\"DEFAULT_MODEL\").unwrap_or_default(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            if !provider.model_string.is_empty() {\n                models.extend(parse_model_config(\n                    &provider.model_string,\n                    &provider.api_key,\n                    &provider.endpoint,\n                )?);\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"DEFAULT_MODEL not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\nfn parse_model_id(i: &str) -> IResult<&str, &str> {\n    take_while1(|c: char| c != '[' && !c.is_whitespace())(i)\n}\n\nfn is_short_name_char(c: char) -> bool {\n    c.is_alphanumeric() || c == '_' || c == '-'\n}\n\nfn is_human_name_char(c: char) -> bool {\n    c != ')'\n}\n\n// Parses a single model from this syntax: <model ID>[<short name>](<human-readable name>)\nfn parse_model<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Model> {\n    map(\n        (\n            parse_model_id,\n            delimited(tag(\"[\"), take_while1(is_short_name_char), tag(\"]\")),\n            delimited(tag(\"(\"), take_while1(is_human_name_char), tag(\")\")),\n        ),\n        |(id, short_name, name)| Model {\n            id: id.to_string(),\n            short_name: short_name.to_string(),\n            name: name.to_string(),\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        },\n    )\n    .parse(i)\n}\n\n// Parses a white-space separated list of models.\nfn parse_models<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Vec<Model>> {\n    all_consuming(separated_list0(multispace1, |i| {\n        parse_model(i, api_key, endpoint)\n    }))\n    .parse(i)\n}\n\n#[cfg(test)]\nmod parser_tests {\n    use super::*;\n\n    #[test]\n    fn test_parse_model_nom() {\n        let input = \"deepseek-1[short1](Deepseek 1)\";\n        let (rest, parsed) = parse_model(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(rest, \"\");\n        assert_eq!(\n            parsed,\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"short1\".to_string(),\n                name: \"Deepseek 1\".to_string(),\n                api_key: \"key\".to_string(),\n                endpoint: \"endpoint\".to_string()\n            }\n        );\n\n        let input2 = \"openrouter-2[o](OpenRouter 2)\";\n        let (_, parsed2) = parse_model(input2, \"\", \"\").unwrap();\n        assert_eq!(parsed2.id, \"openrouter-2\");\n        assert_eq!(parsed2.short_name, \"o\");\n        assert_eq!(parsed2.name, \"OpenRouter 2\");\n    }\n\n    #[test]\n    fn test_parse_model_nom_invalid() {\n        assert!(parse_model(\"invalidmodel\", \"\", \"\").is_err());\n        assert!(parse_model(\"id[short](name\", \"\", \"\").is_err());\n        assert!(parse_model(\"idshort](name)\", \"\", \"\").is_err());\n    }\n\n    #[test]\n    fn test_parse_models() {\n        let input = \"deepseek-1[short1](Deepseek 1) openrouter-2[o](OpenRouter 2)\";\n        let (rest, parsed) = parse_models(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(\n            parsed,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                },\n                Model {\n                    id: \"openrouter-2\".to_string(),\n                    short_name: \"o\".to_string(),\n                    name: \"OpenRouter 2\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                }\n            ]\n        );\n        assert_eq!(rest, \"\");\n    }\n}\n\nfn parse_model_config(\n    models_list: &str,\n    api_key: &str,\n    endpoint: &str,\n) -> Result<Vec<Model>, String> {\n    match parse_models(models_list, api_key, endpoint) {\n        Ok((_, models)) => Ok(models),\n        Err(e) => Err(formatget!(\"Invalid model syntax: {}\", e)),\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                model_string: \"deepseek-1[short1](Deepseek 1) deepseek-2[short2](Deepseek 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"deepseek-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter 1) openrouter-2[p](OpenRouter 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_parses_missing_short_name() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2\".to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"openrouter-2\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2[p](OpenRouter-2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"unknown_model\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"DEFAULT_MODEL\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"d\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_with_empty_flag() {\n        let model_list = setup_model_list();\n        let flags = vec![\"\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"d\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n}\n"}, "ca93df33ed1f9027eceacf01ef240ab41e8c4c65": {"t": 1785756949.622697, "n": 15220, "h": "9b68c0da0aec9b63", "c": "use gettextrs::gettext;\nuse nom::{\n    IResult, Parser,\n    bytes::complete::{tag, take_while1},\n    character::complete::multispace1,\n    combinator::{all_consuming, map},\n    multi::separated_list0,\n    sequence::delimited,\n};\n\nuse crate::formatget;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    model_string: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        let mut providers = Vec::new();\n        let provider_configs = [\n            (\"DEEPSEEK\", \"Deepseek\", DEEPSEEK_API_ENDPOINT),\n            (\"MISTRAL\", \"Mistral\", MISTRAL_API_ENDPOINT),\n            (\"OPENROUTER\", \"OpenRouter\", OPENROUTER_API_ENDPOINT),\n        ];\n        for (env_prefix, name, endpoint) in provider_configs.iter() {\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                let model_string =\n                    std::env::var(format!(\"{}_MODELS\", env_prefix)).unwrap_or_default();\n                providers.push(Provider {\n                    name: name.to_string(),\n                    api_key,\n                    endpoint: endpoint.to_string(),\n                    model_string,\n                });\n            }\n        }\n        Config {\n            providers,\n            default_model_id: std::env::var(\"DEFAULT_MODEL\").unwrap_or_default(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            if !provider.model_string.is_empty() {\n                models.extend(parse_model_config(\n                    &provider.model_string,\n                    &provider.api_key,\n                    &provider.endpoint,\n                )?);\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"DEFAULT_MODEL not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\nfn parse_model_id(i: &str) -> IResult<&str, &str> {\n    take_while1(|c: char| c != '[' && !c.is_whitespace())(i)\n}\n\nfn is_short_name_char(c: char) -> bool {\n    c.is_alphanumeric() || c == '_' || c == '-'\n}\n\nfn is_human_name_char(c: char) -> bool {\n    c != ')'\n}\n\n// Parses a single model from this syntax: <model ID>[<short name>](<human-readable name>)\nfn parse_model<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Model> {\n    map(\n        (\n            parse_model_id,\n            delimited(tag(\"[\"), take_while1(is_short_name_char), tag(\"]\")),\n            delimited(tag(\"(\"), take_while1(is_human_name_char), tag(\")\")),\n        ),\n        |(id, short_name, name)| Model {\n            id: id.to_string(),\n            short_name: short_name.to_string(),\n            name: name.to_string(),\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        },\n    )\n    .parse(i)\n}\n\n// Parses a white-space separated list of models.\nfn parse_models<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Vec<Model>> {\n    all_consuming(separated_list0(multispace1, |i| {\n        parse_model(i, api_key, endpoint)\n    }))\n    .parse(i)\n}\n\n#[cfg(test)]\nmod parser_tests {\n    use super::*;\n\n    #[test]\n    fn test_parse_model_nom() {\n        let input = \"deepseek-1[short1](Deepseek 1)\";\n        let (rest, parsed) = parse_model(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(rest, \"\");\n        assert_eq!(\n            parsed,\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"short1\".to_string(),\n                name: \"Deepseek 1\".to_string(),\n                api_key: \"key\".to_string(),\n                endpoint: \"endpoint\".to_string()\n            }\n        );\n\n        let input2 = \"openrouter-2[o](OpenRouter 2)\";\n        let (_, parsed2) = parse_model(input2, \"\", \"\").unwrap();\n        assert_eq!(parsed2.id, \"openrouter-2\");\n        assert_eq!(parsed2.short_name, \"o\");\n        assert_eq!(parsed2.name, \"OpenRouter 2\");\n    }\n\n    #[test]\n    fn test_parse_model_nom_invalid() {\n        assert!(parse_model(\"invalidmodel\", \"\", \"\").is_err());\n        assert!(parse_model(\"id[short](name\", \"\", \"\").is_err());\n        assert!(parse_model(\"idshort](name)\", \"\", \"\").is_err());\n    }\n\n    #[test]\n    fn test_parse_models() {\n        let input = \"deepseek-1[short1](Deepseek 1) openrouter-2[o](OpenRouter 2)\";\n        let (rest, parsed) = parse_models(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(\n            parsed,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                },\n                Model {\n                    id: \"openrouter-2\".to_string(),\n                    short_name: \"o\".to_string(),\n                    name: \"OpenRouter 2\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                }\n            ]\n        );\n        assert_eq!(rest, \"\");\n    }\n}\n\nfn parse_model_config(\n    models_list: &str,\n    api_key: &str,\n    endpoint: &str,\n) -> Result<Vec<Model>, String> {\n    match parse_models(models_list, api_key, endpoint) {\n        Ok((_, models)) => Ok(models),\n        Err(e) => Err(formatget!(\"Invalid model syntax: {}\", e)),\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                model_string: \"deepseek-1[short1](Deepseek 1) deepseek-2[short2](Deepseek 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"deepseek-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(\n            model_list.models,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                },\n                Model {\n                    id: \"deepseek-2\".to_string(),\n                    short_name: \"short2\".to_string(),\n                    name: \"Deepseek 2\".to_string(),\n                    api_key: \"key1\".to_string(),\n                    endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                }\n            ]\n        );\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter 1) openrouter-2[p](OpenRouter 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_parses_missing_short_name() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2\".to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"openrouter-2\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2[p](OpenRouter-2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"unknown_model\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"DEFAULT_MODEL\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"d\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_with_empty_flag() {\n        let model_list = setup_model_list();\n        let flags = vec![\"\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"d\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n}\n"}, "d9e0235315bdbb2ff7e3430046aae98aba69f505": {"t": 1785756951.0311203, "n": 15504, "h": "b44f608b66ab31c4", "c": "use gettextrs::gettext;\nuse nom::{\n    IResult, Parser,\n    bytes::complete::{tag, take_while1},\n    character::complete::multispace1,\n    combinator::{all_consuming, map},\n    multi::separated_list0,\n    sequence::delimited,\n};\n\nuse crate::formatget;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    model_string: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        let mut providers = Vec::new();\n        let provider_configs = [\n            (\"DEEPSEEK\", \"Deepseek\", DEEPSEEK_API_ENDPOINT),\n            (\"MISTRAL\", \"Mistral\", MISTRAL_API_ENDPOINT),\n            (\"OPENROUTER\", \"OpenRouter\", OPENROUTER_API_ENDPOINT),\n        ];\n        for (env_prefix, name, endpoint) in provider_configs.iter() {\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                let model_string =\n                    std::env::var(format!(\"{}_MODELS\", env_prefix)).unwrap_or_default();\n                providers.push(Provider {\n                    name: name.to_string(),\n                    api_key,\n                    endpoint: endpoint.to_string(),\n                    model_string,\n                });\n            }\n        }\n        Config {\n            providers,\n            default_model_id: std::env::var(\"DEFAULT_MODEL\").unwrap_or_default(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            if !provider.model_string.is_empty() {\n                models.extend(parse_model_config(\n                    &provider.model_string,\n                    &provider.api_key,\n                    &provider.endpoint,\n                )?);\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"DEFAULT_MODEL not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\nfn parse_model_id(i: &str) -> IResult<&str, &str> {\n    take_while1(|c: char| c != '[' && !c.is_whitespace())(i)\n}\nfn is_short_name_char(c: char) -> bool {\n    c.is_alphanumeric() || c == '_' || c == '-'\n}\nfn is_human_name_char(c: char) -> bool {\n    c != ')'\n}\n// Parses a single model from this syntax: <model ID>[<short name>](<human-readable name>)\nfn parse_model<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Model> {\n    map(\n        (\n            parse_model_id,\n            delimited(tag(\"[\"), take_while1(is_short_name_char), tag(\"]\")),\n            delimited(tag(\"(\"), take_while1(is_human_name_char), tag(\")\")),\n        ),\n        |(id, short_name, name)| Model {\n            id: id.to_string(),\n            short_name: short_name.to_string(),\n            name: name.to_string(),\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        },\n    )\n    .parse(i)\n}\n\n// Parses a white-space separated list of models.\nfn parse_models<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Vec<Model>> {\n    all_consuming(separated_list0(multispace1, |i| {\n        parse_model(i, api_key, endpoint)\n    }))\n    .parse(i)\n}\n\n#[cfg(test)]\nmod parser_tests {\n    use super::*;\n\n    #[test]\n    fn test_parse_model_nom() {\n        let input = \"deepseek-1[short1](Deepseek 1)\";\n        let (rest, parsed) = parse_model(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(rest, \"\");\n        assert_eq!(\n            parsed,\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"short1\".to_string(),\n                name: \"Deepseek 1\".to_string(),\n                api_key: \"key\".to_string(),\n                endpoint: \"endpoint\".to_string()\n            }\n        );\n\n        let input2 = \"openrouter-2[o](OpenRouter 2)\";\n        let (_, parsed2) = parse_model(input2, \"\", \"\").unwrap();\n        assert_eq!(parsed2.id, \"openrouter-2\");\n        assert_eq!(parsed2.short_name, \"o\");\n        assert_eq!(parsed2.name, \"OpenRouter 2\");\n    }\n\n    #[test]\n    fn test_parse_model_nom_invalid() {\n        assert!(parse_model(\"invalidmodel\", \"\", \"\").is_err());\n        assert!(parse_model(\"id[short](name\", \"\", \"\").is_err());\n        assert!(parse_model(\"idshort](name)\", \"\", \"\").is_err());\n    }\n\n    #[test]\n    fn test_parse_models() {\n        let input = \"deepseek-1[short1](Deepseek 1) openrouter-2[o](OpenRouter 2)\";\n        let (rest, parsed) = parse_models(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(\n            parsed,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                },\n                Model {\n                    id: \"openrouter-2\".to_string(),\n                    short_name: \"o\".to_string(),\n                    name: \"OpenRouter 2\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                }\n            ]\n        );\n        assert_eq!(rest, \"\");\n    }\n}\n\nfn parse_model_config(\n    models_list: &str,\n    api_key: &str,\n    endpoint: &str,\n) -> Result<Vec<Model>, String> {\n    match parse_models(models_list, api_key, endpoint) {\n        Ok((_, models)) => Ok(models),\n        Err(e) => Err(formatget!(\"Invalid model syntax: {}\", e)),\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                model_string: \"deepseek-1[short1](Deepseek 1) deepseek-2[short2](Deepseek 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"deepseek-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"deepseek-1\");\n        assert_eq!(model_list.models[0].short_name, \"short1\");\n        assert_eq!(model_list.models[0].name, \"Deepseek 1\");\n        assert_eq!(model_list.models[0].api_key, \"key1\");\n        assert_eq!(model_list.models[0].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.models[1].id, \"deepseek-2\");\n        assert_eq!(model_list.models[1].short_name, \"short2\");\n        assert_eq!(model_list.models[1].name, \"Deepseek 2\");\n        assert_eq!(model_list.models[1].api_key, \"key1\");\n        assert_eq!(model_list.models[1].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter 1) openrouter-2[p](OpenRouter 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_parses_missing_short_name() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2\".to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"openrouter-2\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2[p](OpenRouter-2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"unknown_model\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"DEFAULT_MODEL\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"d\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_empty_and_valid_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"\".to_string(), \"clear_history\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"d\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"[o]OpenRouter\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_empty() {\n        let model_list = ModelList {\n            models: vec![],\n            default_model_index: 0,\n        };\n        let models = model_list.list_model_flags();\n        assert_eq!(models, Vec::<&str>::new());\n    }\n}\n"}, "1485d7489a91d08ee4085de888d61ffbdd54f361": {"t": 1785756952.601263, "n": 15505, "h": "dc880be96815db07", "c": "use gettextrs::gettext;\nuse nom::{\n    IResult, Parser,\n    bytes::complete::{tag, take_while1},\n    character::complete::multispace1,\n    combinator::{all_consuming, map},\n    multi::separated_list0,\n    sequence::delimited,\n};\n\nuse crate::formatget;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    model_string: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_id: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub id: String,\n    pub short_name: String,\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        let mut providers = Vec::new();\n        let provider_configs = [\n            (\"DEEPSEEK\", \"Deepseek\", DEEPSEEK_API_ENDPOINT),\n            (\"MISTRAL\", \"Mistral\", MISTRAL_API_ENDPOINT),\n            (\"OPENROUTER\", \"OpenRouter\", OPENROUTER_API_ENDPOINT),\n        ];\n        for (env_prefix, name, endpoint) in provider_configs.iter() {\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                let model_string =\n                    std::env::var(format!(\"{}_MODELS\", env_prefix)).unwrap_or_default();\n                providers.push(Provider {\n                    name: name.to_string(),\n                    api_key,\n                    endpoint: endpoint.to_string(),\n                    model_string,\n                });\n            }\n        }\n        Config {\n            providers,\n            default_model_id: std::env::var(\"DEFAULT_MODEL\").unwrap_or_default(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            if !provider.model_string.is_empty() {\n                models.extend(parse_model_config(\n                    &provider.model_string,\n                    &provider.api_key,\n                    &provider.endpoint,\n                )?);\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.id == cfg.default_model_id)\n            .ok_or_else(|| gettext(\"DEFAULT_MODEL not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.short_name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| format!(\"[{}]{}\", m.short_name, m.name))\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags_without_default(&self) -> Vec<String> {\n        let d = &self.default_model().id;\n        self.models\n            .iter()\n            .filter(|m| &m.id != d)\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .map(|m| m.short_name.clone())\n            .collect::<Vec<String>>()\n    }\n}\n\nfn parse_model_id(i: &str) -> IResult<&str, &str> {\n    take_while1(|c: char| c != '[' && !c.is_whitespace())(i)\n}\nfn is_short_name_char(c: char) -> bool {\n    c.is_alphanumeric() || c == '_' || c == '-'\n}\nfn is_human_name_char(c: char) -> bool {\n    c != ')'\n}\n// Parses a single model from this syntax: <model ID>[<short name>](<human-readable name>)\nfn parse_model<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Model> {\n    map(\n        (\n            parse_model_id,\n            delimited(tag(\"[\"), take_while1(is_short_name_char), tag(\"]\")),\n            delimited(tag(\"(\"), take_while1(is_human_name_char), tag(\")\")),\n        ),\n        |(id, short_name, name)| Model {\n            id: id.to_string(),\n            short_name: short_name.to_string(),\n            name: name.to_string(),\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        },\n    )\n    .parse(i)\n}\n\n// Parses a white-space separated list of models.\nfn parse_models<'a>(i: &'a str, api_key: &str, endpoint: &str) -> IResult<&'a str, Vec<Model>> {\n    all_consuming(separated_list0(multispace1, |i| {\n        parse_model(i, api_key, endpoint)\n    }))\n    .parse(i)\n}\n\n#[cfg(test)]\nmod parser_tests {\n    use super::*;\n\n    #[test]\n    fn test_parse_model_nom() {\n        let input = \"deepseek-1[short1](Deepseek 1)\";\n        let (rest, parsed) = parse_model(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(rest, \"\");\n        assert_eq!(\n            parsed,\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"short1\".to_string(),\n                name: \"Deepseek 1\".to_string(),\n                api_key: \"key\".to_string(),\n                endpoint: \"endpoint\".to_string()\n            }\n        );\n\n        let input2 = \"openrouter-2[o](OpenRouter 2)\";\n        let (_, parsed2) = parse_model(input2, \"\", \"\").unwrap();\n        assert_eq!(parsed2.id, \"openrouter-2\");\n        assert_eq!(parsed2.short_name, \"o\");\n        assert_eq!(parsed2.name, \"OpenRouter 2\");\n    }\n\n    #[test]\n    fn test_parse_model_nom_invalid() {\n        assert!(parse_model(\"invalidmodel\", \"\", \"\").is_err());\n        assert!(parse_model(\"id[short](name\", \"\", \"\").is_err());\n        assert!(parse_model(\"idshort](name)\", \"\", \"\").is_err());\n    }\n\n    #[test]\n    fn test_parse_models() {\n        let input = \"deepseek-1[short1](Deepseek 1) openrouter-2[o](OpenRouter 2)\";\n        let (rest, parsed) = parse_models(input, \"key\", \"endpoint\").unwrap();\n        assert_eq!(\n            parsed,\n            vec![\n                Model {\n                    id: \"deepseek-1\".to_string(),\n                    short_name: \"short1\".to_string(),\n                    name: \"Deepseek 1\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                },\n                Model {\n                    id: \"openrouter-2\".to_string(),\n                    short_name: \"o\".to_string(),\n                    name: \"OpenRouter 2\".to_string(),\n                    api_key: \"key\".to_string(),\n                    endpoint: \"endpoint\".to_string()\n                }\n            ]\n        );\n        assert_eq!(rest, \"\");\n    }\n}\n\nfn parse_model_config(\n    models_list: &str,\n    api_key: &str,\n    endpoint: &str,\n) -> Result<Vec<Model>, String> {\n    match parse_models(models_list, api_key, endpoint) {\n        Ok((_, models)) => Ok(models),\n        Err(e) => Err(formatget!(\"Invalid model syntax: {}\", e)),\n    }\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                id: \"deepseek-1\".to_string(),\n                short_name: \"d\".to_string(),\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                id: \"openrouter-2\".to_string(),\n                short_name: \"o\".to_string(),\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_id: \"\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                model_string: \"deepseek-1[short1](Deepseek 1) deepseek-2[short2](Deepseek 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"deepseek-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"deepseek-1\");\n        assert_eq!(model_list.models[0].short_name, \"short1\");\n        assert_eq!(model_list.models[0].name, \"Deepseek 1\");\n        assert_eq!(model_list.models[0].api_key, \"key1\");\n        assert_eq!(model_list.models[0].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.models[1].id, \"deepseek-2\");\n        assert_eq!(model_list.models[1].short_name, \"short2\");\n        assert_eq!(model_list.models[1].name, \"Deepseek 2\");\n        assert_eq!(model_list.models[1].api_key, \"key1\");\n        assert_eq!(model_list.models[1].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.default_model_name(), \"Deepseek 1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter 1) openrouter-2[p](OpenRouter 2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let model_list = ModelList::new(&cfg).expect(\"new()\");\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].id, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, \"o\".to_string());\n    }\n\n    #[test]\n    fn test_new_parses_missing_short_name() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2\".to_string(),\n            }],\n            default_model_id: \"openrouter-1\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"openrouter-2\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1[o](OpenRouter-1) openrouter-2[p](OpenRouter-2)\"\n                    .to_string(),\n            }],\n            default_model_id: \"unknown_model\".to_string(),\n        };\n        let err = ModelList::new(&cfg).unwrap_err();\n        assert!(err.contains(\"DEFAULT_MODEL\"), \"{}\", err);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"d\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"deepseek-1\");\n        assert_eq!(model.short_name, \"d\");\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n        assert_eq!(model.short_name, \"o\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_empty_and_valid_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"\".to_string(), \"clear_history\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"d\".to_string(), \"o\".to_string()];\n        let result = model_list.select_model(&flags);\n        let model = result.expect(\"select_model()\");\n        assert_eq!(model.id, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_model_flags(), vec![\"d\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"OpenRouter: -o\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            id: \"only-model\".to_string(),\n            short_name: \"o\".to_string(),\n            name: \"Only Model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_empty() {\n        let model_list = ModelList {\n            models: vec![],\n            default_model_index: 0,\n        };\n        let models = model_list.list_model_flags();\n        assert_eq!(models, Vec::<&str>::new());\n    }\n}\n"}, "3d0ad1a0d14f2c23c7a4e5de2cd350540dfa47b5": {"t": 1785756953.9833262, "n": 13742, "h": "e59da3d340f94faf", "c": "use gettextrs::gettext;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    model_string: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    providers: Vec<Provider>,\n    default_model_name: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub name: String,\n    pub short_name: Option<String>,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        let mut providers = Vec::new();\n        let provider_configs = [\n            (\"DEEPSEEK\", \"Deepseek\", DEEPSEEK_API_ENDPOINT),\n            (\"MISTRAL\", \"Mistral\", MISTRAL_API_ENDPOINT),\n            (\"OPENROUTER\", \"OpenRouter\", OPENROUTER_API_ENDPOINT),\n        ];\n        for (env_prefix, name, endpoint) in provider_configs.iter() {\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                let model_string =\n                    std::env::var(format!(\"{}_MODELS\", env_prefix)).unwrap_or_default();\n                providers.push(Provider {\n                    name: name.to_string(),\n                    api_key,\n                    endpoint: endpoint.to_string(),\n                    model_string,\n                });\n            }\n        }\n        Config {\n            providers,\n            default_model_name: std::env::var(\"DEFAULT_MODEL\").unwrap_or_default(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            if !provider.model_string.is_empty() {\n                models.extend(parse_model_config(\n                    &provider.model_string,\n                    &provider.api_key,\n                    &provider.endpoint,\n                ));\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        let default_model_index = models\n            .iter()\n            .position(|m| m.name == cfg.default_model_name)\n            .ok_or_else(|| gettext(\"DEFAULT_MODEL not found\"))?;\n        Ok(ModelList {\n            models,\n            default_model_index,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self\n                .models\n                .iter()\n                .find(|m| m.name == *f || m.short_name.as_deref() == Some(f))\n            {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let default_name = &self.default_model().name;\n        self.models\n            .iter()\n            .filter(|m| &m.name != default_name)\n            .map(|m| {\n                if let Some(short_name) = &m.short_name {\n                    format!(\"-{}|-{}\", m.name, short_name)\n                } else {\n                    format!(\"-{}\", m.name)\n                }\n            })\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .flat_map(|m| [Some(m.name.clone()), m.short_name.clone()])\n            .flatten()\n            .map(|s| s.to_string())\n            .collect::<Vec<String>>()\n    }\n}\n\nfn parse_model_config(models_list: &str, api_key: &str, endpoint: &str) -> Vec<Model> {\n    let model_re = regex::Regex::new(r\"([^(]*)\\(([^)]*)\\)\").unwrap();\n    let models = models_list\n        .split(' ')\n        .map(|s| s.trim())\n        .filter(|s| !s.is_empty())\n        .map(|s| match model_re.captures(s) {\n            Some(cap) => {\n                if cap.len() == 2 {\n                    (cap[1].to_string(), \"\".to_string())\n                } else if cap.len() == 3 {\n                    (cap[1].to_string(), cap[2].to_string())\n                } else {\n                    (s.to_string(), \"\".to_string())\n                }\n            }\n            None => (s.to_string(), \"\".to_string()),\n        })\n        .map(|(name, short)| (name, if short.is_empty() { None } else { Some(short) }))\n        .collect::<Vec<_>>();\n    models\n        .into_iter()\n        .map(|(name, short_name)| Model {\n            name,\n            short_name,\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        })\n        .collect()\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                name: \"deepseek-1\".to_string(),\n                short_name: Some(\"d\".to_string()),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                name: \"openrouter-2\".to_string(),\n                short_name: Some(\"o\".to_string()),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            providers: vec![],\n            default_model_name: \"\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                model_string: \"deepseek-1(d) deepseek-2(e)\".to_string(),\n            }],\n            default_model_name: \"deepseek-1\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"deepseek-1\");\n        assert_eq!(model_list.models[0].short_name, Some(\"d\".to_string()));\n        assert_eq!(model_list.models[0].api_key, \"key1\");\n        assert_eq!(model_list.models[0].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.models[1].name, \"deepseek-2\");\n        assert_eq!(model_list.models[1].short_name, Some(\"e\".to_string()));\n        assert_eq!(model_list.models[1].api_key, \"key1\");\n        assert_eq!(model_list.models[1].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.default_model_name(), \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1(o) openrouter-2(p)\".to_string(),\n            }],\n            default_model_name: \"openrouter-1\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, Some(\"o\".to_string()));\n    }\n\n    #[test]\n    fn test_new_parses_missing_short_name() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1(o) openrouter-2\".to_string(),\n            }],\n            default_model_name: \"openrouter-2\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, Some(\"o\".to_string()));\n        assert_eq!(model_list.models[1].name, \"openrouter-2\");\n        assert_eq!(model_list.models[1].short_name, None);\n        assert_eq!(model_list.default_model_name(), \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_new_unknown_default_model_fails() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1(o) openrouter-2\".to_string(),\n            }],\n            default_model_name: \"unknown_model\".to_string(),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"deepseek-1\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n        assert_eq!(model.short_name, Some(\"d\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n        assert_eq!(model.short_name, Some(\"d\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"openrouter-2\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n        assert_eq!(model.short_name, Some(\"o\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model_short_name() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n        assert_eq!(model.short_name, Some(\"o\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_empty_and_valid_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\n            \"\".to_string(),\n            \"clear_history\".to_string(),\n            \"openrouter-2\".to_string(),\n        ];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"deepseek-1\".to_string(), \"openrouter-2\".to_string()];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags(),\n            vec![\"deepseek-1\", \"d\", \"openrouter-2\", \"o\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"-openrouter-2|-o\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            name: \"only-model\".to_string(),\n            short_name: Some(\"o\".to_string()),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"only-model\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_empty() {\n        let model_list = ModelList {\n            models: vec![],\n            default_model_index: 0,\n        };\n        let models = model_list.list_model_flags();\n        assert_eq!(models, Vec::<&str>::new());\n    }\n}\n"}, "fa24e805efb64cf777e8288b7c04d883a17a3db6": {"t": 1785756955.3619916, "n": 12511, "h": "058f2726dc1d99b9", "c": "use gettextrs::gettext;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Provider {\n    name: String,\n    api_key: String,\n    endpoint: String,\n    model_string: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    providers: Vec<Provider>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub name: String,\n    pub short_name: Option<String>,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst MISTRAL_API_ENDPOINT: &str = \"https://api.mistral.ai/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        let mut providers = Vec::new();\n        let provider_configs = [\n            (\"DEEPSEEK\", \"Deepseek\", DEEPSEEK_API_ENDPOINT),\n            (\"MISTRAL\", \"Mistral\", MISTRAL_API_ENDPOINT),\n            (\"OPENROUTER\", \"OpenRouter\", OPENROUTER_API_ENDPOINT),\n        ];\n        for (env_prefix, name, endpoint) in provider_configs.iter() {\n            if let Ok(api_key) = std::env::var(format!(\"{}_API_KEY\", env_prefix)) {\n                let model_string =\n                    std::env::var(format!(\"{}_MODELS\", env_prefix)).unwrap_or_default();\n                providers.push(Provider {\n                    name: name.to_string(),\n                    api_key,\n                    endpoint: endpoint.to_string(),\n                    model_string,\n                });\n            }\n        }\n        Config { providers }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        for provider in &cfg.providers {\n            if !provider.model_string.is_empty() {\n                models.extend(parse_model_config(\n                    &provider.model_string,\n                    &provider.api_key,\n                    &provider.endpoint,\n                ));\n            }\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        Ok(ModelList {\n            models,\n            default_model_index: 0,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self\n                .models\n                .iter()\n                .find(|m| m.name == *f || m.short_name.as_deref() == Some(f))\n            {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let default_name = &self.default_model().name;\n        self.models\n            .iter()\n            .filter(|m| &m.name != default_name)\n            .map(|m| {\n                if let Some(short_name) = &m.short_name {\n                    format!(\"-{}|-{}\", m.name, short_name)\n                } else {\n                    format!(\"-{}\", m.name)\n                }\n            })\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .flat_map(|m| [Some(m.name.clone()), m.short_name.clone()])\n            .flatten()\n            .map(|s| s.to_string())\n            .collect::<Vec<String>>()\n    }\n}\n\nfn parse_model_config(models_list: &str, api_key: &str, endpoint: &str) -> Vec<Model> {\n    let model_re = regex::Regex::new(r\"([^(]*)\\(([^)]*)\\)\").unwrap();\n    let models = models_list\n        .split(' ')\n        .map(|s| s.trim())\n        .filter(|s| !s.is_empty())\n        .map(|s| match model_re.captures(s) {\n            Some(cap) => {\n                if cap.len() == 2 {\n                    (cap[1].to_string(), \"\".to_string())\n                } else if cap.len() == 3 {\n                    (cap[1].to_string(), cap[2].to_string())\n                } else {\n                    (s.to_string(), \"\".to_string())\n                }\n            }\n            None => (s.to_string(), \"\".to_string()),\n        })\n        .map(|(name, short)| (name, if short.is_empty() { None } else { Some(short) }))\n        .collect::<Vec<_>>();\n    models\n        .into_iter()\n        .map(|(name, short_name)| Model {\n            name,\n            short_name,\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        })\n        .collect()\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                name: \"deepseek-1\".to_string(),\n                short_name: Some(\"d\".to_string()),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                name: \"openrouter-2\".to_string(),\n                short_name: Some(\"o\".to_string()),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config { providers: vec![] };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"Deepseek\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n                model_string: \"deepseek-1(d) deepseek-2(e)\".to_string(),\n            }],\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"deepseek-1\");\n        assert_eq!(model_list.models[0].short_name, Some(\"d\".to_string()));\n        assert_eq!(model_list.models[0].api_key, \"key1\");\n        assert_eq!(model_list.models[0].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.models[1].name, \"deepseek-2\");\n        assert_eq!(model_list.models[1].short_name, Some(\"e\".to_string()));\n        assert_eq!(model_list.models[1].api_key, \"key1\");\n        assert_eq!(model_list.models[1].endpoint, DEEPSEEK_API_ENDPOINT);\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1(o) openrouter-2(p)\".to_string(),\n            }],\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, Some(\"o\".to_string()));\n    }\n\n    #[test]\n    fn test_new_parses_missing_short_name() {\n        let cfg = Config {\n            providers: vec![Provider {\n                name: \"OpenRouter\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n                model_string: \"openrouter-1(o) openrouter-2\".to_string(),\n            }],\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, Some(\"o\".to_string()));\n        assert_eq!(model_list.models[1].name, \"openrouter-2\");\n        assert_eq!(model_list.models[1].short_name, None);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"deepseek-1\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n        assert_eq!(model.short_name, Some(\"d\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n        assert_eq!(model.short_name, Some(\"d\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"openrouter-2\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n        assert_eq!(model.short_name, Some(\"o\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model_short_name() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n        assert_eq!(model.short_name, Some(\"o\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_empty_and_valid_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\n            \"\".to_string(),\n            \"clear_history\".to_string(),\n            \"openrouter-2\".to_string(),\n        ];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"deepseek-1\".to_string(), \"openrouter-2\".to_string()];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags(),\n            vec![\"deepseek-1\", \"d\", \"openrouter-2\", \"o\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"-openrouter-2|-o\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            name: \"only-model\".to_string(),\n            short_name: Some(\"o\".to_string()),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"only-model\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_empty() {\n        let model_list = ModelList {\n            models: vec![],\n            default_model_index: 0,\n        };\n        let models = model_list.list_model_flags();\n        assert_eq!(models, Vec::<&str>::new());\n    }\n}\n"}, "83e7344c4d57d6418ddba554eefd05009abf279a": {"t": 1785756956.8463845, "n": 12074, "h": "65bdfb53af498c29", "c": "use gettextrs::gettext;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    deepseek_models: Option<String>,\n    deepseek_api_key: Option<String>,\n    openrouter_models: Option<String>,\n    openrouter_api_key: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub name: String,\n    pub short_name: Option<String>,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        Config {\n            deepseek_models: std::env::var(\"DEEPSEEK_MODELS\").ok(),\n            deepseek_api_key: std::env::var(\"DEEPSEEK_API_KEY\").ok(),\n            openrouter_models: std::env::var(\"OPENROUTER_MODELS\").ok(),\n            openrouter_api_key: std::env::var(\"OPENROUTER_API_KEY\").ok(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        if let (Some(models_var), Some(api_key_var)) = (&cfg.deepseek_models, &cfg.deepseek_api_key)\n        {\n            models.extend(parse_model_config(\n                models_var,\n                api_key_var,\n                DEEPSEEK_API_ENDPOINT,\n            ));\n        }\n        if let (Some(models_var), Some(api_key_var)) =\n            (&cfg.openrouter_models, &cfg.openrouter_api_key)\n        {\n            models.extend(parse_model_config(\n                models_var,\n                api_key_var,\n                OPENROUTER_API_ENDPOINT,\n            ));\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        Ok(ModelList {\n            models,\n            default_model_index: 0,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self\n                .models\n                .iter()\n                .find(|m| m.name == *f || m.short_name.as_deref() == Some(f))\n            {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_model_flags_human_readable(&self) -> Vec<String> {\n        let default_name = &self.default_model().name;\n        self.models\n            .iter()\n            .filter(|m| &m.name != default_name)\n            .map(|m| {\n                if let Some(short_name) = &m.short_name {\n                    format!(\"-{}|-{}\", m.name, short_name)\n                } else {\n                    format!(\"-{}\", m.name)\n                }\n            })\n            .collect::<Vec<String>>()\n    }\n\n    pub fn list_model_flags(&self) -> Vec<String> {\n        self.models\n            .iter()\n            .flat_map(|m| [Some(m.name.clone()), m.short_name.clone()])\n            .flatten()\n            .map(|s| s.to_string())\n            .collect::<Vec<String>>()\n    }\n}\n\nfn parse_model_config(models_list: &str, api_key: &str, endpoint: &str) -> Vec<Model> {\n    let model_re = regex::Regex::new(r\"([^(]*)\\(([^)]*)\\)\").unwrap();\n    let models = models_list\n        .split(' ')\n        .map(|s| s.trim())\n        .filter(|s| !s.is_empty())\n        .map(|s| match model_re.captures(s) {\n            Some(cap) => {\n                if cap.len() == 2 {\n                    (cap[1].to_string(), \"\".to_string())\n                } else if cap.len() == 3 {\n                    (cap[1].to_string(), cap[2].to_string())\n                } else {\n                    (s.to_string(), \"\".to_string())\n                }\n            }\n            None => (s.to_string(), \"\".to_string()),\n        })\n        .map(|(name, short)| (name, if short.is_empty() { None } else { Some(short) }))\n        .collect::<Vec<_>>();\n    models\n        .into_iter()\n        .map(|(name, short_name)| Model {\n            name,\n            short_name,\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        })\n        .collect()\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                name: \"deepseek-1\".to_string(),\n                short_name: Some(\"d\".to_string()),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                name: \"openrouter-2\".to_string(),\n                short_name: Some(\"o\".to_string()),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            deepseek_models: None,\n            deepseek_api_key: None,\n            openrouter_models: None,\n            openrouter_api_key: None,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            deepseek_models: Some(\"deepseek-1(d) deepseek-2(e)\".to_string()),\n            deepseek_api_key: Some(\"key1\".to_string()),\n            openrouter_models: None,\n            openrouter_api_key: None,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"deepseek-1\");\n        assert_eq!(model_list.models[0].short_name, Some(\"d\".to_string()));\n        assert_eq!(model_list.models[0].api_key, \"key1\");\n        assert_eq!(model_list.models[0].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.models[1].name, \"deepseek-2\");\n        assert_eq!(model_list.models[1].short_name, Some(\"e\".to_string()));\n        assert_eq!(model_list.models[1].api_key, \"key1\");\n        assert_eq!(model_list.models[1].endpoint, DEEPSEEK_API_ENDPOINT);\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            deepseek_models: None,\n            deepseek_api_key: None,\n            openrouter_models: Some(\"openrouter-1(o) openrouter-2(p)\".to_string()),\n            openrouter_api_key: Some(\"key2\".to_string()),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, Some(\"o\".to_string()));\n    }\n\n    #[test]\n    fn test_new_parses_missing_short_name() {\n        let cfg = Config {\n            deepseek_models: None,\n            deepseek_api_key: None,\n            openrouter_models: Some(\"openrouter-1(o) openrouter-2\".to_string()),\n            openrouter_api_key: Some(\"key2\".to_string()),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"openrouter-1\");\n        assert_eq!(model_list.models[0].short_name, Some(\"o\".to_string()));\n        assert_eq!(model_list.models[1].name, \"openrouter-2\");\n        assert_eq!(model_list.models[1].short_name, None);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"deepseek-1\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n        assert_eq!(model.short_name, Some(\"d\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n        assert_eq!(model.short_name, Some(\"d\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"openrouter-2\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n        assert_eq!(model.short_name, Some(\"o\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model_short_name() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"o\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n        assert_eq!(model.short_name, Some(\"o\".to_string()));\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_empty_and_valid_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\n            \"\".to_string(),\n            \"clear_history\".to_string(),\n            \"openrouter-2\".to_string(),\n        ];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\"deepseek-1\".to_string(), \"openrouter-2\".to_string()];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags(),\n            vec![\"deepseek-1\", \"d\", \"openrouter-2\", \"o\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_human_readable_excludes_default() {\n        let model_list = setup_model_list();\n        assert_eq!(\n            model_list.list_model_flags_human_readable(),\n            vec![\"-openrouter-2|-o\"]\n        );\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            name: \"only-model\".to_string(),\n            short_name: Some(\"o\".to_string()),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_model_flags(), vec![\"only-model\", \"o\"]);\n    }\n\n    #[test]\n    fn test_list_models_empty() {\n        let model_list = ModelList {\n            models: vec![],\n            default_model_index: 0,\n        };\n        let models = model_list.list_model_flags();\n        assert_eq!(models, Vec::<&str>::new());\n    }\n}\n"}, "b0ee16df6a092dabd44b3db4a6492c2382aad795": {"t": 1785756958.0075393, "n": 10098, "h": "a64c677a59eaf777", "c": "use gettextrs::gettext;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    deepseek_models: Option<String>,\n    deepseek_api_key: Option<String>,\n    openrouter_models: Option<String>,\n    openrouter_api_key: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        Config {\n            deepseek_models: std::env::var(\"DEEPSEEK_MODELS\").ok(),\n            deepseek_api_key: std::env::var(\"DEEPSEEK_API_KEY\").ok(),\n            openrouter_models: std::env::var(\"OPENROUTER_MODELS\").ok(),\n            openrouter_api_key: std::env::var(\"OPENROUTER_API_KEY\").ok(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        if let (Some(models_var), Some(api_key_var)) = (&cfg.deepseek_models, &cfg.deepseek_api_key)\n        {\n            models.extend(parse_model_config(\n                models_var,\n                api_key_var,\n                DEEPSEEK_API_ENDPOINT,\n            ));\n        }\n        if let (Some(models_var), Some(api_key_var)) =\n            (&cfg.openrouter_models, &cfg.openrouter_api_key)\n        {\n            models.extend(parse_model_config(\n                models_var,\n                api_key_var,\n                OPENROUTER_API_ENDPOINT,\n            ));\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        Ok(ModelList {\n            models,\n            default_model_index: 0,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_models(&self) -> Vec<&str> {\n        self.models\n            .iter()\n            .map(|m| m.name.as_str())\n            .collect::<Vec<&str>>()\n    }\n}\n\nfn parse_model_config(models_list: &str, api_key: &str, endpoint: &str) -> Vec<Model> {\n    let models = models_list\n        .split(' ')\n        .map(|s| s.trim().to_string())\n        .filter(|s| !s.is_empty())\n        .collect::<Vec<_>>();\n    models\n        .into_iter()\n        .map(|name| Model {\n            name,\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        })\n        .collect()\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                name: \"deepseek-1\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                name: \"openrouter-2\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            deepseek_models: None,\n            deepseek_api_key: None,\n            openrouter_models: None,\n            openrouter_api_key: None,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            deepseek_models: Some(\"deepseek-1 deepseek-2\".to_string()),\n            deepseek_api_key: Some(\"key1\".to_string()),\n            openrouter_models: None,\n            openrouter_api_key: None,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"deepseek-1\");\n        assert_eq!(model_list.models[0].api_key, \"key1\");\n        assert_eq!(model_list.models[0].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.models[1].name, \"deepseek-2\");\n        assert_eq!(model_list.models[1].api_key, \"key1\");\n        assert_eq!(model_list.models[1].endpoint, DEEPSEEK_API_ENDPOINT);\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            deepseek_models: None,\n            deepseek_api_key: None,\n            openrouter_models: Some(\"openrouter-1 openrouter-2\".to_string()),\n            openrouter_api_key: Some(\"key2\".to_string()),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"openrouter-1\");\n        assert_eq!(model_list.models[0].api_key, \"key2\");\n        assert_eq!(model_list.models[0].endpoint, OPENROUTER_API_ENDPOINT);\n        assert_eq!(model_list.models[1].name, \"openrouter-2\");\n        assert_eq!(model_list.models[1].api_key, \"key2\");\n        assert_eq!(model_list.models[1].endpoint, OPENROUTER_API_ENDPOINT);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"deepseek-1\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_with_model_prefix() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"openrouter-2\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_with_only_model_prefix() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"openrouter-2\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_leading_and_trailing_spaces() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"openrouter-2\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_no_query_after_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"deepseek-1\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_empty_and_valid_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\n            \"\".to_string(),\n            \"clear_history\".to_string(),\n            \"openrouter-2\".to_string(),\n        ];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\n            \"\".to_string(),\n            \"deepseek-1\".to_string(),\n            \"openrouter-2\".to_string(),\n        ];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let model_list = setup_model_list();\n        assert_eq!(model_list.list_models(), vec![\"deepseek-1\", \"openrouter-2\"]);\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            name: \"only-model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        assert_eq!(model_list.list_models(), vec![\"only-model\"]);\n    }\n\n    #[test]\n    fn test_list_models_empty() {\n        let model_list = ModelList {\n            models: vec![],\n            default_model_index: 0,\n        };\n        let models = model_list.list_models();\n        assert_eq!(models, Vec::<&str>::new());\n    }\n}\n"}, "bc84e9ff07bf239ed903da044de1a24889b2b8ab": {"t": 1785756959.5335178, "n": 10132, "h": "b4982766d17798ca", "c": "use gettextrs::gettext;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    deepseek_models: Option<String>,\n    deepseek_api_key: Option<String>,\n    openrouter_models: Option<String>,\n    openrouter_api_key: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        Config {\n            deepseek_models: std::env::var(\"DEEPSEEK_MODELS\").ok(),\n            deepseek_api_key: std::env::var(\"DEEPSEEK_API_KEY\").ok(),\n            openrouter_models: std::env::var(\"OPENROUTER_MODELS\").ok(),\n            openrouter_api_key: std::env::var(\"OPENROUTER_API_KEY\").ok(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        if let (Some(models_var), Some(api_key_var)) = (&cfg.deepseek_models, &cfg.deepseek_api_key)\n        {\n            models.extend(parse_model_config(\n                models_var,\n                api_key_var,\n                DEEPSEEK_API_ENDPOINT,\n            ));\n        }\n        if let (Some(models_var), Some(api_key_var)) =\n            (&cfg.openrouter_models, &cfg.openrouter_api_key)\n        {\n            models.extend(parse_model_config(\n                models_var,\n                api_key_var,\n                OPENROUTER_API_ENDPOINT,\n            ));\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        Ok(ModelList {\n            models,\n            default_model_index: 0,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the flags contains a model name, it selects the last specified model.\n    // Otherwise, it returns the default model.\n    pub fn select_model(&self, flags: &[String]) -> Result<&Model, String> {\n        // Find the last flag that matches a model name\n        let mut selected: Option<&Model> = None;\n        for f in flags.iter() {\n            if let Some(model) = self.models.iter().find(|m| m.name == *f) {\n                selected = Some(model);\n            }\n        }\n        if let Some(model) = selected {\n            return Ok(model);\n        }\n        Ok(self.default_model())\n    }\n\n    pub fn list_models(&self) -> String {\n        self.models\n            .iter()\n            .map(|m| m.name.as_str())\n            .collect::<Vec<&str>>()\n            .join(\" \")\n    }\n}\n\nfn parse_model_config(models_list: &str, api_key: &str, endpoint: &str) -> Vec<Model> {\n    let models = models_list\n        .split(' ')\n        .map(|s| s.trim().to_string())\n        .filter(|s| !s.is_empty())\n        .collect::<Vec<_>>();\n    models\n        .into_iter()\n        .map(|name| Model {\n            name,\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        })\n        .collect()\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                name: \"deepseek-1\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                name: \"openrouter-2\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            deepseek_models: None,\n            deepseek_api_key: None,\n            openrouter_models: None,\n            openrouter_api_key: None,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            deepseek_models: Some(\"deepseek-1 deepseek-2\".to_string()),\n            deepseek_api_key: Some(\"key1\".to_string()),\n            openrouter_models: None,\n            openrouter_api_key: None,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"deepseek-1\");\n        assert_eq!(model_list.models[0].api_key, \"key1\");\n        assert_eq!(model_list.models[0].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.models[1].name, \"deepseek-2\");\n        assert_eq!(model_list.models[1].api_key, \"key1\");\n        assert_eq!(model_list.models[1].endpoint, DEEPSEEK_API_ENDPOINT);\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            deepseek_models: None,\n            deepseek_api_key: None,\n            openrouter_models: Some(\"openrouter-1 openrouter-2\".to_string()),\n            openrouter_api_key: Some(\"key2\".to_string()),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"openrouter-1\");\n        assert_eq!(model_list.models[0].api_key, \"key2\");\n        assert_eq!(model_list.models[0].endpoint, OPENROUTER_API_ENDPOINT);\n        assert_eq!(model_list.models[1].name, \"openrouter-2\");\n        assert_eq!(model_list.models[1].api_key, \"key2\");\n        assert_eq!(model_list.models[1].endpoint, OPENROUTER_API_ENDPOINT);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"deepseek-1\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_with_model_prefix() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"openrouter-2\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"unknownmodel\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_with_only_model_prefix() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"openrouter-2\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_leading_and_trailing_spaces() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"openrouter-2\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_no_query_after_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(&vec![\"deepseek-1\".to_string()]);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_empty_and_valid_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\n            \"\".to_string(),\n            \"clear_history\".to_string(),\n            \"openrouter-2\".to_string(),\n        ];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_flags_containing_multiple_model_names() {\n        let model_list = setup_model_list();\n        let flags = vec![\n            \"\".to_string(),\n            \"deepseek-1\".to_string(),\n            \"openrouter-2\".to_string(),\n        ];\n        let result = model_list.select_model(&flags);\n        assert!(result.is_ok());\n        let model = result.unwrap();\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let models_str = setup_model_list().list_models();\n        assert!(models_str.eq(\"deepseek-1 openrouter-2\"));\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            name: \"only-model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        let models_str = model_list.list_models();\n        assert_eq!(models_str, \"only-model\");\n    }\n\n    #[test]\n    fn test_list_models_empty() {\n        let model_list = ModelList {\n            models: vec![],\n            default_model_index: 0,\n        };\n        let models_str = model_list.list_models();\n        assert_eq!(models_str, \"\");\n    }\n}\n"}, "657a6b85b209c1c7b41337203898d638847b166a": {"t": 1785756961.141877, "n": 9894, "h": "4b39c6bc5ef96e52", "c": "use gettextrs::gettext;\n\nuse crate::formatget;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Config {\n    deepseek_models: Option<String>,\n    deepseek_api_key: Option<String>,\n    openrouter_models: Option<String>,\n    openrouter_api_key: Option<String>,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct Model {\n    pub name: String,\n    pub api_key: String,\n    pub endpoint: String,\n}\n\n#[derive(Debug, Clone, PartialEq, Eq)]\npub struct ModelList {\n    models: Vec<Model>,\n    default_model_index: usize,\n}\n\nconst DEEPSEEK_API_ENDPOINT: &str = \"https://api.deepseek.com/v1/chat/completions\";\nconst OPENROUTER_API_ENDPOINT: &str = \"https://openrouter.ai/api/v1/chat/completions\";\n\nimpl Config {\n    pub fn from_env() -> Self {\n        Config {\n            deepseek_models: std::env::var(\"DEEPSEEK_MODELS\").ok(),\n            deepseek_api_key: std::env::var(\"DEEPSEEK_API_KEY\").ok(),\n            openrouter_models: std::env::var(\"OPENROUTER_MODELS\").ok(),\n            openrouter_api_key: std::env::var(\"OPENROUTER_API_KEY\").ok(),\n        }\n    }\n}\n\nimpl ModelList {\n    pub fn new(cfg: &Config) -> Result<Self, String> {\n        let mut models = Vec::new();\n        if let (Some(models_var), Some(api_key_var)) = (&cfg.deepseek_models, &cfg.deepseek_api_key)\n        {\n            models.extend(parse_model_config(\n                models_var,\n                api_key_var,\n                DEEPSEEK_API_ENDPOINT,\n            ));\n        }\n        if let (Some(models_var), Some(api_key_var)) =\n            (&cfg.openrouter_models, &cfg.openrouter_api_key)\n        {\n            models.extend(parse_model_config(\n                models_var,\n                api_key_var,\n                OPENROUTER_API_ENDPOINT,\n            ));\n        }\n        if models.is_empty() {\n            return Err(gettext(\"Missing API keys or model configuration\"));\n        }\n        Ok(ModelList {\n            models,\n            default_model_index: 0,\n        })\n    }\n\n    fn default_model(&self) -> &Model {\n        self.models.get(self.default_model_index).unwrap()\n    }\n\n    pub fn default_model_name(&self) -> &str {\n        &self.default_model().name\n    }\n\n    // Selects a model based on the query.\n    // If the query starts with '-model', it selects the specified model.\n    // Otherwise, it returns the default model.\n    // Returns the query without the model prefix and the selected model.\n    pub fn select_model(&self, query: &str) -> Result<(String, &Model), String> {\n        let trimmed = query.trim_start();\n        if !trimmed.starts_with('-') {\n            // No model specified, use default\n            return Ok((query.to_string(), self.default_model()));\n        }\n        let mut parts = trimmed.splitn(2, ' ');\n        let first_word = parts.next().unwrap_or(\"\").trim_start_matches('-');\n        let rest_query = parts.next().unwrap_or(\"\").trim_start().to_string();\n        let model_result = self\n            .models\n            .iter()\n            .find(|model| model.name == first_word)\n            .ok_or_else(|| formatget!(\"Unknown model. Available models: {}\", self.list_models()))?;\n        Ok((rest_query, model_result))\n    }\n\n    pub fn list_models(&self) -> String {\n        self.models\n            .iter()\n            .map(|m| m.name.as_str())\n            .collect::<Vec<&str>>()\n            .join(\" \")\n    }\n}\n\nfn parse_model_config(models_list: &str, api_key: &str, endpoint: &str) -> Vec<Model> {\n    let models = models_list\n        .split(' ')\n        .map(|s| s.trim().to_string())\n        .filter(|s| !s.is_empty())\n        .collect::<Vec<_>>();\n    models\n        .into_iter()\n        .map(|name| Model {\n            name,\n            api_key: api_key.to_string(),\n            endpoint: endpoint.to_string(),\n        })\n        .collect()\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n\n    fn setup_model_list() -> ModelList {\n        let models = vec![\n            Model {\n                name: \"deepseek-1\".to_string(),\n                api_key: \"key1\".to_string(),\n                endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n            },\n            Model {\n                name: \"openrouter-2\".to_string(),\n                api_key: \"key2\".to_string(),\n                endpoint: OPENROUTER_API_ENDPOINT.to_string(),\n            },\n        ];\n        ModelList {\n            models,\n            default_model_index: 0,\n        }\n    }\n\n    #[test]\n    fn test_new_returns_error_when_no_env_vars() {\n        let cfg = Config {\n            deepseek_models: None,\n            deepseek_api_key: None,\n            openrouter_models: None,\n            openrouter_api_key: None,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_err());\n        assert_eq!(\n            result.unwrap_err(),\n            \"Missing API keys or model configuration\"\n        );\n    }\n\n    #[test]\n    fn test_new_parses_deepseek_env_vars() {\n        let cfg = Config {\n            deepseek_models: Some(\"deepseek-1 deepseek-2\".to_string()),\n            deepseek_api_key: Some(\"key1\".to_string()),\n            openrouter_models: None,\n            openrouter_api_key: None,\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"deepseek-1\");\n        assert_eq!(model_list.models[0].api_key, \"key1\");\n        assert_eq!(model_list.models[0].endpoint, DEEPSEEK_API_ENDPOINT);\n        assert_eq!(model_list.models[1].name, \"deepseek-2\");\n        assert_eq!(model_list.models[1].api_key, \"key1\");\n        assert_eq!(model_list.models[1].endpoint, DEEPSEEK_API_ENDPOINT);\n    }\n\n    #[test]\n    fn test_new_parses_openrouter_env_vars() {\n        let cfg = Config {\n            deepseek_models: None,\n            deepseek_api_key: None,\n            openrouter_models: Some(\"openrouter-1 openrouter-2\".to_string()),\n            openrouter_api_key: Some(\"key2\".to_string()),\n        };\n        let result = ModelList::new(&cfg);\n        assert!(result.is_ok());\n        let model_list = result.unwrap();\n        assert_eq!(model_list.models.len(), 2);\n        assert_eq!(model_list.models[0].name, \"openrouter-1\");\n        assert_eq!(model_list.models[0].api_key, \"key2\");\n        assert_eq!(model_list.models[0].endpoint, OPENROUTER_API_ENDPOINT);\n        assert_eq!(model_list.models[1].name, \"openrouter-2\");\n        assert_eq!(model_list.models[1].api_key, \"key2\");\n        assert_eq!(model_list.models[1].endpoint, OPENROUTER_API_ENDPOINT);\n    }\n\n    #[test]\n    fn test_select_model_with_empty_query() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(\"\");\n        assert!(result.is_ok());\n        let (query, model) = result.unwrap();\n        assert_eq!(query, \"\");\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_default() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(\"hello world\");\n        assert!(result.is_ok());\n        let (query, model) = result.unwrap();\n        assert_eq!(query, \"hello world\");\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_select_model_with_model_prefix() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(\"-openrouter-2 test prompt\");\n        assert!(result.is_ok());\n        let (query, model) = result.unwrap();\n        assert_eq!(query, \"test prompt\");\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_unknown_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(\"-unknownmodel some query\");\n        assert!(result.is_err());\n        let err = result.unwrap_err();\n        assert!(err.contains(\"Unknown model\"));\n        assert!(err.contains(\"deepseek-1\"));\n        assert!(err.contains(\"openrouter-2\"));\n    }\n\n    #[test]\n    fn test_select_model_with_only_model_prefix() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(\"-openrouter-2\");\n        assert!(result.is_ok());\n        let (query, model) = result.unwrap();\n        assert_eq!(query, \"\");\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_leading_and_trailing_spaces() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(\"   -openrouter-2   prompt   \");\n        assert!(result.is_ok());\n        let (query, model) = result.unwrap();\n        assert_eq!(query, \"prompt   \");\n        assert_eq!(model.name, \"openrouter-2\");\n    }\n\n    #[test]\n    fn test_select_model_with_no_query_after_model() {\n        let model_list = setup_model_list();\n        let result = model_list.select_model(\"   -deepseek-1   \");\n        assert!(result.is_ok());\n        let (query, model) = result.unwrap();\n        assert_eq!(query, \"\");\n        assert_eq!(model.name, \"deepseek-1\");\n    }\n\n    #[test]\n    fn test_list_models_returns_all_model_names() {\n        let models_str = setup_model_list().list_models();\n        assert!(models_str.eq(\"deepseek-1 openrouter-2\"));\n    }\n\n    #[test]\n    fn test_list_models_with_single_model() {\n        let models = vec![Model {\n            name: \"only-model\".to_string(),\n            api_key: \"key\".to_string(),\n            endpoint: DEEPSEEK_API_ENDPOINT.to_string(),\n        }];\n        let model_list = ModelList {\n            models,\n            default_model_index: 0,\n        };\n        let models_str = model_list.list_models();\n        assert_eq!(models_str, \"only-model\");\n    }\n\n    #[test]\n    fn test_list_models_empty() {\n        let model_list = ModelList {\n            models: vec![],\n            default_model_index: 0,\n        };\n        let models_str = model_list.list_models();\n        assert_eq!(models_str, \"\");\n    }\n}\n"}}, "ai/src/main.rs": {"6572516a1786d0c8e65b6c5961ac78e2aa679b9e": {"t": 1785745831.8445892, "n": 43405, "h": "9d41782579d2a5e3", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n    config: &Config,\n    channel_name: &str,\n) -> Result<String, String> {\n    let timeout_seconds = config.get_timeout(model, channel_name);\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: config.get_max_tokens(model) as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::AskAgent {\n                query: weather::weather_error_prompt(),\n                extra_history: formatget!(\"Weather API error: {}\", err),\n            }),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::AskAgent {\n                query: weather::weather_error_prompt(),\n                extra_history: formatget!(\"Weather API error: {}\", err),\n            }),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| {\n            input.config.get_channel_model_temperature(\n                &input.receiver,\n                &input.model.provider,\n                &input.model.id,\n            )\n        })\n        .or(input.model.temperature);\n\n    let result = &call_api(\n        &input.model,\n        &prompt,\n        &temperature,\n        &input.config,\n        &input.receiver,\n    )?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n\n    let result = result.trim();\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test-channel\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test channel prompt\"\n\n[channels.\"#test-channel\".models.litellm.\"test-model\"]\ntemperature = 0.8\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp =\n            config.get_channel_model_temperature(\"#test-channel\", \"litellm\", \"test-model\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp =\n            config.get_channel_model_temperature(\"#unknown\", \"litellm\", \"test-model\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test prompt\"\n\n[channels.\"#test\".models.litellm.\"test-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let provider = \"litellm\";\n        let model_id = \"test-model\";\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", provider, model_id));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#unknown\", provider, model_id));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\".models.deepseek.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let provider = \"deepseek\";\n        let model_id = \"default-model\";\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", provider, model_id));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n\n    #[test]\n    fn test_model_temperature_fallback() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.deepseek.\"other-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Model temperature is used when no flag or channel-model temperature\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.provider, &model.id))\n            .or(model.temperature);\n\n        assert_eq!(temperature, Some(0.3));\n    }\n\n    #[test]\n    fn test_temperature_precedence_flag_over_channel_over_model() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.deepseek.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Flag overrides channel-model and model\n        let flags = [\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.9));\n\n        // Channel-model overrides model\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.5));\n\n        // Model is used when no channel-model config\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.3));\n    }\n}\n"}, "f3bec27527acdc809042b4335c5c6189213fab29": {"t": 1785756916.484607, "n": 43405, "h": "9d41782579d2a5e3", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n    config: &Config,\n    channel_name: &str,\n) -> Result<String, String> {\n    let timeout_seconds = config.get_timeout(model, channel_name);\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: config.get_max_tokens(model) as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::AskAgent {\n                query: weather::weather_error_prompt(),\n                extra_history: formatget!(\"Weather API error: {}\", err),\n            }),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::AskAgent {\n                query: weather::weather_error_prompt(),\n                extra_history: formatget!(\"Weather API error: {}\", err),\n            }),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| {\n            input.config.get_channel_model_temperature(\n                &input.receiver,\n                &input.model.provider,\n                &input.model.id,\n            )\n        })\n        .or(input.model.temperature);\n\n    let result = &call_api(\n        &input.model,\n        &prompt,\n        &temperature,\n        &input.config,\n        &input.receiver,\n    )?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n\n    let result = result.trim();\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test-channel\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test channel prompt\"\n\n[channels.\"#test-channel\".models.litellm.\"test-model\"]\ntemperature = 0.8\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp =\n            config.get_channel_model_temperature(\"#test-channel\", \"litellm\", \"test-model\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp =\n            config.get_channel_model_temperature(\"#unknown\", \"litellm\", \"test-model\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test prompt\"\n\n[channels.\"#test\".models.litellm.\"test-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let provider = \"litellm\";\n        let model_id = \"test-model\";\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", provider, model_id));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#unknown\", provider, model_id));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\".models.deepseek.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let provider = \"deepseek\";\n        let model_id = \"default-model\";\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", provider, model_id));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n\n    #[test]\n    fn test_model_temperature_fallback() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.deepseek.\"other-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Model temperature is used when no flag or channel-model temperature\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.provider, &model.id))\n            .or(model.temperature);\n\n        assert_eq!(temperature, Some(0.3));\n    }\n\n    #[test]\n    fn test_temperature_precedence_flag_over_channel_over_model() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.deepseek.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Flag overrides channel-model and model\n        let flags = [\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.9));\n\n        // Channel-model overrides model\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.5));\n\n        // Model is used when no channel-model config\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.3));\n    }\n}\n"}, "dc40b2ff3dac2a3d17da1464591d9d77c382ebcc": {"t": 1785756919.350165, "n": 43372, "h": "82b4a1aca80efd3e", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n    config: &Config,\n    channel_name: &str,\n) -> Result<String, String> {\n    let timeout_seconds = config.get_timeout(model, channel_name);\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: config.get_max_tokens(model) as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::AskAgent {\n                query: weather::weather_error_prompt(),\n                extra_history: formatget!(\"Weather API error: {}\", err),\n            }),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::AskAgent {\n                query: weather::weather_error_prompt(),\n                extra_history: formatget!(\"Weather API error: {}\", err),\n            }),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| {\n            input.config.get_channel_model_temperature(\n                &input.receiver,\n                &input.model.provider,\n                &input.model.id,\n            )\n        })\n        .or(input.model.temperature);\n\n    let result = &call_api(\n        &input.model,\n        &prompt,\n        &temperature,\n        &input.config,\n        &input.receiver,\n    )?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test-channel\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test channel prompt\"\n\n[channels.\"#test-channel\".models.litellm.\"test-model\"]\ntemperature = 0.8\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp =\n            config.get_channel_model_temperature(\"#test-channel\", \"litellm\", \"test-model\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp =\n            config.get_channel_model_temperature(\"#unknown\", \"litellm\", \"test-model\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test prompt\"\n\n[channels.\"#test\".models.litellm.\"test-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let provider = \"litellm\";\n        let model_id = \"test-model\";\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", provider, model_id));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#unknown\", provider, model_id));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\".models.deepseek.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let provider = \"deepseek\";\n        let model_id = \"default-model\";\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", provider, model_id));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n\n    #[test]\n    fn test_model_temperature_fallback() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.deepseek.\"other-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Model temperature is used when no flag or channel-model temperature\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.provider, &model.id))\n            .or(model.temperature);\n\n        assert_eq!(temperature, Some(0.3));\n    }\n\n    #[test]\n    fn test_temperature_precedence_flag_over_channel_over_model() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.deepseek.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Flag overrides channel-model and model\n        let flags = [\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.9));\n\n        // Channel-model overrides model\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.5));\n\n        // Model is used when no channel-model config\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.3));\n    }\n}\n"}, "ed0f7a7bace554eededb0901383945f0d2d66448": {"t": 1785756920.8452506, "n": 43090, "h": "748b5ebcfb1ed66b", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n    config: &Config,\n    channel_name: &str,\n) -> Result<String, String> {\n    let timeout_seconds = config.get_timeout(model, channel_name);\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: config.get_max_tokens(model) as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| {\n            input.config.get_channel_model_temperature(\n                &input.receiver,\n                &input.model.provider,\n                &input.model.id,\n            )\n        })\n        .or(input.model.temperature);\n\n    let result = &call_api(\n        &input.model,\n        &prompt,\n        &temperature,\n        &input.config,\n        &input.receiver,\n    )?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test-channel\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test channel prompt\"\n\n[channels.\"#test-channel\".models.litellm.\"test-model\"]\ntemperature = 0.8\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp =\n            config.get_channel_model_temperature(\"#test-channel\", \"litellm\", \"test-model\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp =\n            config.get_channel_model_temperature(\"#unknown\", \"litellm\", \"test-model\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test prompt\"\n\n[channels.\"#test\".models.litellm.\"test-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let provider = \"litellm\";\n        let model_id = \"test-model\";\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", provider, model_id));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#unknown\", provider, model_id));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\".models.deepseek.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let provider = \"deepseek\";\n        let model_id = \"default-model\";\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", provider, model_id));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n\n    #[test]\n    fn test_model_temperature_fallback() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.deepseek.\"other-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Model temperature is used when no flag or channel-model temperature\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.provider, &model.id))\n            .or(model.temperature);\n\n        assert_eq!(temperature, Some(0.3));\n    }\n\n    #[test]\n    fn test_temperature_precedence_flag_over_channel_over_model() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.deepseek.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Flag overrides channel-model and model\n        let flags = [\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.9));\n\n        // Channel-model overrides model\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.5));\n\n        // Model is used when no channel-model config\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.provider, &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.3));\n    }\n}\n"}, "deb1e1c932e5c063942af53b2f692e7dd7658e13": {"t": 1785756922.327591, "n": 42746, "h": "ac4476875579624c", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n    config: &Config,\n    channel_name: &str,\n) -> Result<String, String> {\n    let timeout_seconds = config.get_timeout(model, channel_name);\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: config.get_max_tokens(model) as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| {\n            input\n                .config\n                .get_channel_model_temperature(&input.receiver, &input.model.id)\n        })\n        .or(input.model.temperature);\n\n    let result = &call_api(\n        &input.model,\n        &prompt,\n        &temperature,\n        &input.config,\n        &input.receiver,\n    )?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test-channel\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test channel prompt\"\n\n[channels.\"#test-channel\".models.\"test-model\"]\ntemperature = 0.8\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_model_temperature(\"#test-channel\", \"test-model\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_model_temperature(\"#unknown\", \"test-model\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\"]\ndefault_model = \"test-model\"\nsystem_prompt = \"Test prompt\"\n\n[channels.\"#test\".models.\"test-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let model_id = \"test-model\";\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", model_id));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#unknown\", model_id));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels.\"#test\".models.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let model_id = \"default-model\";\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", model_id));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n\n    #[test]\n    fn test_model_temperature_fallback() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.\"other-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Model temperature is used when no flag or channel-model temperature\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.id))\n            .or(model.temperature);\n\n        assert_eq!(temperature, Some(0.3));\n    }\n\n    #[test]\n    fn test_temperature_precedence_flag_over_channel_over_model() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels.\"#test\".models.\"default-model\"]\ntemperature = 0.5\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Flag overrides channel-model and model\n        let flags = [\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.9));\n\n        // Channel-model overrides model\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#test\", &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.5));\n\n        // Model is used when no channel-model config\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_model_temperature(\"#other\", &model.id))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.3));\n    }\n}\n"}, "b89d714b530ae0d9d4ed73ce1e24f73ad29df59f": {"t": 1785756924.0598438, "n": 42141, "h": "7338ed8c957b604d", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n    config: &Config,\n) -> Result<String, String> {\n    let timeout_seconds = config.get_timeout(model);\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: config.get_max_tokens(model) as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver))\n        .or(input.model.temperature);\n\n    let result = &call_api(&input.model, &prompt, &temperature, &input.config)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n\n    #[test]\n    fn test_model_temperature_fallback() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Model temperature is used when no flag or channel temperature\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#other\"))\n            .or(model.temperature);\n\n        assert_eq!(temperature, Some(0.3));\n    }\n\n    #[test]\n    fn test_temperature_precedence_flag_over_channel_over_model() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\", temperature = 0.3 }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).expect(\"ModelList::new()\");\n        let model = models\n            .select_model_for_channel(&[], \"default-model\")\n            .unwrap();\n\n        // Flag overrides channel and model\n        let flags = [\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.9));\n\n        // Channel overrides model\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.5));\n\n        // Model is used when no channel config\n        let flags: [&String; 0] = [];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#other\"))\n            .or(model.temperature);\n        assert_eq!(temperature, Some(0.3));\n    }\n}\n"}, "28e2a96b35926edb405ff5dddc6833300dcbb988": {"t": 1785756925.7151551, "n": 37992, "h": "d0b2174f26fb5363", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n    config: &Config,\n) -> Result<String, String> {\n    let timeout_seconds = config.get_timeout(model);\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: config.get_max_tokens(model) as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature, &input.config)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "8b877592ad4abe70c1f0ff117c79631e4a56ceaa": {"t": 1785756928.9291728, "n": 38142, "h": "c5d98d84abe29f12", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS,\n    MAX_TOKENS_WITH_REASONING, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n    timeout_seconds: u64,\n) -> Result<String, String> {\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: if model.reasoning {\n            MAX_TOKENS_WITH_REASONING as i32\n        } else {\n            MAX_TOKENS as i32\n        },\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let timeout_seconds = input.config.get_timeout(&input.model);\n    let result = &call_api(&input.model, &prompt, &temperature, timeout_seconds)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "b31f3d72eb8375b38a08a3255f26fe995059332a": {"t": 1785756930.4325187, "n": 38151, "h": "e6995f2a61e0bd1c", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS,\n    MAX_TOKENS_WITH_REASONING, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n    timeout_seconds: u64,\n) -> Result<String, String> {\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: if model.reasoning {\n            MAX_TOKENS_WITH_REASONING as i32\n        } else {\n            MAX_TOKENS as i32\n        },\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let timeout_seconds = input.config.get_timeout(input.model.reasoning);\n    let result = &call_api(&input.model, &prompt, &temperature, timeout_seconds)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "67222dfb7f511e6d911762dce981e2130561a169": {"t": 1785756932.0207007, "n": 38098, "h": "47b74d20538df5c4", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS,\n    MAX_TOKENS_WITH_REASONING, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let timeout_seconds = if model.reasoning { 40 } else { 20 };\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(timeout_seconds))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: if model.reasoning {\n            MAX_TOKENS_WITH_REASONING as i32\n        } else {\n            MAX_TOKENS as i32\n        },\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| {\n        if e.is_timeout() {\n            formatget!(\"API error: Request timed out (%d seconds)\", timeout_seconds)\n        } else if e.is_connect() {\n            formatget!(\"API error: Failed to connect to server: {}\", e)\n        } else {\n            formatget!(\"API error: {}\", e)\n        }\n    })?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "ee9fc0906f8877f826b98c84dcd6fa39dd487183": {"t": 1785756933.502336, "n": 37753, "h": "3fa0d89c8eb33e0d", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS,\n    MAX_TOKENS_WITH_REASONING, MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: if model.reasoning {\n            MAX_TOKENS_WITH_REASONING as i32\n        } else {\n            MAX_TOKENS as i32\n        },\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| formatget!(\"API error: {}\", e))?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "f3cc3dada24f9e4a56e7659eb23e76a99b9c59cc": {"t": 1785756935.0179596, "n": 37621, "h": "edca07004e2e6ad2", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS,\n    MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n        .map_err(|e| formatget!(\"HTTP client error: {}\", e))?;\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = response.map_err(|e| formatget!(\"API error: {}\", e))?;\n\n    let mut body = String::new();\n    resp.read_to_string(&mut body)\n        .map_err(|e| formatget!(\"Failed to read response: {}\", e))?;\n\n    let json: serde_json::Value =\n        serde_json::from_str(&body).map_err(|e| formatget!(\"Invalid response: {}\", e))?;\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    content.ok_or_else(|| formatget!(\"Invalid response: {}\", body))\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", err);\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = locate_config_path().ok_or_else(|| gettext(\"Config file not found.\"))?;\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "df4f4b3b264af22a7522f55c42d662489bc7a356": {"t": 1785756936.617607, "n": 37927, "h": "9ed3ae766d0ea5d5", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS,\n    MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history_for_joined_users(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "aa46e80ad19a4d5f192b48192e4de158e88e501a": {"t": 1785756937.9824758, "n": 37910, "h": "5edfe727cff5324c", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS,\n    MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => gettext(\n            \"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\",\n        ),\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag && let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\n                    \"alice\".to_string(),\n                    \"charlie\".to_string(),\n                    \"david\".to_string()\n                ]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "c38ecec64c92edbb9b013175a1074b1d6e99efce": {"t": 1785756939.560329, "n": 37864, "h": "b02d5df57086bb55", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS,\n    MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => {\n            gettext(\"Usage: !ai join <user...>  (Join your chat history with other users' histories.)\")\n        }\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { sender: String, users: Vec<String> },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let users: Vec<String> = args\n                .split_whitespace()\n                .filter(|u| !u.is_empty() && *u != sender)\n                .map(|u| u.to_string())\n                .collect();\n            if users.is_empty() {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                sender: sender.to_string(),\n                users,\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { sender, users } => {\n            for user in users {\n                memory\n                    .join_users(sender, user, receiver)\n                    .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string()],\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_parse_command_join_multiple_users() {\n        let command = parse_command(\"join\", \"alice charlie david\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                sender: \"bob\".to_string(),\n                users: vec![\"alice\".to_string(), \"charlie\".to_string(), \"david\".to_string()]\n            }\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_multiple_users() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            sender: \"bob\".to_string(),\n            users: vec![\"alice\".to_string(), \"charlie\".to_string()],\n        };\n        let _result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"charlie\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 3);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "b4bb79950405e87ee3d76df890497d798d215a4d": {"t": 1785756940.9441257, "n": 36534, "h": "a2a101441f2a1b61", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS,\n    MEMORY_RETENTION,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> Output {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    Output::RawMessage(formatget!(\n        \"Usage: !ai [flags...] [command] <query>.  Flags: -c (clear history), -t=<val> (temp), {} (select model).\\nCommands: join <user...>, solo [user], joined, weather <city>, forecast <city>.  Models: {}, Default: {}\\nBy default, history is split by user and deleted after {} minutes. The bot does not have access to a full chat log, it sees only your history.\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name,\n        MEMORY_RETENTION.whole_minutes(),\n    ))\n}\n\nfn usage_for_command(command: &CommandForHelp) -> String {\n    match command {\n        CommandForHelp::Join => {\n            gettext(\"Usage: !ai join <user>  (Join your chat history with another user's history.)\")\n        }\n        CommandForHelp::Solo => gettext(\n            \"Usage: !ai solo [user]  (Disable the shared history for yourself or the given user.)\",\n        ),\n        CommandForHelp::Joined => {\n            gettext(\"Usage: !ai joined  (List the users with whom you share the chat history.)\")\n        }\n        CommandForHelp::Weather => gettext(\n            \"Usage: !ai weather <city>  (Tell the AI the current weather. Weather data by https://open-meteo.com.)\",\n        ),\n        CommandForHelp::Forecast => gettext(\n            \"Usage: !ai forecast <city>  (Tell the AI this week's weather forecast. Weather data by https://open-meteo.com.)\",\n        ),\n    }\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // clear history\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // temperature\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\n#[derive(Debug)]\nenum Output {\n    // A message which needs to be sanitized before displaying it.\n    AgentMessage(String),\n    // A trusted message safe to display as-is.\n    RawMessage(String),\n}\n\nimpl std::fmt::Display for Output {\n    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {\n        match self {\n            Output::AgentMessage(msg) => write!(f, \"{}\", msg),\n            Output::RawMessage(msg) => write!(f, \"{}\", msg),\n        }\n    }\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(Output::AgentMessage(msg)) => {\n            println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key)))\n        }\n        Ok(Output::RawMessage(msg)) => println!(\"{}\", msg),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandForHelp {\n    Join,\n    Solo,\n    Joined,\n    Weather,\n    Forecast,\n}\n\n#[derive(Debug, PartialEq)]\nenum Command {\n    None,\n    Join { user_a: String, user_b: String },\n    Solo { user: String },\n    Joined { user: String },\n    Help { command: Option<CommandForHelp> },\n    Weather { city: String },\n    Forecast { city: String },\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n    ShowUsage,\n    ShowCustomUsage(String),\n}\n\nfn parse_command(command: &str, args: &str, sender: &str) -> Result<Command, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                });\n            }\n            Ok(Command::Join {\n                user_a: sender.to_string(),\n                user_b: target_user.to_string(),\n            })\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n            Ok(Command::Solo {\n                user: target.to_string(),\n            })\n        }\n        \"joined\" => Ok(Command::Joined {\n            user: sender.to_string(),\n        }),\n        \"help\" => {\n            let command = args.split_whitespace().next().unwrap_or(\"\").to_lowercase();\n            match command.as_ref() {\n                \"join\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Join),\n                }),\n                \"joined\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Joined),\n                }),\n                \"solo\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Solo),\n                }),\n                \"weather\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Weather),\n                }),\n                \"forecast\" => Ok(Command::Help {\n                    command: Some(CommandForHelp::Forecast),\n                }),\n                _ => Ok(Command::Help { command: None }),\n            }\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Weather),\n                    });\n                }\n                Ok(Command::Weather {\n                    city: city.to_string(),\n                })\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(Command::Help {\n                        command: Some(CommandForHelp::Forecast),\n                    });\n                }\n                Ok(Command::Forecast {\n                    city: city.to_string(),\n                })\n            } else {\n                Ok(Command::None)\n            }\n        }\n    }\n}\n\nfn process_command(\n    command: &Command,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    match command {\n        Command::Join { user_a, user_b } => {\n            memory\n                .join_users(user_a, user_b, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                user_a,\n                memory\n                    .get_joined_users_excluding_self(user_a, receiver)\n                    .join(\", \")\n            )))\n        }\n        Command::Solo { user } => {\n            memory\n                .make_user_solo(user, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n            Ok(CommandResult::Message(formatget!(\"{} is now solo.\", user)))\n        }\n        Command::Joined { user } => {\n            let other_users = memory.get_joined_users_excluding_self(user, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    user\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                user,\n                other_users.join(\", \")\n            )))\n        }\n        Command::Help { command: None } => Ok(CommandResult::ShowUsage),\n        Command::Help {\n            command: Some(command),\n        } => Ok(CommandResult::ShowCustomUsage(usage_for_command(command))),\n        Command::Weather { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::weather_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::Forecast { city } => match weather::get_weather(city) {\n            Ok(w) => Ok(CommandResult::AskAgent {\n                query: weather::forecast_prompt().to_string(),\n                extra_history: format!(\n                    \"{}{}\",\n                    formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                    w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                        + &formatget!(\"The current local time is {}.\", t))\n                ),\n            }),\n            Err(err) => Ok(CommandResult::Message(err)),\n        },\n        Command::None => Ok(CommandResult::NotACommand),\n    }\n}\n\nfn run(input: &Input) -> Result<Output, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(Output::RawMessage(gettext(\n            \"!ai is only available in channels.\",\n        )));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(Output::RawMessage(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE)));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let parsed_command = parse_command(command, args, &input.sender)?;\n    let query = match process_command(&parsed_command, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(Output::RawMessage(result)),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n        CommandResult::ShowUsage => {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n        CommandResult::ShowCustomUsage(usage) => return Ok(Output::RawMessage(usage)),\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t flag\n        let user_temperature_flag = input.flags.iter().any(|flag| flag.starts_with(\"t=\"));\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(Output::AgentMessage(result))\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_parse_command_join_valid() {\n        let command = parse_command(\"join\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Join {\n                user_a: \"bob\".to_string(),\n                user_b: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_empty_user() {\n        let command = parse_command(\"join\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_join_self() {\n        let command = parse_command(\"join\", \"bob\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_with_user() {\n        let command = parse_command(\"solo\", \"alice\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"alice\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_solo_without_user() {\n        let command = parse_command(\"solo\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Solo {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_joined() {\n        let command = parse_command(\"joined\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Joined {\n                user: \"bob\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_help_variants() {\n        let command = parse_command(\"help\", \"\", \"bob\").unwrap();\n        assert_eq!(command, Command::Help { command: None });\n\n        let command = parse_command(\"help\", \"join\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Join)\n            }\n        );\n\n        let command = parse_command(\"help\", \"solo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Solo)\n            }\n        );\n\n        let command = parse_command(\"help\", \"joined\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Joined)\n            }\n        );\n\n        let command = parse_command(\"help\", \"weather\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n\n        let command = parse_command(\"help\", \"forecast\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_valid() {\n        let command = parse_command(\"weather\", \"Tokyo\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Weather {\n                city: \"Tokyo\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_weather_empty() {\n        let command = parse_command(\"weather\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Weather)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_valid() {\n        let command = parse_command(\"forecast\", \"New York\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Forecast {\n                city: \"New York\".to_string()\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_forecast_empty() {\n        let command = parse_command(\"forecast\", \"\", \"bob\").unwrap();\n        assert_eq!(\n            command,\n            Command::Help {\n                command: Some(CommandForHelp::Forecast)\n            }\n        );\n    }\n\n    #[test]\n    fn test_parse_command_unknown() {\n        let command = parse_command(\"unknown\", \"args\", \"bob\").unwrap();\n        assert_eq!(command, Command::None);\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Join {\n            user_a: \"bob\".to_string(),\n            user_b: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_solo() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Solo {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is now solo.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_joined() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Joined {\n            user: \"alice\".to_string(),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"alice is not sharing memory with anyone.\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_help() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::Help { command: None };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::ShowUsage);\n\n        let command = Command::Help {\n            command: Some(CommandForHelp::Join),\n        };\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::ShowCustomUsage(usage_for_command(&CommandForHelp::Join))\n        );\n    }\n\n    #[test]\n    fn test_process_command_none() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let command = Command::None;\n        let result = process_command(&command, \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "3a3d94c9910fff65eef8154a0abec90bfbb3788d": {"t": 1785756942.8336496, "n": 27893, "h": "e9257f0cab62add9", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME,\n    MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> String {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: format!(\n                            \"{}{}\",\n                            formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                            w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                                + &formatget!(\"The current local time is {}.\", t))\n                        ),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else if command == gettext(\"forecast\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: forecast <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::forecast_prompt().to_string(),\n                        extra_history: format!(\n                            \"{}{}\",\n                            formatget!(\"Weather forecast for {}: {}.\", w.city, w.forecast),\n                            w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                                + &formatget!(\"The current local time is {}.\", t))\n                        ),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "67fdaf598cac9dbe676cdca1a4266f742f11a00a": {"t": 1785756944.3070433, "n": 26816, "h": "2e9a4ed55094ca74", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME,\n    MAX_LINE_LENGTH_CUTOFF, MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH_CUTOFF);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> String {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: format!(\n                            \"{}{}\",\n                            formatget!(\"The weather in {} is: {}.\", w.city, w.weather,),\n                            w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                                + &formatget!(\"The current local time is {}.\", t))\n                        ),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "755c45e724532f583cf45a7d0d31bb65ca4ecc9c": {"t": 1785756945.8254073, "n": 26802, "h": "6ec12983320c8c3b", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> String {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: format!(\n                            \"{}{}\",\n                            formatget!(\"The weather in {} is: {}.\", w.city, w.weather,),\n                            w.local_time.map_or(\"\".to_string(), |t| \" \".to_string()\n                                + &formatget!(\"The current local time is {}.\", t))\n                        ),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "9775080ac11a0718fbf9cf8142966f79ade9322a": {"t": 1785756947.226075, "n": 26534, "h": "fc5aeb8d882b8106", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> String {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(0);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "6546c6065b0f7499a57957344fb0b5c6de9b9019": {"t": 1785756948.8047125, "n": 26534, "h": "5a0c2fe15b6a6f86", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> String {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug, Default)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_default();\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "333e17c4296d29f4f96c3b76c82cdcb1b4a27832": {"t": 1785756950.131048, "n": 26571, "h": "cc7d732fce5197a3", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList, config: &Config, channel: &str) -> String {\n    let default_model_id = config.get_channel_default_model(channel);\n    let default_model_name = models\n        .select_model_for_channel(&[], default_model_id)\n        .map(|m| m.name.clone())\n        .unwrap_or_else(|_| default_model_id.to_string());\n\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default(default_model_id)\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models\n            .list_model_flags_human_readable(default_model_id)\n            .join(\" \"),\n        default_model_name\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(\n    query: &str,\n    models: &ModelList,\n    config: &Config,\n    channel: &str,\n) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query)\n        .map_err(|err| format!(\"{}.  {}\", err, usage(models, config, channel)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_else(|_| EnvVars {\n        vars: HashMap::new(),\n    });\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n        &config,\n        &receiver,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models, &input.config, &input.receiver));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "b70495a3f139bca6d1d9012444661676a5b9605e": {"t": 1785756951.6522374, "n": 26089, "h": "50bcc9d64dfa2e52", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList) -> String {\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default()\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models.list_model_flags_human_readable().join(\" \"),\n        models.default_model_name()\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(query: &str, models: &ModelList) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query).map_err(|err| format!(\"{}.  {}\", err, usage(models)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_else(|_| EnvVars {\n        vars: HashMap::new(),\n    });\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let query = if input.config.is_compiler_explorer_enabled() {\n        match compilerx::process_shortlinks(&query, &input.config_path) {\n            Ok(q) => q,\n            Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n            Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n            Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n        }\n    } else {\n        query\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "bc33434ef96a18c49de14b1f2838376aa6312e21": {"t": 1785756953.0144188, "n": 25979, "h": "f231076d307a5719", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList) -> String {\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default()\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models.list_model_flags_human_readable().join(\" \"),\n        models.default_model_name()\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(query: &str, models: &ModelList) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query).map_err(|err| format!(\"{}.  {}\", err, usage(models)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_else(|_| EnvVars {\n        vars: HashMap::new(),\n    });\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let query = match compilerx::process_shortlinks(&query, &input.config_path) {\n        Ok(q) => q,\n        Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n        Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n        Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "c9240b511ca471e20a3f0859282ce59ac92056f0": {"t": 1785756954.359666, "n": 25959, "h": "5a9c9ddc5eb420ce", "c": "mod compilerx;\nmod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::compilerx::CompilerError;\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList) -> String {\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default()\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models.list_model_flags_human_readable().join(\" \"),\n        models.default_model_name()\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(query: &str, models: &ModelList) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query).map_err(|err| format!(\"{}.  {}\", err, usage(models)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_else(|_| EnvVars {\n        vars: HashMap::new(),\n    });\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let query = match compilerx::process_shortlinks(&query) {\n        Ok(q) => q,\n        Err(CompilerError::MultipleShortlinks(msg)) => msg.to_string(),\n        Err(CompilerError::InvalidResponse(msg)) => msg.to_string(),\n        Err(e) => return Err(format!(\"Failed to process shortlinks: {}\", e)),\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "89df245e7893537a770c7da7ddf83e57ffbc2966": {"t": 1785756955.7936995, "n": 25598, "h": "06aa86035fa756ce", "c": "mod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    MAX_TOKENS, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: MAX_TOKENS as i32,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList) -> String {\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default()\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models.list_model_flags_human_readable().join(\" \"),\n        models.default_model_name()\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(query: &str, models: &ModelList) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query).map_err(|err| format!(\"{}.  {}\", err, usage(models)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_else(|_| EnvVars {\n        vars: HashMap::new(),\n    });\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "8869cf839d7bfacf31b31e7d729ac2cfc68f09e4": {"t": 1785756957.2435834, "n": 25572, "h": "44e99145f7a5f90e", "c": "mod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: 300,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList) -> String {\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default()\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models.list_model_flags_human_readable().join(\" \"),\n        models.default_model_name()\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(query: &str, models: &ModelList) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query).map_err(|err| format!(\"{}.  {}\", err, usage(models)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_else(|_| EnvVars {\n        vars: HashMap::new(),\n    });\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n\n        // Only show model prefix if user explicitly used a model flag\n        let user_model_flag = input\n            .flags\n            .iter()\n            .any(|flag| input.models.list_model_flags().contains(&flag.to_string()));\n        if user_model_flag && input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n\n        // Only show temperature prefix if user explicitly used -t or -temperature flag\n        let user_temperature_flag = input.flags.iter().any(|flag| {\n            flag.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || flag.starts_with(\"t=\")\n        });\n        if user_temperature_flag {\n            if let Some(t) = temperature {\n                flag_state.push(format!(\"t={}\", t));\n            }\n        }\n\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "0b468d8db9cdb1eb26354dd3f94e11bf6872c2ec": {"t": 1785756958.7274318, "n": 24999, "h": "8163239219675e2a", "c": "mod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::collections::HashMap;\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: 300,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList) -> String {\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default()\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models.list_model_flags_human_readable().join(\" \"),\n        models.default_model_name()\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\n#[derive(Debug)]\nstruct EnvVars {\n    vars: HashMap<String, String>,\n}\n\nimpl EnvVars {\n    fn from_file(path: &Path) -> Result<Self, Box<dyn std::error::Error>> {\n        let env_file_path = path.join(ENV_FILE_NAME);\n        let mut vars = HashMap::new();\n\n        for item in dotenvy::from_path_iter(&env_file_path)? {\n            let (key, value) = item?;\n            vars.insert(key, value);\n        }\n\n        Ok(EnvVars { vars })\n    }\n\n    fn get(&self, key: &str) -> Option<&String> {\n        self.vars.get(key)\n    }\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(query: &str, models: &ModelList) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query).map_err(|err| format!(\"{}.  {}\", err, usage(models)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    let env_vars = EnvVars::from_file(&config_path).unwrap_or_else(|_| EnvVars {\n        vars: HashMap::new(),\n    });\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path, &env_vars)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").as_deref() == Ok(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n        if input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n        if let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        let env_vars = EnvVars::from_file(&config_dir).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir, &env_vars).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "406158a184a45cb49731c9cc9b2f70db9d1a4ea4": {"t": 1785756960.1803627, "n": 24349, "h": "8aefed35a24d2e22", "c": "mod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: 300,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList) -> String {\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default()\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models.list_model_flags_human_readable().join(\" \"),\n        models.default_model_name()\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\nfn load_env(path: &Path) {\n    let _ = dotenvy::from_path(path.join(ENV_FILE_NAME));\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(query: &str, models: &ModelList) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query).map_err(|err| format!(\"{}.  {}\", err, usage(models)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    config: Config,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    load_env(&config_path);\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let config = Config::new(&config_path)?;\n    let models = ModelList::new(&config)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n    )?;\n\n    let channel_default_model = config.get_channel_default_model(&receiver);\n    let model = models\n        .select_model_for_channel(&flags, channel_default_model)?\n        .clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").ok().as_deref() == Some(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n        .or_else(|| input.config.get_channel_temperature(&input.receiver));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n        if input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n        if let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n\n    #[test]\n    fn test_channel_configuration_integration() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::{Config, ModelList};\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        dotenvy::from_path(&env_file).unwrap();\n\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(&config_path, r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.litellm]\nendpoint = \"http://test.example.com\"\nmodels = [\n  { id = \"test-model\", short_name = \"t\", name = \"Test Model\" },\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test-channel\" = { default_model = \"test-model\", temperature = 0.8, system_prompt = \"Test channel prompt\" }\n\"##).unwrap();\n\n        let config = Config::new(&config_dir).expect(\"Config::new()\");\n        let models = ModelList::new(&config).unwrap();\n\n        // Test channel-specific model selection\n        let channel_default = config.get_channel_default_model(\"#test-channel\");\n        assert_eq!(channel_default, \"test-model\");\n\n        let selected_model = models\n            .select_model_for_channel(&vec![], channel_default)\n            .unwrap();\n        assert_eq!(selected_model.id, \"test-model\");\n\n        // Test channel-specific temperature\n        let channel_temp = config.get_channel_temperature(\"#test-channel\");\n        assert_eq!(channel_temp, Some(0.8));\n\n        // Test channel-specific system prompt\n        let channel_prompt = config.get_channel_system_prompt(\"#test-channel\");\n        assert_eq!(channel_prompt, Some(\"Test channel prompt\"));\n\n        // Test fallback for unconfigured channel\n        let unconfigured_default = config.get_channel_default_model(\"#unknown\");\n        assert_eq!(unconfigured_default, \"default-model\");\n\n        let unconfigured_temp = config.get_channel_temperature(\"#unknown\");\n        assert_eq!(unconfigured_temp, None);\n\n        let unconfigured_prompt = config.get_channel_system_prompt(\"#unknown\");\n        assert_eq!(unconfigured_prompt, None);\n    }\n\n    #[test]\n    fn test_temperature_extraction_with_channel_config() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"LITELLM_API_KEY=test-key\\n\").unwrap();\n        dotenvy::from_path(&env_file).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { default_model = \"test-model\", temperature = 0.5, system_prompt = \"Test prompt\" }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir).expect(\"Config::new()\");\n\n        // Test channel-specific temperature fallback (simulating the logic from run())\n        let flags: Vec<String> = vec![];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.5));\n\n        // Test for unconfigured channel\n        let temperature_unconfigured = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#unknown\"));\n\n        assert_eq!(temperature_unconfigured, None);\n    }\n\n    #[test]\n    fn test_temperature_flag_overrides_channel() {\n        use crate::constants::CONFIG_FILE_NAME;\n        use crate::model::Config;\n\n        let temp_dir = tempfile::tempdir().unwrap();\n        let config_dir = temp_dir.path();\n        let env_file = config_dir.join(\".env\");\n        std::fs::write(&env_file, \"DEEPSEEK_API_KEY=test-key\\n\").unwrap();\n        dotenvy::from_path(&env_file).unwrap();\n\n        std::fs::create_dir_all(&config_dir).unwrap();\n        let config_path = config_dir.join(CONFIG_FILE_NAME);\n        std::fs::write(\n            &config_path,\n            r##\"\n[general]\ndefault_model = \"default-model\"\n\n[providers.deepseek]\nmodels = [\n  { id = \"default-model\", short_name = \"d\", name = \"Default Model\" }\n]\n\n[channels]\n\"#test\" = { temperature = 0.5 }\n\"##,\n        )\n        .unwrap();\n\n        let config = Config::new(&config_dir).expect(\"Config::new()\");\n\n        // Temperature flag should override channel temperature (simulating the logic from run())\n        let flags = vec![\"t=0.9\".to_string()];\n        let temperature = flags\n            .iter()\n            .find(|f| f.starts_with(\"temperature=\") || f.starts_with(\"t=\"))\n            .and_then(|f| f.split('=').nth(1))\n            .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)))\n            .or_else(|| config.get_channel_temperature(\"#test\"));\n\n        assert_eq!(temperature, Some(0.9));\n    }\n}\n"}, "3e8fe7f8804b40a20a3637ae3dfebb3da49630ca": {"t": 1785756962.1644702, "n": 18591, "h": "6eed82005f2903c0", "c": "mod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, ENV_FILE_NAME, MAX_LINE_LENGTH,\n    TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: 300,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList) -> String {\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default()\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models.list_model_flags_human_readable().join(\" \"),\n        models.default_model_name()\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\nfn load_env(path: &Path) {\n    let _ = dotenvy::from_path(path.join(ENV_FILE_NAME));\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(query: &str, models: &ModelList) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query).map_err(|err| format!(\"{}.  {}\", err, usage(models)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    load_env(&config_path);\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let models = ModelList::new(&Config::new(&config_path)?)?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n    )?;\n\n    let model = models.select_model(&flags)?.clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").ok().as_deref() == Some(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config_path,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n        if input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n        if let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n}\n"}, "3d80b177828df85205632b31b83fa9d7f57f4a10": {"t": 1785756963.7674105, "n": 18567, "h": "b6d199e2386b5339", "c": "mod constants;\nmod gettext;\nmod memory;\nmod model;\nmod prompt;\nmod unicodebytelimit;\nmod weather;\n\nuse crate::constants::{\n    CLEAR_MEMORY_FLAG, CLEAR_MEMORY_MESSAGE, CONFIG_FILE_NAME, MAX_LINE_LENGTH, TEMPERATURE_FLAG,\n};\nuse crate::memory::{Memory, Sender};\nuse crate::model::{Config, Model, ModelList};\nuse crate::prompt::{Message, build_prompt};\nuse crate::unicodebytelimit::UnicodeByteLimit;\n\nuse formatx::formatx;\nuse gettextrs::{TextDomain, gettext, ngettext};\nuse std::io::Read;\nuse std::path::{Path, PathBuf};\nuse std::time::Duration;\n\n#[derive(Debug, Clone, PartialEq, Eq)]\nstruct Flag {\n    name: String,\n    requires_value: bool,\n}\n\nimpl Flag {\n    fn new(name: String, requires_value: bool) -> Self {\n        Flag {\n            name,\n            requires_value,\n        }\n    }\n}\n\nfn call_api(\n    model: &Model,\n    prompt: &Vec<Message>,\n    temperature: &Option<f64>,\n) -> Result<String, String> {\n    let client = match reqwest::blocking::Client::builder()\n        .timeout(Duration::from_secs(20))\n        .build()\n    {\n        Ok(c) => c,\n        Err(e) => return Err(formatget!(\"HTTP client error: {}\", e)),\n    };\n\n    #[derive(Debug, serde::Serialize)]\n    struct Payload<'a> {\n        model: &'a str,\n        messages: &'a Vec<Message>,\n        max_tokens: i32,\n        #[serde(skip_serializing_if = \"Option::is_none\")]\n        temperature: &'a Option<f64>,\n    }\n\n    let payload = serde_json::json!(Payload {\n        model: &model.id,\n        messages: prompt,\n        max_tokens: 300,\n        temperature,\n    });\n\n    let response = client\n        .post(model.endpoint.clone())\n        .header(\"Authorization\", format!(\"Bearer {}\", model.api_key))\n        .header(\"Content-Type\", \"application/json\")\n        .json(&payload)\n        .send();\n\n    let mut resp = match response {\n        Ok(r) => r,\n        Err(e) => return Err(formatget!(\"API error: {}\", e)),\n    };\n\n    let mut body = String::new();\n    if let Err(e) = resp.read_to_string(&mut body) {\n        return Err(formatget!(\"Failed to read response: {}\", e));\n    }\n\n    let json: serde_json::Value = match serde_json::from_str(&body) {\n        Ok(j) => j,\n        Err(e) => return Err(formatget!(\"Invalid response: {}\", e)),\n    };\n\n    let content = json[\"choices\"]\n        .get(0)\n        .and_then(|c| c[\"message\"][\"content\"].as_str())\n        .map(|s| s.to_string());\n\n    match content {\n        Some(text) => Ok(text),\n        None => Err(formatget!(\"Invalid response: {}\", body)),\n    }\n}\n\nfn sanitize_output(s: &str, api_key: &Option<&str>) -> String {\n    // An HTTPS request error might expose the API key by accident, so we redact it to be safe.\n    // This is irrelevant for successful requests because the LLM doesn't know the API key.\n    let redacted = match api_key {\n        Some(k) => s.replace(k, \"[REDACTED]\"),\n        None => s.to_string(),\n    };\n    let s_no_newlines: String = redacted.chars().filter(|&c| c != '\\n').collect();\n    let truncated = s_no_newlines.unicode_byte_limit(MAX_LINE_LENGTH);\n    if truncated != s_no_newlines {\n        format!(\"{}...\", truncated)\n    } else {\n        truncated.to_string()\n    }\n}\n\nfn usage(models: &ModelList) -> String {\n    formatget!(\n        \"Usage: !ai [{}] [-{}|-c] [-{}=1.0|-t=1.0] <query>.  Models: {}.  Default: {}\",\n        models\n            .list_model_flags_without_default()\n            .into_iter()\n            .map(|f| format!(\"-{}\", f))\n            .collect::<Vec<_>>()\n            .join(\"|\"),\n        CLEAR_MEMORY_FLAG,\n        TEMPERATURE_FLAG,\n        models.list_model_flags_human_readable().join(\" \"),\n        models.default_model_name()\n    )\n}\n\nfn locate_config_path() -> Option<PathBuf> {\n    let current_dir = std::env::current_dir().ok()?;\n    let config_file = current_dir.join(CONFIG_FILE_NAME);\n    if config_file.exists() && config_file.is_file() {\n        return Some(current_dir);\n    }\n    None\n}\n\nfn load_env(path: &Path) {\n    let _ = dotenvy::from_path(path.join(CONFIG_FILE_NAME));\n}\n\nfn textdomain_dir() -> Option<String> {\n    // start in the executable directory, walk up to find the \"gettext\" directory\n    let mut dir = std::env::current_exe()\n        .ok()\n        .and_then(|p| p.parent().map(|d| d.to_path_buf()))\n        .unwrap_or_else(|| std::env::current_dir().unwrap());\n    loop {\n        let gettext_dir = dir.join(\"gettext\");\n        if gettext_dir.is_dir() {\n            return Some(gettext_dir.to_string_lossy().into_owned());\n        }\n        if !dir.pop() {\n            break;\n        }\n    }\n    None\n}\n\n// Extracts known flags from the query. Returns the flags and the remaining query.\n// Example:\n// Input: [Flag{name:\"foo\", requires_value:false}, Flag{name:\"t\", requires_value:true}],\n//        \"-foo -t=1.3   rest -of    the   query\"\n// Result: [\"foo\", \"t=1.3\"], \"rest -of    the   query\"\nfn extract_flags(known_flags: &[Flag], query: &str) -> Result<(Vec<String>, String), String> {\n    // Extract all -flags from query from the beginning until query no longer starts with - or we hit the end of string. Use string splitting or something, don't iterate over individual characters.\n    // Collect all extracted flags which are not in known_flags. If non-empty, return all of them in the error.\n    // Otherwise, return the extracted flags and the remaining query.\n    let mut flags = Vec::new();\n    let mut unknown_flags = Vec::new();\n    let mut rest = query.trim_start();\n    while let Some(stripped) = rest.strip_prefix('-') {\n        let (flag_with_value, remaining) = stripped.split_once(' ').unwrap_or((stripped, \"\"));\n        rest = remaining.trim_start();\n        let flag_name = flag_with_value.split('=').next().unwrap_or(flag_with_value);\n\n        if let Some(known_flag) = known_flags.iter().find(|f| f.name == flag_name) {\n            if known_flag.requires_value && !flag_with_value.contains('=') {\n                return Err(formatget!(\n                    \"Flag -{} requires a value (e.g., -{}=1.0)\",\n                    flag_name,\n                    flag_name\n                ));\n            }\n            flags.push(flag_with_value.to_string());\n        } else {\n            unknown_flags.push(flag_with_value.to_string());\n        }\n    }\n    if !unknown_flags.is_empty() {\n        let unknown_flags_str = unknown_flags.join(\", \");\n        let unknown_flags_str = if unknown_flags_str.len() > 60 {\n            unknown_flags_str[..60].to_string() + \"...\"\n        } else {\n            unknown_flags_str\n        };\n        return Err(formatx!(\n            ngettext(\n                \"Unknown flag: {}\",\n                \"Unknown flags: {}\",\n                unknown_flags.len() as u32\n            ),\n            &unknown_flags_str\n        )\n        .unwrap_or_else(|_| format!(\"Unknown flag(s): {}\", &unknown_flags_str)));\n    }\n    Ok((flags, rest.to_string()))\n}\n\nfn parse_command_line(query: &str, models: &ModelList) -> Result<(Vec<String>, String), String> {\n    let known_flags = {\n        let mut known_flags = Vec::new();\n        for model_flag in models.list_model_flags() {\n            known_flags.push(Flag::new(model_flag, false));\n        }\n        known_flags.push(Flag::new(CLEAR_MEMORY_FLAG.to_string(), false));\n        known_flags.push(Flag::new(\"c\".to_string(), false)); // Short for CLEAR_MEMORY_FLAG\n        known_flags.push(Flag::new(TEMPERATURE_FLAG.to_string(), true));\n        known_flags.push(Flag::new(\"t\".to_string(), true)); // Short for TEMPERATURE_FLAG\n        known_flags\n    };\n    let flag_names: Vec<&String> = known_flags.iter().map(|f| &f.name).collect();\n    if flag_names.len()\n        != flag_names\n            .iter()\n            .collect::<std::collections::HashSet<_>>()\n            .len()\n    {\n        return Err(gettext(\n            \"Internal error: duplicate configured flags detected, check your model config\",\n        ));\n    }\n    extract_flags(&known_flags, query).map_err(|err| format!(\"{}.  {}\", err, usage(models)))\n}\n\nstruct Input {\n    config_path: PathBuf,\n    flags: Vec<String>,\n    models: ModelList,\n    model: Model,\n    sender: String,\n    receiver: String,\n    query: String,\n    irc_plugin: bool,\n}\n\nfn main() {\n    let input = setup().unwrap_or_else(|err| {\n        println!(\"{}\", sanitize_output(&err.to_string(), &None));\n        std::process::exit(1);\n    });\n    match run(&input) {\n        Ok(msg) => println!(\"{}\", sanitize_output(&msg, &Some(&input.model.api_key))),\n        Err(err) => {\n            println!(\n                \"{}\",\n                sanitize_output(&err.to_string(), &Some(&input.model.api_key))\n            );\n            std::process::exit(1);\n        }\n    }\n}\n\nfn setup() -> Result<Input, String> {\n    if let Some(dir) = textdomain_dir() {\n        // Ignore errors and use untranslated strings if it fails.\n        let _ = TextDomain::new(\"japanese_tools\")\n            .skip_system_data_paths()\n            .push(&dir)\n            .init();\n    }\n    let config_path = match locate_config_path() {\n        Some(path) => path,\n        None => return Err(gettext(\"Config file not found.\")),\n    };\n    load_env(&config_path);\n\n    let sender = std::env::var(\"DMB_SENDER\").unwrap_or_default();\n    let receiver = std::env::var(\"DMB_RECEIVER\").unwrap_or_default();\n\n    let models = ModelList::new(&Config::from_env())?;\n\n    let (flags, query) = parse_command_line(\n        &std::env::args().skip(1).collect::<Vec<_>>().join(\" \"),\n        &models,\n    )?;\n\n    let model = models.select_model(&flags)?.clone();\n\n    Ok(Input {\n        query,\n        sender,\n        receiver,\n        flags,\n        models,\n        model,\n        config_path,\n        irc_plugin: std::env::var(\"IRC_PLUGIN\").ok().as_deref() == Some(\"1\"),\n    })\n}\n\n#[derive(Debug, PartialEq)]\nenum CommandResult {\n    NotACommand,\n    Message(String),\n    AskAgent {\n        extra_history: String,\n        query: String,\n    },\n}\n\nfn process_command(\n    command: &str,\n    args: &str,\n    sender: &str,\n    receiver: &str,\n    memory: &mut Memory,\n) -> Result<CommandResult, String> {\n    let command = command.to_lowercase();\n    match command.as_ref() {\n        \"join\" => {\n            let target_user = args.trim();\n            if target_user.is_empty() || target_user == sender {\n                return Ok(CommandResult::Message(\n                    gettext(\"Usage: join <username>\").to_string(),\n                ));\n            }\n\n            memory\n                .join_users(sender, target_user, receiver)\n                .map_err(|e| format!(\"Failed to join users: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} joined memory with the group: {}\",\n                sender,\n                memory\n                    .get_joined_users_excluding_self(sender, receiver)\n                    .join(\", \")\n            )))\n        }\n        \"solo\" => {\n            let arg = args.trim();\n            let target = if arg.is_empty() { sender } else { arg };\n\n            memory\n                .make_user_solo(target, receiver)\n                .map_err(|e| format!(\"Failed to make user solo: {}\", e))?;\n\n            Ok(CommandResult::Message(formatget!(\n                \"{} is now solo.\",\n                target\n            )))\n        }\n        \"joined\" => {\n            let other_users = memory.get_joined_users_excluding_self(sender, receiver);\n            if other_users.is_empty() {\n                return Ok(CommandResult::Message(formatget!(\n                    \"{} is not sharing memory with anyone.\",\n                    sender\n                )));\n            }\n            Ok(CommandResult::Message(formatget!(\n                \"{} is sharing memory with: {}\",\n                sender,\n                other_users.join(\", \")\n            )))\n        }\n        _ => {\n            if command == gettext(\"weather\").to_lowercase() {\n                let city = args.trim();\n                if city.is_empty() {\n                    return Ok(CommandResult::Message(format!(\n                        \"{}  ({})\",\n                        gettext(\"Usage: weather <city>\"),\n                        gettext(\"Weather data by https://open-meteo.com\")\n                    )));\n                }\n                match weather::get_weather(city) {\n                    Ok(w) => Ok(CommandResult::AskAgent {\n                        query: weather::weather_prompt().to_string(),\n                        extra_history: formatget!(\"The weather in {} is: {}.\", w.city, w.weather),\n                    }),\n                    Err(err) => Ok(CommandResult::Message(err)),\n                }\n            } else {\n                Ok(CommandResult::NotACommand)\n            }\n        }\n    }\n}\n\nfn run(input: &Input) -> Result<String, String> {\n    // Prevent usage in private messages\n    if input.irc_plugin && !input.receiver.starts_with('#') {\n        return Ok(gettext(\"!ai is only available in channels.\"));\n    }\n\n    let mut memory = Memory::new_from_path(&input.config_path)?;\n\n    let history_cleared = input.flags.contains(&CLEAR_MEMORY_FLAG.to_string())\n        || input.flags.contains(&\"c\".to_string());\n    if history_cleared {\n        memory\n            .clear_history(&input.sender, &input.receiver)\n            .map_err(|e| format!(\"Failed to clear history: {}\", e))?;\n    }\n\n    let query = input.query.trim();\n    if query.is_empty() {\n        if history_cleared {\n            return Ok(format!(\"[{}]\", CLEAR_MEMORY_MESSAGE));\n        } else {\n            return Ok(usage(&input.models));\n        }\n    }\n\n    let (command, args) = query.split_once(' ').unwrap_or((query, \"\"));\n    let query = match process_command(command, args, &input.sender, &input.receiver, &mut memory)? {\n        CommandResult::Message(result) => return Ok(result),\n        CommandResult::NotACommand => query.to_string(),\n        CommandResult::AskAgent {\n            query,\n            extra_history,\n        } => {\n            memory\n                .add_to_history(&input.sender, Sender::User, &input.receiver, &extra_history)\n                .map_err(|e| format!(\"Failed to add extra history: {}\", e))?;\n            query\n        }\n    };\n\n    let prompt = build_prompt(\n        &query,\n        &input.sender,\n        &input.receiver,\n        &memory,\n        &input.config_path,\n    );\n    memory\n        .add_to_history(&input.sender, Sender::User, &input.receiver, &query)\n        .map_err(|e| format!(\"Failed to add user query to history: {}\", e))?;\n\n    let temperature = input\n        .flags\n        .iter()\n        .find(|f| f.starts_with(&(TEMPERATURE_FLAG.to_string() + \"=\")) || f.starts_with(\"t=\"))\n        .and_then(|f| f.split('=').nth(1))\n        .and_then(|s| s.parse::<f64>().ok().map(|t| t.clamp(0.0, 2.0)));\n\n    let result = &call_api(&input.model, &prompt, &temperature)?;\n\n    memory\n        .add_to_history(&input.sender, Sender::Assistant, &input.receiver, result)\n        .map_err(|e| format!(\"Failed to add assistant response to history: {}\", e))?;\n\n    let flag_state = {\n        let mut flag_state: Vec<String> = Vec::new();\n        if input.model.name != input.models.default_model_name() {\n            flag_state.push(input.model.short_name.clone());\n        }\n        if let Some(t) = temperature {\n            flag_state.push(format!(\"t={}\", t));\n        }\n        flag_state.join(\" \")\n    };\n    let result = if flag_state.is_empty() {\n        result.to_string()\n    } else {\n        format!(\"[{}] {}\", flag_state, result)\n    };\n\n    let result = if history_cleared {\n        format!(\"[{}] {}\", CLEAR_MEMORY_MESSAGE, &result)\n    } else {\n        result\n    };\n\n    // Prevent triggering other bots that might be present in the same channel.\n    let result = match result.chars().next() {\n        Some('!') => \" \".to_string() + &result,\n        _ => result,\n    };\n    Ok(result)\n}\n\n#[cfg(test)]\nmod tests {\n    use super::*;\n    use tempfile::tempdir;\n\n    #[test]\n    fn test_extract_flags() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"bar\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let (flags, query) =\n            extract_flags(&known_flags, \"-foo -t=1.3 -bar   rest -of    the   query\").unwrap();\n        assert_eq!(flags, vec![\"foo\", \"t=1.3\", \"bar\"]);\n        assert_eq!(query, \"rest -of    the   query\");\n    }\n\n    #[test]\n    fn test_extract_flags_temperature_requires_value() {\n        let known_flags = vec![\n            Flag::new(\"foo\".to_string(), false),\n            Flag::new(\"t\".to_string(), true),\n        ];\n        let result = extract_flags(&known_flags, \"-t foo\");\n        assert!(result.is_err());\n        let error = result.unwrap_err();\n        assert!(error.contains(\"Flag -t requires a value\"));\n        assert!(error.contains(\"-t=1.0\"));\n    }\n\n    #[test]\n    fn test_process_command_join() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"alice\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(\n            result,\n            CommandResult::Message(\"bob joined memory with the group: alice\".to_string())\n        );\n\n        // Verify the users are actually joined\n        let joined_users = memory.get_joined_users(\"bob\", \"receiver1\");\n        assert!(joined_users.contains(&\"alice\".to_string()));\n        assert!(joined_users.contains(&\"bob\".to_string()));\n        assert_eq!(joined_users.len(), 2);\n    }\n\n    #[test]\n    fn test_process_command_join_empty_user() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_join_self() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"join\", \"bob\", \"bob\", \"receiver1\", &mut memory);\n        assert!(result.is_ok());\n        assert_eq!(\n            result.unwrap(),\n            CommandResult::Message(\"Usage: join <username>\".to_string())\n        );\n    }\n\n    #[test]\n    fn test_process_command_unknown() {\n        let dir = tempdir().unwrap();\n        let mut memory = Memory::new_from_path(dir.path()).unwrap();\n\n        let result = process_command(\"unknown\", \"args\", \"bob\", \"receiver1\", &mut memory).unwrap();\n        assert_eq!(result, CommandResult::NotACommand);\n    }\n}\n"}}}}