135 lines
5.0 KiB
C
135 lines
5.0 KiB
C
|
#ifndef R_TOOLS_H
|
||
|
#define R_TOOLS_H
|
||
|
|
||
|
#include <json-c/json.h>
|
||
|
#include <json-c/json_object.h>
|
||
|
#include <string.h>
|
||
|
#include "http_curl.h"
|
||
|
|
||
|
struct json_object * tool_description_http_get();
|
||
|
|
||
|
struct json_object * tools_descriptions(){
|
||
|
struct json_object *root = json_object_new_array();
|
||
|
json_object_array_add(root, tool_description_http_get());
|
||
|
return root;
|
||
|
}
|
||
|
|
||
|
|
||
|
char * tool_function_http_get(char * url) {
|
||
|
return curl_get(url);
|
||
|
}
|
||
|
|
||
|
struct json_object * tool_description_http_get(){
|
||
|
// Create root object
|
||
|
struct json_object *root = json_object_new_object();
|
||
|
json_object_object_add(root, "type", json_object_new_string("function"));
|
||
|
|
||
|
// Create function object
|
||
|
struct json_object *function = json_object_new_object();
|
||
|
json_object_object_add(function, "name", json_object_new_string("http_fetch"));
|
||
|
json_object_object_add(function, "description", json_object_new_string("Get the contents of an url."));
|
||
|
|
||
|
// Create parameters object
|
||
|
struct json_object *parameters = json_object_new_object();
|
||
|
json_object_object_add(parameters, "type", json_object_new_string("object"));
|
||
|
|
||
|
// Create properties object
|
||
|
struct json_object *properties = json_object_new_object();
|
||
|
struct json_object *url = json_object_new_object();
|
||
|
json_object_object_add(url, "type", json_object_new_string("string"));
|
||
|
json_object_object_add(url, "description", json_object_new_string("fetch url contents."));
|
||
|
json_object_object_add(properties, "url", url);
|
||
|
|
||
|
json_object_object_add(parameters, "properties", properties);
|
||
|
|
||
|
// Add required array
|
||
|
struct json_object *required = json_object_new_array();
|
||
|
json_object_array_add(required, json_object_new_string("url"));
|
||
|
json_object_object_add(parameters, "required", required);
|
||
|
|
||
|
// Add additionalProperties as false
|
||
|
json_object_object_add(parameters, "additionalProperties", json_object_new_boolean(0));
|
||
|
|
||
|
// Add parameters to function object
|
||
|
json_object_object_add(function, "parameters", parameters);
|
||
|
|
||
|
// Add strict mode
|
||
|
json_object_object_add(function, "strict", json_object_new_boolean(1));
|
||
|
|
||
|
// Add function object to root
|
||
|
json_object_object_add(root, "function", function);
|
||
|
|
||
|
|
||
|
return root;
|
||
|
}
|
||
|
|
||
|
struct json_object * tools_execute(struct json_object * tools_array){
|
||
|
//struct json_object * tools_result = json_object_new_object();
|
||
|
struct json_object * tools_result_messages = json_object_new_array();
|
||
|
int array_len = json_object_array_length(tools_array);
|
||
|
// Iterate over array
|
||
|
for (int i = 0; i < array_len; i++) {
|
||
|
struct json_object *obj = json_object_array_get_idx(tools_array, i);
|
||
|
struct json_object * tool_result = json_object_new_object();
|
||
|
|
||
|
|
||
|
json_object_object_add(tool_result, "tool_call_id", json_object_new_string(json_object_get_string(json_object_object_get(obj, "id"))));
|
||
|
|
||
|
json_object_object_add(tool_result, "role", json_object_new_string("tool"));
|
||
|
|
||
|
// Get "id"
|
||
|
struct json_object *id_obj;
|
||
|
if (json_object_object_get_ex(obj, "id", &id_obj)) {
|
||
|
|
||
|
|
||
|
printf("ID: %s\n", json_object_get_string(id_obj));
|
||
|
}
|
||
|
|
||
|
// Get "type"
|
||
|
struct json_object *type_obj;
|
||
|
if (json_object_object_get_ex(obj, "type", &type_obj)) {
|
||
|
printf("Type: %s\n", json_object_get_string(type_obj));
|
||
|
}
|
||
|
|
||
|
// Get "function" object
|
||
|
struct json_object *function_obj;
|
||
|
if (json_object_object_get_ex(obj, "function", &function_obj)) {
|
||
|
// Get function "name"
|
||
|
struct json_object *name_obj;
|
||
|
char * function_name = NULL;
|
||
|
if (json_object_object_get_ex(function_obj, "name", &name_obj)) {
|
||
|
function_name = (char *)json_object_get_string(name_obj);
|
||
|
printf("Function Name: %s\n", json_object_get_string(name_obj));
|
||
|
}
|
||
|
|
||
|
// Get function "arguments"
|
||
|
if(!strcmp(function_name,"http_fetch")){
|
||
|
|
||
|
|
||
|
struct json_object *arguments_obj;
|
||
|
if (json_object_object_get_ex(function_obj, "arguments", &arguments_obj)) {
|
||
|
struct json_object *arguments = json_tokener_parse(json_object_get_string(arguments_obj));;
|
||
|
struct json_object *url_obj;
|
||
|
if (json_object_object_get_ex(arguments, "url", &url_obj)) {
|
||
|
char * url = (char *)json_object_get_string(url_obj);
|
||
|
char * http_result = curl_get(url);
|
||
|
printf("URL: %s\n", json_object_get_string(url_obj));
|
||
|
printf("content: %s\n",http_result);
|
||
|
json_object_object_add(tool_result, "content", json_object_new_string(http_result));
|
||
|
}
|
||
|
|
||
|
printf("Function Arguments: %s\n", json_object_get_string(arguments_obj));
|
||
|
}
|
||
|
}
|
||
|
json_object_array_add(tools_result_messages, tool_result);
|
||
|
}
|
||
|
|
||
|
printf("\n"); // Add spacing for readability
|
||
|
}
|
||
|
|
||
|
//json_object_object_add(tools_result, "messages", tools_result_messages);
|
||
|
return tools_result_messages;
|
||
|
}
|
||
|
|
||
|
#endif
|