luboslenco
129 lines
4.1 KiB
C
129 lines
4.1 KiB
C
|
|
#include "../global.h"
|
|
|
|
static i32 text_to_text_node_backend = CONSOLE_MODEL_QWEN;
|
|
|
|
static char *text_to_text_node_guide = "Reply with C code only - no explanations. Do not chain statements - declare intermediary variables. Place the code "
|
|
"inside 'void main()' function. Wrap the code in a ```c markdown fence.";
|
|
|
|
static char *text_to_text_node_project_contents(void) {
|
|
buffer_t *encoded = util_encode_project(g_project);
|
|
char *json = armpack_decode_to_json_omit_buffers(encoded);
|
|
return string("/* Current project state:\n%s\n*/\n", json);
|
|
}
|
|
|
|
string_array_t *text_to_text_node_qwen_args(char *dir) {
|
|
char *prompt_file = string("%s%sprompt.txt", dir, PATH_SEP);
|
|
string_array_t *argv = any_array_create_from_raw(
|
|
(void *[]){
|
|
string("%s/%s", dir, neural_node_llama_bin()),
|
|
"-m",
|
|
string("%s/Qwen3.6-27B-Q4_K_M.gguf", dir),
|
|
"-ngl",
|
|
"99",
|
|
"-c",
|
|
"20000",
|
|
"--single-turn",
|
|
"--prompt-cache",
|
|
string("%s/context.bin", dir),
|
|
"--file",
|
|
prompt_file,
|
|
NULL,
|
|
},
|
|
13);
|
|
return argv;
|
|
}
|
|
|
|
string_array_t *text_to_text_node_grok_args(char *dir) {
|
|
char *prompt_file = string("%s%sprompt.txt", dir, PATH_SEP);
|
|
string_array_t *argv = any_array_create_from_raw(
|
|
(void *[]){
|
|
"grok",
|
|
"--prompt-file",
|
|
prompt_file,
|
|
"--output-format",
|
|
"plain",
|
|
"--tools",
|
|
"",
|
|
NULL,
|
|
},
|
|
10);
|
|
return argv;
|
|
}
|
|
|
|
string_array_t *text_to_text_node_claude_args(char *dir, char *prompt) {
|
|
string_array_t *argv = any_array_create_from_raw(
|
|
(void *[]){
|
|
"claude",
|
|
"--print",
|
|
"--output-format",
|
|
"text",
|
|
"--tools",
|
|
"",
|
|
"--append-system-prompt-file",
|
|
string("%s%sapi.h", dir, PATH_SEP),
|
|
prompt,
|
|
NULL,
|
|
},
|
|
10);
|
|
return argv;
|
|
}
|
|
|
|
void text_to_text_node_check_result(void (*done)(char *)) {
|
|
iron_delay_idle_sleep();
|
|
if (iron_exec_async_done == 1) {
|
|
char *file = string("%s%soutput.txt", neural_node_dir(), PATH_SEP);
|
|
if (iron_file_exists(file)) {
|
|
buffer_t *b = iron_load_blob(file);
|
|
char *s = sys_buffer_to_string(b);
|
|
|
|
if (text_to_text_node_backend == CONSOLE_MODEL_QWEN) {
|
|
i32 think_end = string_last_index_of(s, "</think>\n\n");
|
|
i32 eot = string_last_index_of(s, "[end of text]");
|
|
if (think_end >= 0 && eot > think_end) {
|
|
s = substring(s, think_end + 10, eot);
|
|
}
|
|
}
|
|
s = trim_end(s);
|
|
|
|
done(s);
|
|
base_redraw_console();
|
|
}
|
|
sys_remove_update(text_to_text_node_check_result);
|
|
}
|
|
}
|
|
|
|
void text_to_text_node_clear(void) {
|
|
char *dir = neural_node_dir();
|
|
iron_delete_file(string("%s%sprompt.txt", dir, PATH_SEP));
|
|
iron_delete_file(string("%s%scontext.bin", dir, PATH_SEP));
|
|
iron_delete_file(string("%s%soutput.txt", dir, PATH_SEP));
|
|
iron_delete_file(string("%s%sapi.h", dir, PATH_SEP));
|
|
}
|
|
|
|
void text_to_text_node_run(char *prompt, void (*done)(char *)) {
|
|
char *dir = neural_node_dir();
|
|
if (string_equals(file_read_directory(dir)->buffer[0], "")) {
|
|
iron_create_directory(dir);
|
|
}
|
|
text_to_text_node_backend = g_config->console_model;
|
|
|
|
char *reference = string("%s\n%s\n%s\n", minic_api_header_generate(), text_to_text_node_project_contents(), text_to_text_node_guide);
|
|
|
|
string_array_t *argv;
|
|
if (text_to_text_node_backend == CONSOLE_MODEL_CLAUDE) {
|
|
iron_file_save_bytes(string("%s%sapi.h", dir, PATH_SEP), (buffer_t *)u8_array_create_from_string(reference), 0);
|
|
argv = text_to_text_node_claude_args(dir, prompt);
|
|
}
|
|
else {
|
|
char *prompt_file = string("%s%sprompt.txt", dir, PATH_SEP);
|
|
char *full = string("%s%s", reference, prompt);
|
|
iron_file_save_bytes(prompt_file, (buffer_t *)u8_array_create_from_string(full), 0);
|
|
argv = text_to_text_node_backend == CONSOLE_MODEL_GROK ? text_to_text_node_grok_args(dir) : text_to_text_node_qwen_args(dir);
|
|
}
|
|
|
|
iron_exec_async_output_file = string("%s%soutput.txt", dir, PATH_SEP);
|
|
iron_exec_async(argv->buffer[0], argv->buffer);
|
|
sys_notify_on_update(text_to_text_node_check_result, done);
|
|
}
|