Everything safe.

This commit is contained in:
retoor 2025-01-04 08:35:39 +01:00
parent de80b013c0
commit 016e335a23
6 changed files with 56 additions and 75 deletions

3
chat.h
View File

@ -16,7 +16,6 @@ void chat_free(){
_prompt = NULL;
}
char * chat_json(char * role, char * message){
chat_free();
message_add(role,message);
@ -28,4 +27,4 @@ json_object_object_add(root_object, "temperature", json_object_new_double(prompt
return (char *)json_object_to_json_string_ext(root_object, JSON_C_TO_STRING_PRETTY);
}
#endif
#endif

17
http.h
View File

@ -96,22 +96,20 @@ char *http_post(const char *hostname, char *url, char *data)
SSL_set_tlsext_host_name(ssl, hostname);
SSL_set_fd(ssl, sock);
int buffer_size = 4096;
char *buffer = (char *)malloc(buffer_size);
if (SSL_connect(ssl) <= 0)
{
ERR_print_errors_fp(stderr);
}
else
{
//printf("Connected with %s encryption\n", SSL_get_cipher(ssl));
size_t len = strlen(data);
char *request = (char *)malloc(len + 4096);
request[0] = 0;
sprintf(request,
"POST %s HTTP/1.1\r\n"
"Content-Length: %ld\r\n"
@ -120,11 +118,11 @@ char *http_post(const char *hostname, char *url, char *data)
"Authorization: Bearer %s\r\n"
"Connection: close\r\n\r\n%s",
url, len, api_key, data);
SSL_write(ssl, request, strlen(request));
free(request);
free(request);
int bytes;
int bytes_total = 0;
while ((bytes = SSL_read(ssl, buffer + bytes_total, buffer_size - 1)) > 0)
@ -173,7 +171,6 @@ char *http_get(const char *hostname, char *url)
}
else
{
//printf("Connected with %s encryption\n", SSL_get_cipher(ssl));
char request[buffer_size];
request[0] = 0;
@ -207,4 +204,4 @@ char *http_get(const char *hostname, char *url)
return buffer;
}
#endif
#endif

96
main.c
View File

@ -5,29 +5,29 @@
#include <locale.h>
#include <stdio.h>
char * get_prompt_from_args(int c, char **argv){
char * get_prompt_from_args(int c, char **argv) {
char * prompt = malloc(1024*1024 + 1);
prompt[0] = 0;
for(int i = 1; i < c; i++){
for(int i = 1; i < c; i++) {
if(argv[i][0] == '-')
break;
strncat(prompt, argv[i], 1024*1024);
if(i < c - 1){
if(i < c - 1) {
strncat(prompt, " ", 1024*1024);
}else{
} else {
strncat(prompt, ".", 1024*1024);
}
}
if(!*prompt){
if(!*prompt) {
free(prompt);
return NULL;
}
return prompt;
}
bool try_prompt(int argc,char*argv[]){
char * prompt = get_prompt_from_args(argc, argv);
if(prompt != NULL){
bool try_prompt(int argc,char*argv[]) {
char * prompt = get_prompt_from_args(argc, argv);
if(prompt != NULL) {
char * response = openai_chat("user",prompt);
parse_markdown_to_ansi(response);
printf("\n");
@ -40,59 +40,54 @@ bool try_prompt(int argc,char*argv[]){
void help();
void render(char *);
void serve(){
void serve() {
render("Starting server. *Put executables in a dir named cgi-bin and they will behave as webpages.*");
int res = system("python3 -m http.server --cgi");
// Thanks tsoding!
(void)res;
}
void render(char * content){
void render(char * content) {
parse_markdown_to_ansi(content);
printf("\n\n");
}
void repl(){
void repl() {
line_init();
setbuf(stdout, NULL);
char *line;
char *previous_line = NULL;
while((line = line_read("> "))){
if(!line || !*line){
while((line = line_read("> "))) {
if(!line || !*line) {
line = previous_line;
}
if(!line || !*line)
continue;
previous_line = line;
if(line[0] == '!'){
if(line[0] == '!') {
plugin_run(line + 1);
continue;
}
if(!strncmp(line,"exit", 4)){
if(!strncmp(line,"exit", 4)) {
exit(0);
}
if(!strncmp(line,"help",4)){
if(!strncmp(line,"help",4)) {
help();
continue;
}
if(!strncmp(line,"serve",5)){
if(!strncmp(line,"serve",5)) {
serve();
}
if(!strncmp(line,"spar ",5)){
if(!strncmp(line,"spar ",5)) {
char * response = line+5;
while(true){
while(true) {
render(response);
sleep(2);
//line = line_read("> ");
//if(!*line)
response = openai_chat("user",response);
}
response = openai_chat("user",response);
}
}
if(!strncmp(line,"ls",2) || !strncmp(line,"list",4)){
if(!strncmp(line,"ls",2) || !strncmp(line,"list",4)) {
int offset = 2;
if(!strncmp(line,"list",4)){
if(!strncmp(line,"list",4)) {
offset = 4;
}
char * command = (char *)malloc(strlen(line) + 42);
@ -112,8 +107,7 @@ void repl(){
}
}
void help(){
void help() {
char help_text[1024*1024] = {0};
char * template = "# Help\n"
"Written by retoor@molodetz.nl.\n\n"
@ -133,13 +127,13 @@ void help(){
" - **google search** and actions with those results.\n"
" - **reminders**.\n"
" - predefined **templates** for **reviewing** / **refactoring** so you can personalize.\n";
sprintf(help_text,template,prompt_temperature,prompt_model,prompt_max_tokens);
render(help_text);
sprintf(help_text,template,prompt_temperature,prompt_model,prompt_max_tokens);
render(help_text);
}
void openai_include(char * path){
void openai_include(char * path) {
FILE * file = fopen(path,"r");
if(file == NULL){
if(file == NULL) {
return;
}
fseek(file, 0, SEEK_END);
@ -148,7 +142,7 @@ void openai_include(char * path){
char * buffer = (char *)malloc(size);
size_t read = fread(buffer,1,size,file);
if(read == 0){
if(read == 0) {
return;
}
@ -159,27 +153,27 @@ void openai_include(char * path){
free(buffer);
}
void init(){
line_init();
const char *locale = setlocale(LC_ALL, NULL);
char payload[4096] = {0};
sprintf(payload, "User locale is %s. User lang is %s.\n"
"You are Retoor. Use a lot of markdown in response.\n"
"Be confident and short in answers.\n"
"You divide things by zero if you have to."
, locale, locale);
printf("%s","Loading...");
openai_system(payload);
openai_include("context.txt");
printf("%s", "\rLoaded! Type help for feautures.\n");
void init() {
line_init();
const char *locale = setlocale(LC_ALL, NULL);
char payload[4096] = {0};
sprintf(payload, "User locale is %s. User lang is %s.\n"
"You are Retoor. Use a lot of markdown in response.\n"
"Be confident and short in answers.\n"
"You divide things by zero if you have to."
, locale, locale);
printf("%s","Loading...");
openai_system(payload);
openai_include("context.txt");
printf("%s", "\rLoaded! Type help for feautures.\n");
}
int main(int argc, char *argv[]){
int main(int argc, char *argv[]) {
init();
if(try_prompt(argc,argv))
return 0;
repl();
return 0;
}
}

View File

@ -29,4 +29,4 @@ void message_free(){
_message_array = NULL;
}
}
#endif
#endif

View File

@ -27,7 +27,6 @@ bool openai_system(char * content){
return is_done;
}
char *openai_chat(char * role, char * content){
const char *hostname = "api.openai.com";
char *url = "/v1/chat/completions";
@ -50,7 +49,6 @@ char *openai_chat(char * role, char * content){
return NULL;
}
// Get the first element of the "choices" array
struct json_object *first_choice = json_object_array_get_idx(choices_array, 0);
if (!first_choice) {
fprintf(stderr, "Failed to get the first element of 'choices'.\n");
@ -58,7 +56,6 @@ char *openai_chat(char * role, char * content){
return NULL;
}
// Extract the "message" object
struct json_object *message_object;
if (!json_object_object_get_ex(first_choice, "message", &message_object)) {
fprintf(stderr, "Failed to get 'message' object.\n");
@ -66,19 +63,14 @@ char *openai_chat(char * role, char * content){
return NULL;
}
// Print the "message" object
// printf("Message object:\n%s\n", json_object_to_json_string_ext(message_object, JSON_C_TO_STRING_PRETTY));
message_add("assistant",(char *)json_object_get_string(json_object_object_get(message_object, "content")));
// Clean up
free(data);
free(result);
result = strdup((char *)json_object_get_string(json_object_object_get(message_object, "content")));
json_object_put(parsed_json);
//printf("Parsed JSON:\n%s\n", json_object_to_json_string_ext(parsed_json, JSON_C_TO_STRING_PRETTY));
return result;
}
#endif
#endif

View File

@ -8,9 +8,8 @@ bool plugin_construct(){
if(plugin_initialized)
return true;
Py_Initialize();
Py_Initialize();
// Check if Python initialized successfully
if (!Py_IsInitialized()) {
fprintf(stderr, "Failed to initialize Python interpreter\n");
return plugin_initialized;