llm 1.3.8

A Rust library unifying multiple LLM backends.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145

use crossterm::event::{KeyCode, KeyEvent};

use crate::provider::backend_env_key;
use crate::runtime::{OnboardingState, OnboardingStep};

use super::super::{OverlayAction, OverlayResult};

pub(super) fn handle_onboarding(state: &mut OnboardingState, key: KeyEvent) -> OverlayResult {
    match state.step {
        OnboardingStep::Welcome => handle_onboarding_welcome(state, key),
        OnboardingStep::Provider => handle_onboarding_provider(state, key),
        OnboardingStep::ApiKey => handle_onboarding_key(state, key),
        OnboardingStep::Preferences => handle_onboarding_preferences(state, key),
        OnboardingStep::Confirm => handle_onboarding_confirm(state, key),
    }
}

fn handle_onboarding_welcome(state: &mut OnboardingState, key: KeyEvent) -> OverlayResult {
    match key.code {
        KeyCode::Enter => {
            state.next_step();
            OverlayResult::action(OverlayAction::Handled)
        }
        _ => OverlayResult::action(OverlayAction::None),
    }
}

fn handle_onboarding_provider(state: &mut OnboardingState, key: KeyEvent) -> OverlayResult {
    match key.code {
        KeyCode::Esc => {
            state.prev_step();
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Up | KeyCode::Char('k') => {
            state.error = None;
            state.select_prev();
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Down | KeyCode::Char('j') => {
            state.error = None;
            state.select_next();
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Enter => {
            state.error = None;
            advance_onboarding(state);
            OverlayResult::action(OverlayAction::Handled)
        }
        _ => OverlayResult::action(OverlayAction::None),
    }
}

fn handle_onboarding_key(state: &mut OnboardingState, key: KeyEvent) -> OverlayResult {
    match key.code {
        KeyCode::Esc => {
            state.prev_step();
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Backspace => {
            state.error = None;
            state.api_key.pop();
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Char(ch) => {
            state.error = None;
            state.api_key.push(ch);
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Enter => {
            if requires_key(state) && state.api_key.trim().is_empty() {
                state.set_error("API key required.");
            } else {
                state.next_step();
            }
            OverlayResult::action(OverlayAction::Handled)
        }
        _ => OverlayResult::action(OverlayAction::None),
    }
}

fn handle_onboarding_preferences(state: &mut OnboardingState, key: KeyEvent) -> OverlayResult {
    match key.code {
        KeyCode::Esc => {
            state.prev_step();
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Left | KeyCode::Right => {
            state.mode = toggle_mode(state.mode);
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Up | KeyCode::Down => {
            state.theme = toggle_theme(&state.theme);
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Enter => {
            state.next_step();
            OverlayResult::action(OverlayAction::Handled)
        }
        _ => OverlayResult::action(OverlayAction::None),
    }
}

fn handle_onboarding_confirm(state: &mut OnboardingState, key: KeyEvent) -> OverlayResult {
    match key.code {
        KeyCode::Esc => {
            state.prev_step();
            OverlayResult::action(OverlayAction::Handled)
        }
        KeyCode::Enter => OverlayResult::action(OverlayAction::FinishOnboarding),
        _ => OverlayResult::action(OverlayAction::None),
    }
}

fn advance_onboarding(state: &mut OnboardingState) {
    if state.selected_provider().is_none() {
        state.set_error("Select a provider.");
        return;
    }
    if requires_key(state) {
        state.step = OnboardingStep::ApiKey;
        return;
    }
    state.step = OnboardingStep::Preferences;
}

fn requires_key(state: &OnboardingState) -> bool {
    state
        .selected_provider()
        .and_then(|provider| backend_env_key(&provider.backend))
        .is_some()
}

fn toggle_mode(mode: crate::config::NavigationMode) -> crate::config::NavigationMode {
    match mode {
        crate::config::NavigationMode::Simple => crate::config::NavigationMode::Vi,
        crate::config::NavigationMode::Vi => crate::config::NavigationMode::Simple,
    }
}

fn toggle_theme(current: &str) -> String {
    match current.to_lowercase().as_str() {
        "codex" => "mono".to_string(),
        _ => "codex".to_string(),
    }
}