2025-03-22 03:15:49 +01:00
|
|
|
#ifndef R_H
|
|
|
|
#define R_H
|
2025-03-28 06:56:36 +01:00
|
|
|
#include "auth.h"
|
|
|
|
#include "utils.h"
|
2025-03-22 03:15:49 +01:00
|
|
|
#include <stdbool.h>
|
2025-03-28 01:55:04 +01:00
|
|
|
#include <string.h>
|
2025-03-22 03:15:49 +01:00
|
|
|
bool is_verbose = false;
|
|
|
|
|
2025-03-28 21:38:50 +01:00
|
|
|
#ifndef RD
|
2025-03-28 20:50:10 +01:00
|
|
|
#ifndef OLLAMA
|
|
|
|
char *models_api_url = "https://api.openai.com/v1/models";
|
|
|
|
// char *completions_api_url = "https://ollama.molodetz.nl/v1/chat/completions";
|
|
|
|
char *completions_api_url = "https://api.openai.com/v1/chat/completions";
|
|
|
|
char *advanced_model = "gpt-4o-mini";
|
|
|
|
char *fast_model = "gpt-3.5-turbo";
|
2025-03-28 21:38:50 +01:00
|
|
|
#endif
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#ifdef RD
|
|
|
|
|
|
|
|
char *models_api_url = "https://api.openai.com/v1/models";
|
|
|
|
char *completions_api_url = "https://api.deepinfra.com/v1/openai/chat/completions";
|
2025-03-28 23:00:17 +01:00
|
|
|
char *advanced_model = "meta-llama/Llama-3.3-70B-Instruct-Turbo";
|
2025-03-28 21:38:50 +01:00
|
|
|
//char *advanced_model = "meta-llama/Meta-Llama-3.1-8B-Instruct";
|
2025-03-28 23:00:17 +01:00
|
|
|
//char *advanced_model = "google/gemini-1.5-flash";
|
2025-03-28 21:38:50 +01:00
|
|
|
char *fast_model = "Qwen/Qwen2.5-Coder-32B-Instruct";
|
|
|
|
|
|
|
|
#endif
|
|
|
|
#ifdef OLLAMA
|
2025-03-28 20:50:10 +01:00
|
|
|
char *models_api_url = "https://ollama.molodetz.nl/v1/models";
|
|
|
|
char *completions_api_url = "https://ollama.molodetz.nl/v1/chat/completions";
|
|
|
|
char *advanced_model = "qwen2.5:3b";
|
2025-03-28 23:00:17 +01:00
|
|
|
//char *advanced_model = "qwen2.5-coder:0.5b";
|
2025-03-28 20:50:10 +01:00
|
|
|
char *fast_model = "qwen2.5:0.5b";
|
|
|
|
#endif
|
|
|
|
|
2025-03-28 06:56:36 +01:00
|
|
|
char *_model = NULL;
|
2025-03-28 01:55:04 +01:00
|
|
|
|
2025-03-28 06:55:22 +01:00
|
|
|
#define DB_FILE "~/.r.db"
|
|
|
|
#define PROMPT_TEMPERATURE 0.1
|
|
|
|
|
2025-03-28 01:55:04 +01:00
|
|
|
void set_prompt_model(const char *model) {
|
2025-03-28 06:56:36 +01:00
|
|
|
if (_model != NULL) {
|
|
|
|
free(_model);
|
|
|
|
}
|
|
|
|
_model = strdup(model);
|
2025-03-28 01:55:04 +01:00
|
|
|
}
|
|
|
|
|
2025-03-28 06:56:36 +01:00
|
|
|
const char *get_prompt_model() {
|
|
|
|
if (auth_type != AUTH_TYPE_API_KEY) {
|
|
|
|
if (_model == NULL) {
|
2025-03-28 20:50:10 +01:00
|
|
|
_model = strdup(fast_model);
|
2025-03-28 01:55:04 +01:00
|
|
|
}
|
2025-03-28 06:56:36 +01:00
|
|
|
} else if (_model == NULL) {
|
2025-03-28 20:50:10 +01:00
|
|
|
_model = strdup(advanced_model);
|
2025-03-28 06:56:36 +01:00
|
|
|
}
|
|
|
|
return _model;
|
2025-03-28 01:55:04 +01:00
|
|
|
}
|
|
|
|
|
2025-03-28 06:56:36 +01:00
|
|
|
#endif
|