aipack 0.7.7-WIP

Command Agent runner to accelerate production coding with genai.
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

use crate::support::os::{OsType, current_os};
use crossterm::{
	cursor::MoveToColumn,
	execute,
	style::{Attribute, Color, Print, ResetColor, SetAttribute, SetForegroundColor},
	terminal::{Clear, ClearType},
};
use genai::ModelIden;
use std::io::stdout;

#[allow(unused_must_use)] // TODO: need to remove and make this function return error
pub fn print_key_env_missing(missing_env_name: &str, model_iden: &ModelIden, _interactive: bool) {
	let mut stdout = stdout();

	let model_name: &str = &model_iden.model_name;
	let provider_name: &str = model_iden.adapter_kind.as_str();
	let os_set_env_line = os_set_env_line(missing_env_name);

	execute!(
		stdout,
		Clear(ClearType::CurrentLine),
		MoveToColumn(0),
		SetForegroundColor(Color::Red),
		Print("\n======== Environment Error\n\n"),
		Clear(ClearType::CurrentLine),
		MoveToColumn(0),
		ResetColor,
		SetForegroundColor(Color::Red),
		SetAttribute(Attribute::Bold),
		Clear(ClearType::CurrentLine),
		MoveToColumn(0),
		Print("Error: "),
		ResetColor,
		Print("Cannot connect to model "),
		SetForegroundColor(Color::Yellow),
		Print("'"),
		Print(model_name),
		Print("'"),
		ResetColor,
		Print(" for provider "),
		SetForegroundColor(Color::Yellow),
		Print("'"),
		Print(provider_name),
		Print("'."),
		ResetColor,
		SetForegroundColor(Color::Red),
		SetAttribute(Attribute::Bold),
		Print("\n"),
		Clear(ClearType::CurrentLine),
		MoveToColumn(0),
		Print("Cause: "),
		ResetColor,
		Print("Environment variable "),
		SetForegroundColor(Color::Magenta),
		Print(format!("'{}'", missing_env_name)),
		ResetColor,
		Print(" missing. Make sure to set it for this terminal."),
		SetAttribute(Attribute::Bold),
		Print("\n"),
		Clear(ClearType::CurrentLine),
		MoveToColumn(0),
		Print("\n"),
		Clear(ClearType::CurrentLine),
		MoveToColumn(0),
		Print("You can set environment variable like:"),
		Print("\n"),
		Clear(ClearType::CurrentLine),
		MoveToColumn(0),
		Print("\n"),
		Clear(ClearType::CurrentLine),
		MoveToColumn(0),
		ResetColor,
		SetForegroundColor(Color::Green),
		Print(os_set_env_line),
		Print("\n\n"),
		Clear(ClearType::CurrentLine),
		MoveToColumn(0),
		SetForegroundColor(Color::Red),
		Print("========================\n"),
		ResetColor,
	);
}

pub fn os_set_env_line(env_name: &str) -> String {
	match current_os() {
		OsType::Mac | OsType::Linux | OsType::Unknown => format!(r#"export {env_name}="YOUR_{env_name}_VALUE_HERE""#),
		OsType::Windows => format!(r#"$env:{env_name} = 'YOUR_{env_name}_VALUE_HERE'"#),
	}
}