Files
galaxy/app/src/integration_testing/agent_mode/user_defaults.rs
T

47 lines
1.6 KiB
Rust

use std::collections::HashMap;
// User default keys
const IS_ACTIVE_AI_ENABLED: &str = "IsActiveAIEnabled";
const INTELLIGENT_AUTOSUGGESTIONS_ENABLED: &str = "IntelligentAutosuggestionsEnabled";
const NATURAL_LANGUAGE_AUTOSUGGESTIONS_ENABLED: &str = "NaturalLanguageAutosuggestionsEnabled";
const AGENT_MODE_QUERY_SUGGESTIONS_ENABLED: &str = "AgentModeQuerySuggestionsEnabled";
const CODE_SUGGESTIONS_ENABLED: &str = "CodeSuggestionsEnabled";
pub fn user_defaults_map_with_active_ai(enabled: bool) -> HashMap<String, String> {
HashMap::from_iter([
(
INTELLIGENT_AUTOSUGGESTIONS_ENABLED.to_owned(),
enabled.to_string(),
),
(
AGENT_MODE_QUERY_SUGGESTIONS_ENABLED.to_owned(),
enabled.to_string(),
),
(CODE_SUGGESTIONS_ENABLED.to_owned(), enabled.to_string()),
(
NATURAL_LANGUAGE_AUTOSUGGESTIONS_ENABLED.to_owned(),
enabled.to_string(),
),
(IS_ACTIVE_AI_ENABLED.to_owned(), enabled.to_string()),
])
}
/// User defaults for predictable AI input behavior needed in evals.
///
/// This allows tests to more reliably enter and exit AI input mode.
///
/// * UDI is enabled
/// * Natural language detection is disabled
pub fn user_defaults_map_for_ai_input() -> HashMap<String, String> {
HashMap::from_iter([
(
"AIAutoDetectionEnabled".to_owned(),
serde_json::to_string(&false).unwrap(),
),
(
"InputBoxTypeSetting".to_owned(),
serde_json::to_string("Universal").unwrap(),
),
])
}