talksik
87 lines
2.5 KiB
C
87 lines
2.5 KiB
C
#include "intelligence.h"
|
|
#include "utils.h"
|
|
#include <curl/curl.h>
|
|
#include <string.h>
|
|
|
|
static char *find_ai_response(char *json_response) {
|
|
printf("Parsing response: %s \n", json_response);
|
|
char *start = strstr(json_response, "\"text\":\"");
|
|
if (!start)
|
|
return NULL;
|
|
|
|
start += 8; // Skip past "text":"
|
|
char *end = strchr(start, '"');
|
|
if (!end)
|
|
return NULL;
|
|
|
|
size_t len = end - start;
|
|
char *result = malloc(len + 1);
|
|
strncpy(result, start, len);
|
|
result[len] = '\0';
|
|
return result;
|
|
}
|
|
|
|
char *ai_ask(char *question) {
|
|
CURL *curl;
|
|
CURLcode res;
|
|
struct curl_slist *headers = NULL;
|
|
|
|
// Initialize response buffer
|
|
struct curl_response_data response = {0};
|
|
response.data = malloc(1);
|
|
response.size = 0;
|
|
|
|
// Create JSON payload
|
|
char json_payload[2048];
|
|
snprintf(json_payload, sizeof(json_payload),
|
|
"{"
|
|
"\"model\":\"claude-3-haiku-20240307\","
|
|
"\"max_tokens\":256,"
|
|
"\"messages\":[{\"role\":\"user\",\"content\":\"Make it one sentence answer: %s\"}]"
|
|
"}",
|
|
question);
|
|
|
|
curl = curl_easy_init();
|
|
if (curl) {
|
|
char *anthropic_api_key = getenv("ANTHROPIC_API_KEY");
|
|
if (!anthropic_api_key) {
|
|
printf("ANTHROPIC_API_KEY not set\n");
|
|
free(response.data);
|
|
curl_easy_cleanup(curl);
|
|
return NULL;
|
|
}
|
|
char auth_header[256];
|
|
snprintf(auth_header, sizeof(auth_header), "x-api-key: %s",
|
|
anthropic_api_key);
|
|
headers = curl_slist_append(headers, "content-type: application/json");
|
|
headers = curl_slist_append(headers, auth_header);
|
|
headers = curl_slist_append(headers, "anthropic-version: 2023-06-01");
|
|
|
|
curl_easy_setopt(curl, CURLOPT_URL,
|
|
"https://api.anthropic.com/v1/messages");
|
|
curl_easy_setopt(curl, CURLOPT_HTTPHEADER, headers);
|
|
curl_easy_setopt(curl, CURLOPT_POST, 1L);
|
|
curl_easy_setopt(curl, CURLOPT_POSTFIELDS, json_payload);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, curl_write_callback_response);
|
|
curl_easy_setopt(curl, CURLOPT_WRITEDATA, &response);
|
|
|
|
res = curl_easy_perform(curl);
|
|
if (res != CURLE_OK) {
|
|
fprintf(stderr, "curl failed: %s\n", curl_easy_strerror(res));
|
|
free(response.data);
|
|
curl_slist_free_all(headers);
|
|
curl_easy_cleanup(curl);
|
|
return NULL;
|
|
}
|
|
|
|
curl_slist_free_all(headers);
|
|
curl_easy_cleanup(curl);
|
|
}
|
|
|
|
// Parse AI response from complete response
|
|
char *answer = find_ai_response(response.data);
|
|
free(response.data);
|
|
|
|
return answer;
|
|
}
|