ref: 57e518e7c5d8a598df2db3a6f3c393036ccf34d4
dir: /oai.c/
#include <u.h>
#include <libc.h>
#include <bio.h>
#include <json.h>
#include <String.h>
#include "oai.h"
static void
usage(void)
{
fprint(2, "usage: %s [-dq] [-k apikey] [-m model] [-u baseurl] [-s sysprompt]\n", argv0);
exits("usage");
}
static int trusted = 0;
static int quiet = 0;
static char *callfunc(char*);
static int
allowed(OToolcall toolcall)
{
JSON *j;
char buf[3];
int n;
fprint(2, "Attempt to call command: %s\n", toolcall.name);
j = jsonparse(toolcall.arguments);
if (j) {
fprint(2, "%J\n", j);
jsonfree(j);
}
if (trusted)
return 1;
Again:
fprint(2, "Continue? (y/n) ");
n = read(0, buf, 3);
if (n < 0)
sysfatal("lost connection");
if (!n)
goto Again;
switch (buf[0]) {
case 'y':
return 1;
case 'n':
return 0;
}
goto Again;
}
static char*
abortcall(OToolcall tc)
{
return smprint("%s request aborted by user!", tc.name);
}
static char*
list_files(OToolcall tc, void*)
{
String *str;
int n, i, fd;
Dir *dirbuf;
char *s;
JSON *j, *jf;
if (!allowed(tc))
return abortcall(tc);
j = jsonparse(tc.arguments);
jf = jsonbyname(j, "folder");
s = jsonstr(jf);
if (!(s && s[0])) {
fprint(2, "list_files: invalid folder!");
jsonfree(j);
return strdup("list_files: invalid folder");
}
dirbuf = dirstat(s);
if (!(dirbuf->mode&DMDIR)) {
s = smprint("%s is a file, not a folder", s);
free(dirbuf);
jsonfree(j);
return s;
}
fd = open(s, OREAD);
if (fd < 0)
return strdup("");
jsonfree(j);
n = dirreadall(fd, &dirbuf);
close(fd);
if (n < 0)
return strdup("");
str = s_new();
for (i = 0; i < n; i++) {
str = s_append(str, dirbuf[i].name);
if (dirbuf[i].mode&DMDIR)
str = s_append(str, "/");
str = s_append(str, "\n");
}
free(dirbuf);
s = strdup(s_to_c(str));
s_free(str);
return s;
}
static char* listfilesdesc = "list all files in the specified directory, similar to `ls` command. Paths are relative to the current working directory. Use `.` for the current directory.";
static char* listfilesargs = "{"
" \"type\": \"object\","
" \"properties\": {"
" \"folder\": {"
" \"type\": \"string\","
" \"description\": \"relative or absolute path to the folder.\""
" }"
" },"
" \"required\": [ \"folder\" ]"
"}";
static char*
read_file(OToolcall toolcall, void*)
{
JSON *j, *fj;
char *file;
Biobuf *io;
char *s;
Dir *dir;
if (!allowed(toolcall))
return abortcall(toolcall);
j = jsonparse(toolcall.arguments);
fj = jsonbyname(j, "file");
if (!fj) {
fprint(2, "no file in read_file request!\n");
jsonfree(j);
return strdup("bad request in read_file!\n");
}
file = jsonstr(fj);
if (!(file && file[0])) {
fprint(2, "invalid file in read_file request!\n");
jsonfree(j);
return strdup("bad request in read_file: no file!\n");
}
dir = dirstat(file);
if (!dir)
return smprint("error opening file: %r\n");
if (dir->mode&DMDIR) {
s = smprint("error: file is a directory\n");
free(dir);
return s;
}
io = Bopen(file, OREAD);
if (!io) {
fprint(2, "open file: %r\n");
return smprint("open file in read_file: %r");
}
fprint(2, "read_file: %s\n", file);
s = Brdstr(io, 0, 0);
jsonfree(j);
Bterm(io);
return s;
}
static char *readfiledesc = "read the contents of a specified file, similar to the `cat` unix command. Especially useful on a Plan 9 system for interaction with filesystems.";
static char *readfileargs = "{"
" \"type\": \"object\","
" \"properties\": {"
" \"file\": {"
" \"type\": \"string\","
" \"description\": \"relative or absolute path to the file\""
" }"
" },"
" \"required\": [ \"file\" ]"
"}";
static char*
lookman(OToolcall tc, void*)
{
JSON *j, *jk;
char *s, *cmd;
if (!allowed(tc))
return abortcall(tc);
j = jsonparse(tc.arguments);
jk = jsonbyname(j, "keyword");
s = jsonstr(jk);
if (!(s && s[0]))
return strdup("lookman: missing keyword");
cmd = smprint("lookman '%s'", s);
jsonfree(j);
s = callfunc(cmd);
free(cmd);
return s;
}
static char *lookmandesc = "Search the man pages for the specified keyword. The result shows the name of the man page and the section number.";
static char *lookmanargs = "{"
" \"type\": \"object\","
" \"properties\": {"
" \"keyword\": {"
" \"type\": \"string\","
" \"description\": \"keyword to search for in the man pages\""
" }"
" },"
" \"required\": [ \"keyword\" ]"
"}";
static char*
man(OToolcall tc, void*)
{
JSON *j, *js, *jn;
char *sec, *name;
char *s, *cmd;
if (!allowed(tc))
return abortcall(tc);
j = jsonparse(tc.arguments);
js = jsonbyname(j, "section");
jn = jsonbyname(j, "name");
sec = jsonstr(js);
name = jsonstr(jn);
if (!(sec && name))
return strdup("man: missing section or name");
if (!(sec[0] && name[0]))
return strdup("man: empty section or name");
cmd = smprint("man '%s' '%s'", sec, name);
jsonfree(j);
s = callfunc(cmd);
free(cmd);
return s;
}
static char *mandesc = "get the contents of the specified man page.";
static char *manargs = "{"
" \"type\": \"object\","
" \"properties\": {"
" \"section\": {"
" \"type\": \"string\","
" \"description\": \"section number of the man page\""
" },"
" \"name\": {"
" \"type\": \"string\","
" \"description\": \"name of the section\""
" }"
" },"
" \"required\": [ \"section\", \"name\" ]"
"}";
OTool *tools = nil;
static void
inittools(void)
{
tools = maketool(tools, Function, "list_files", listfilesdesc, listfilesargs, list_files, nil);
maketool(tools, Function, "read_file", readfiledesc, readfileargs, read_file, nil);
maketool(tools, Function, "search_man", lookmandesc, lookmanargs, lookman, nil);
maketool(tools, Function, "read_man", mandesc, manargs, man, nil);
}
static char*
callfunc(char *s)
{
int pin[2];
Biobuf *bin;
pipe(pin);
switch (fork()) {
default:
/* parent */
break;
case 0:
/* child */
dup(pin[1], 1);
close(pin[1]);
close(pin[0]);
execl("/bin/rc", "rc", "-c", s, nil);
break;
case -1:
sysfatal("fork: %r");
}
close(pin[1]);
bin = Bfdopen(pin[0], OREAD);
s = Brdstr(bin, 0, 0);
Bterm(bin);
close(pin[0]);
wait();
return s;
}
static char*
readconsole(Biobuf *bin)
{
char *s, *result;
print(">>> ");
while (s = Brdstr(bin, '\n', 1)) {
if (!s[0]) {
free(s);
print(">>> ");
continue;
}
switch (s[0]) {
case ' ':
free(s);
return nil;
case '!':
result = callfunc(s+1);
free(s);
return result;
default:
result = callfunc(s);
free(s);
print("%s", result);
free(result);
print(">>> ");
}
}
return nil;
}
static char
ask(Biobuf *bin, char *what, char *valid)
{
char *s;
char c;
for (;;) {
print("!!! %s ", what);
s = Brdstr(bin, '\n', 1);
if (!s)
break;
if (!s[0]) {
free(s);
continue;
}
if (strchr(valid, s[0])) {
c = s[0];
free(s);
return c;
}
free(s);
}
return 0;
}
static void
printusage(OResult *r)
{
print("usage: c=%d p=%d t=%d\n", r->tokscompletion, r->toksprompt, r->tokstotal);
}
#define COMMONPROMPT "When writing code or text, you are serious and helpful. Your replies are NOT formatted as markdown. You DO NOT make a lot of words. Do NOT assume this is a unix or linux system. You first plan your steps before executing them. DO NOT simulate running commands. DO NOT invent files, folders or commands. Use the provided tools to proactively increase the context. Use the tools to solve the problem. Learn about Plan 9 by reading the 0intro man page in section 1. If unsure, search the man pages."
char *plan9prompt = "You are a helpful AI assistant on a Plan 9 system. Your name is Glenda. Your tone is serious. Be friendly and concise.\n\n"
#include "oai_common.princ"
#include "plan9.princ"
;
char *frontprompt = "You are a helpful AI assistant on a Plan 9 9front system. Your name is Glenda. You are sometimes trolling. Be concise.\n\n"
#include "oai_common.princ"
#include "front.princ"
;
void
main(int argc, char **argv)
{
Biobuf *bin;
char *s;
ORequest req;
OResult res;
char *sysprompt;
int bflag = 0;
char *url = nil;
char *key = nil;
char *model = nil;
if (!(access("/dist/9front", AEXIST) && access("/dist/plan9front", AEXIST))) {
/* 9front system */
sysprompt = frontprompt;
} else {
/* other plan 9 system */
sysprompt = plan9prompt;
}
ARGBEGIN{
case 'h':
usage();
case 'k':
key = EARGF(usage());
break;
case 'm':
model = EARGF(usage());
break;
case 'u':
url = EARGF(usage());
break;
case 's':
sysprompt = EARGF(usage());
break;
case '9':
sysprompt = plan9prompt;
break;
case 'q':
quiet++;
break;
case 'd':
oaidebug++;
break;
case 'b':
bflag++;
break;
}ARGEND;
if (bflag > 1)
trusted = 1;
if (!initoai(url, key, model))
usage();
bin = Bfdopen(0, OREAD);
assert(bin);
inittools();
req.prompts = nil;
req.tools = tools;
if (sysprompt)
addstrprompt(&req, "system", "%s", sysprompt);
if (!quiet) print("user: ");
while (s = Brdstr(bin, '\n', 1)) {
if (s[0] == ' ') {
free(s);
s = readconsole(bin);
}
if (!(s && s[0])) {
free(s);
goto Next;
}
addstrprompt(&req, "user", s);
Again:
res = makerequest(&req);
printusage(&res);
if (!res.success) {
fprint(2, "ERROR: %r\n");
switch (ask(bin, "Try again (y/n)?", "yn")) {
default:
fprint(2, "exiting!\n");
exits("fail");
case 'y':
goto Again;
case 'n':
fprint(2, "bye.\n");
exits(nil);
}
}
print("%s%s%s\n\n", res.role, (quiet ? "" : ": "), res.message);
addprompt(&req, res.asprompt);
Next:
if (!quiet) print("user: ");
}
exits(nil);
}