This commit is contained in:
retoor 2025-03-19 22:02:39 +01:00
parent 1882ed4eb3
commit 8e7e5f96c3
2 changed files with 20 additions and 6 deletions

17
auth.h
View File

@ -19,10 +19,27 @@ enum AUTH_TYPE {
int auth_type = AUTH_TYPE_FREE;
void auth_init() {
char *api_key = NULL;
api_key = getenv("R_KEY");
if (api_key) {
auth_type = AUTH_TYPE_API_KEY;
return;
}
api_key = getenv("OPENAI_API_KEY");
if (api_key) {
auth_type = AUTH_TYPE_API_KEY;
return;
}
auth_type = AUTH_TYPE_FREE;
return;
}
const char *resolve_api_key() {
static char *api_key = NULL;
api_key = getenv("R_KEY");
if (api_key) {
auth_type = AUTH_TYPE_API_KEY;
return api_key;
}
api_key = getenv("OPENAI_API_KEY");

9
main.c
View File

@ -289,6 +289,7 @@ bool openai_include(char *path) {
void init() {
setbuf(stdout, NULL);
line_init();
auth_init();
const char *locale = setlocale(LC_ALL, NULL);
char payload[4096] = {0};
sprintf(payload, "Your locale is %s. User lang is %s.", locale, locale);
@ -296,12 +297,8 @@ void init() {
openai_system(payload);
if(!openai_include(".rcontext.txt")){
openai_include("~/.rcontext.txt");
}
#ifndef FREE_VERSION
fprintf(stderr, "%s", "\r✅ Commercial version. Type help for features.\n");
#else
fprintf(stderr, "%s","\r✅ Free version (GPT-3.5 Turbo), for you by retoor.\n");
#endif
}
fprintf(stderr, "\r \r");
}
int main(int argc, char *argv[]) {