rho-coding-agent 1.8.2

A lightweight agent harness inspired by Pi
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

use super::{PickerAction, PickerItem, UiPicker};
use rho_providers::{
    auth::login_dispatch::ProviderAuthentication, credentials::CredentialStore, model::catalog,
    provider,
};

pub(super) const ALL_REFRESHABLE_PROVIDERS: &str = "all";

pub(super) fn login_group_picker() -> UiPicker {
    let items = catalog::login_groups()
        .into_iter()
        .map(|group| PickerItem {
            label: group.prompt,
            detail: None,
            preview: None,
            badge: None,
            value: group.id,
        })
        .collect();
    UiPicker::new(
        "select provider to login",
        "type regex filter, tab complete, up/down select, enter confirm, esc cancel",
        items,
        PickerAction::LoginGroup,
    )
}

pub(super) fn login_method_picker(group: catalog::LoginGroup) -> UiPicker {
    let title = format!("select {} login method", group.prompt);
    let items = group
        .methods
        .into_iter()
        .map(|method| PickerItem {
            label: method.prompt,
            detail: None,
            preview: None,
            badge: None,
            value: method.target.provider,
        })
        .collect();
    UiPicker::new(
        title,
        "type regex filter, tab complete, up/down select, enter confirm, esc cancel",
        items,
        PickerAction::LoginProvider,
    )
}

pub(super) fn refresh_model_list_picker(available_auths: &[String]) -> UiPicker {
    let mut items = vec![PickerItem {
        label: "All configured providers".into(),
        detail: Some("Refresh every available provider with model discovery support.".into()),
        preview: None,
        badge: None,
        value: ALL_REFRESHABLE_PROVIDERS.into(),
    }];
    items.extend(
        provider::providers()
            .iter()
            .filter(|descriptor| descriptor.model_refresh.is_some())
            .filter(|descriptor| available_auths.iter().any(|auth| auth == descriptor.auth))
            .map(|descriptor| PickerItem {
                label: descriptor.display_name.into(),
                detail: Some(format!(
                    "Refresh cached {} models.",
                    descriptor.display_name
                )),
                preview: None,
                badge: None,
                value: descriptor.name.into(),
            }),
    );
    UiPicker::new(
        "Refresh model lists",
        "type regex filter, enter refresh, esc back",
        items,
        PickerAction::RefreshModelList,
    )
}

pub(super) fn logout_provider_picker(
    store: &dyn CredentialStore,
) -> rho_providers::credentials::CredentialResult<UiPicker> {
    let mut targets = Vec::new();
    for target in catalog::login_targets() {
        if ProviderAuthentication::has_stored_credentials(store, &target.provider)? {
            targets.push(target);
        }
    }
    Ok(provider_picker_for_targets(
        "logout",
        PickerAction::LogoutProvider,
        targets,
    ))
}

fn provider_picker_for_targets(
    verb: &str,
    action: PickerAction,
    targets: Vec<catalog::LoginTarget>,
) -> UiPicker {
    let items = targets
        .into_iter()
        .map(|target| PickerItem {
            label: target.provider.clone(),
            detail: Some(target.label),
            preview: None,
            badge: None,
            value: target.provider,
        })
        .collect();

    UiPicker::new(
        format!("select provider to {verb}"),
        "type regex filter, tab complete, up/down select, enter confirm, esc cancel",
        items,
        action,
    )
}

#[cfg(test)]
#[path = "provider_picker_tests.rs"]
mod tests;