-
Notifications
You must be signed in to change notification settings - Fork 235
/
tongueworm.pl
92 lines (86 loc) · 4.19 KB
/
tongueworm.pl
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
use Irssi;
use LWP::UserAgent;
use JSON::PP;
use strict;
use warnings;
our $VERSION = "1.0.0";
our %IRSSI = (
authors => 'terminaldweller',
contact => 'https://terminaldweller.com',
name => 'tongueworm',
description => 'rewrites the input line using openai chatgpt',
license => 'GPL3 or newer',
url => 'https://github.com/irssi/scripts.irssi.org',
);
# adds the tongueworm command. the default question will just rephrase your input line
# while keeping the language the same. technically you can ask it to even translate it.
# you can bind the function like so:
# /bind ^R command tongueworm
# the above bind the command to ctrl-r. please note that if you're using vim_mode,
# then the command will only bind in INSERT mode. if you want in NORMAL mode use
# the facilities vim_mode provides to bind the function.
# settings settable by the user:
# /set wormtongue_openai_api_key XXXXXXXXXXXX
# /set wormtongue_model gpt-3.5-turbo
# /set wormtongue_role user
# /set wormtongue_temperature 700
# the temperature value is divided by 1000, so a value of 700 would become 700/1000, i.e. 0.7
# /set wormtongue_debug 0
# /set wormtongue_request my_awesome_request
Irssi::settings_add_str('misc','wormtongue_openai_api_key', '');
Irssi::settings_add_str('misc','wormtongue_model', 'gpt-3.5-turbo');
Irssi::settings_add_str('misc','wormtongue_role', 'user');
Irssi::settings_add_int('misc','wormtongue_temperature', 700);
Irssi::settings_add_bool('misc','wormtongue_debug', 0);
Irssi::settings_add_str('misc', 'wormtongue_request', 'rephrase this: ');
Irssi::settings_add_str('misc', 'wormtongue_provider', 'ollama');
Irssi::settings_add_str('misc', 'wormtongue_server_endpoint', '');
Irssi::settings_add_int('misc', 'wormtongue_timeout', 5000);
Irssi::settings_add_str('misc', 'wormtongue_system_prompt', 'you are good at rephrasing sentences');
sub wormtongue {
my $debug = Irssi::settings_get_bool('wormtongue_debug');
my $timeout = Irssi::settings_get_int('wormtongue_timeout')/1000;
my $ua = LWP::UserAgent->new(timeout => $timeout);
my $server_endpoint = Irssi::settings_get_str('wormtongue_server_endpoint');
my $req = HTTP::Request->new(POST => $server_endpoint);
$req->header('Content-Type'=>'application/json');
my $openai_api_key = Irssi::settings_get_str('wormtongue_openai_api_key');
$req->header('Authorization'=>"Bearer $openai_api_key");
my $ai_model = Irssi::settings_get_str('wormtongue_model');
my $ai_role = Irssi::settings_get_str('wormtongue_role');
my $ai_temp = Irssi::settings_get_int('wormtongue_temperature')/1000;
my $provider = Irssi::settings_get_str('wormtongue_provider');
my $question = Irssi::settings_get_str('wormtongue_request');
my $content = $question.Irssi::parse_special('$L', 0, 0);
my $system_prompt = Irssi::settings_get_str('wormtongue_system_prompt');
my $post_data = "";
if ($provider eq "chatgpt") {
$post_data = '{"model" : "'.$ai_model.'", "temperature" : '.$ai_temp.', "messages" : [{"role" : "'.$ai_role.'","content" : "'.$content.'"}]}';
Irssi::print($post_data) if ($debug == 1);
} elsif ($provider eq "ollama") {
$post_data = '{"model" : "'.$ai_model.'", "format" : "json", "prompt" : "'.$content.'" , "system" : "'.$system_prompt.'", "stream": false, "options" : {"temperature" : '.$ai_temp.'}}';
Irssi::print($post_data) if ($debug == 1);
}
$req->content($post_data);
my $resp = $ua->request($req);
my $result = "";
Irssi::print($resp) if ($debug == 1);
if ($resp->is_success) {
my $message = $resp->decoded_content;
Irssi::print("Received reply: $message") if ($debug == 1);
my $json_parser = JSON::PP->new;
my $data = $json_parser->decode($message);
if ($provider eq "chatgpt"){
$result = $data->{choices}[0]{message}{content};
}elsif ($provider eq "ollama") {
$result = $data->{response};
}
Irssi::print($result) if ($debug == 1);
Irssi::gui_input_set($result);
}
else {
Irssi::print("HTTP POST error code: ".$resp->code);
Irssi::print("HTTP POST error message: ".$resp->message);
}
}
Irssi::command_bind('tongueworm', \&wormtongue);