sapiens_cli 0.2.0

The cli interface of a simple bot to conquer the world
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

//! Main for sapiens_cli
use std::net::IpAddr;
use std::rc::Rc;
use std::str::FromStr;

use clap::Parser;
use dotenvy::dotenv_override;
use huelib::bridge;
use sapiens::tools::Toolbox;
use sapiens::{something, Config};
use sapiens_tools::conclude::ConcludeTool;
use sapiens_tools::hue::room::RoomTool;
use sapiens_tools::hue::status::StatusTool;
use sapiens_tools::python::PythonTool;

// Simplify adding tools:
// TODO(ssoudan) https://pyo3.rs/v0.17.3/conversions/traits for Input and Output structs?
//
// Usability:
// TODO(ssoudan) Conditional loading of tools
// TODO(ssoudan) More tools: search, wiki, wx, arxiv, negotiate
// TODO(ssoudan) Discord bot with long-lived conversations
// TODO(ssoudan) allow the bot to share its doubt and ask for help
// TODO(ssoudan) Crontab-like scheduling: get a summary of the news daily at 8am
// TODO(ssoudan) better errors for python code
//
// Deployability:
// TODO(ssoudan) use insta for some of the tests
// TODO(ssoudan) more tests
// TODO(ssoudan) logging
// TODO(ssoudan) monitoring
//
// Adoption:
// TODO(ssoudan) More documentation and examples
// TODO(ssoudan) A site?
//
// Explore:
// TODO(ssoudan) other models?
// TODO(ssoudan) memory?
// TODO(ssoudan) vector stores?
// TODO(ssoudan) prompt optimization
// TODO(ssoudan) multiple models - critic?
// TODO(ssoudan) multi-stage evaluation

/// A bot that can do things - or at least try to.
#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    /// Model to use
    #[arg(long, default_value = "gpt-3.5-turbo")]
    model: String,

    /// Maximum number of steps to execute
    #[arg(short, long, default_value_t = 10)]
    max_steps: u32,

    /// Task to execute
    #[arg(short, long, default_value = "Tell me a joke.")]
    task: String,

    /// Show the warmup prompt
    #[arg(long)]
    show_warmup_prompt: bool,
}

impl From<Args> for Config {
    fn from(args: Args) -> Self {
        Self {
            model: args.model,
            max_steps: args.max_steps,
            show_warmup_prompt: args.show_warmup_prompt,
            ..Default::default()
        }
    }
}

#[tokio::main]
async fn main() {
    let args = Args::parse();

    let _ = dotenv_override();

    let bridge_ip = match std::env::var("HUE_BRIDGE_IP") {
        Ok(ip) => IpAddr::from_str(&ip).expect("Invalid IP address"),
        Err(_) => {
            println!("HUE_BRIDGE_IP env not set. Trying to discover bridge.");
            let bridge_ip = bridge::discover_nupnp().unwrap().pop().unwrap();
            println!(
                "Discovered bridge at IP address: HUE_BRIDGE_IP={}",
                bridge_ip
            );
            bridge_ip
        }
    };

    let username = match std::env::var("HUE_USERNAME") {
        Ok(username) => username,
        Err(_) => {
            println!("HUE_USERNAME env not set. Trying to register a new user.");

            // Register a new user.
            let username =
                bridge::register_user(bridge_ip, "sapiens").expect("Failed to register user");
            println!(
                "Registered a new user - pass it as env: \nHUE_USERNAME={}",
                username
            );
            username
        }
    };

    let openai_client = async_openai::Client::new().with_api_key(
        std::env::var("OPENAI_API_KEY").expect("OPENAI_API_KEY not set in configuration file"),
    );

    let bridge = bridge::Bridge::new(bridge_ip, username);
    let bridge = Rc::new(bridge);

    let mut toolbox = Toolbox::default();

    toolbox.add_tool(RoomTool::new(bridge.clone()));
    toolbox.add_tool(StatusTool::new(bridge));
    toolbox.add_terminal_tool(ConcludeTool::default());
    toolbox.add_advanced_tool(PythonTool::default());

    // Sanitation
    // remove environment variables that could be used to access the host
    for (k, _) in std::env::vars() {
        std::env::remove_var(&k);
    }

    let task = args.task.clone();
    something(toolbox, openai_client, args.into(), task).await;
}