1
0

utils.hpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673
  1. #pragma once
  2. #include "common.h"
  3. #include "log.h"
  4. #include "llama.h"
  5. #ifndef NDEBUG
  6. // crash the server in debug mode, otherwise send an http 500 error
  7. #define CPPHTTPLIB_NO_EXCEPTIONS 1
  8. #endif
  9. // increase max payload length to allow use of larger context size
  10. #define CPPHTTPLIB_FORM_URL_ENCODED_PAYLOAD_MAX_LENGTH 1048576
  11. #include "httplib.h"
  12. // Change JSON_ASSERT from assert() to GGML_ASSERT:
  13. #define JSON_ASSERT GGML_ASSERT
  14. #include "json.hpp"
  15. #include <random>
  16. #include <sstream>
  17. #include <string>
  18. #include <vector>
  19. #include <memory>
  20. #define DEFAULT_OAICOMPAT_MODEL "gpt-3.5-turbo"
  21. using json = nlohmann::ordered_json;
  22. #define SLT_INF(slot, fmt, ...) LOG_INF("slot %12.*s: id %2d | task %d | " fmt, 12, __func__, (slot).id, (slot).id_task, __VA_ARGS__)
  23. #define SLT_WRN(slot, fmt, ...) LOG_WRN("slot %12.*s: id %2d | task %d | " fmt, 12, __func__, (slot).id, (slot).id_task, __VA_ARGS__)
  24. #define SLT_ERR(slot, fmt, ...) LOG_ERR("slot %12.*s: id %2d | task %d | " fmt, 12, __func__, (slot).id, (slot).id_task, __VA_ARGS__)
  25. #define SLT_DBG(slot, fmt, ...) LOG_DBG("slot %12.*s: id %2d | task %d | " fmt, 12, __func__, (slot).id, (slot).id_task, __VA_ARGS__)
  26. #define SRV_INF(fmt, ...) LOG_INF("srv %12.*s: " fmt, 12, __func__, __VA_ARGS__)
  27. #define SRV_WRN(fmt, ...) LOG_WRN("srv %12.*s: " fmt, 12, __func__, __VA_ARGS__)
  28. #define SRV_ERR(fmt, ...) LOG_ERR("srv %12.*s: " fmt, 12, __func__, __VA_ARGS__)
  29. #define SRV_DBG(fmt, ...) LOG_DBG("srv %12.*s: " fmt, 12, __func__, __VA_ARGS__)
  30. #define QUE_INF(fmt, ...) LOG_INF("que %12.*s: " fmt, 12, __func__, __VA_ARGS__)
  31. #define QUE_WRN(fmt, ...) LOG_WRN("que %12.*s: " fmt, 12, __func__, __VA_ARGS__)
  32. #define QUE_ERR(fmt, ...) LOG_ERR("que %12.*s: " fmt, 12, __func__, __VA_ARGS__)
  33. #define QUE_DBG(fmt, ...) LOG_DBG("que %12.*s: " fmt, 12, __func__, __VA_ARGS__)
  34. template <typename T>
  35. static T json_value(const json & body, const std::string & key, const T & default_value) {
  36. // Fallback null to default value
  37. if (body.contains(key) && !body.at(key).is_null()) {
  38. try {
  39. return body.at(key);
  40. } catch (NLOHMANN_JSON_NAMESPACE::detail::type_error const &) {
  41. LOG_WRN("Wrong type supplied for parameter '%s'. Expected '%s', using default value\n", key.c_str(), json(default_value).type_name());
  42. return default_value;
  43. }
  44. } else {
  45. return default_value;
  46. }
  47. }
  48. //
  49. // tokenizer and input processing utils
  50. //
  51. static bool json_is_array_of_numbers(const json & data) {
  52. if (data.is_array()) {
  53. for (const auto & e : data) {
  54. if (!e.is_number_integer()) {
  55. return false;
  56. }
  57. }
  58. return true;
  59. }
  60. return false;
  61. }
  62. // is array having BOTH numbers & strings?
  63. static bool json_is_array_of_mixed_numbers_strings(const json & data) {
  64. bool seen_string = false;
  65. bool seen_number = false;
  66. if (data.is_array()) {
  67. for (const auto & e : data) {
  68. seen_string |= e.is_string();
  69. seen_number |= e.is_number_integer();
  70. if (seen_number && seen_string) {
  71. return true;
  72. }
  73. }
  74. }
  75. return false;
  76. }
  77. /**
  78. * this handles 2 cases:
  79. * - only string, example: "string"
  80. * - mixed string and tokens, example: [12, 34, "string", 56, 78]
  81. */
  82. static llama_tokens tokenize_mixed(const llama_context * ctx, const json & json_prompt, bool add_special, bool parse_special) {
  83. // If `add_bos` is true, we only add BOS, when json_prompt is a string,
  84. // or the first element of the json_prompt array is a string.
  85. llama_tokens prompt_tokens;
  86. if (json_prompt.is_array()) {
  87. bool first = true;
  88. for (const auto & p : json_prompt) {
  89. if (p.is_string()) {
  90. auto s = p.template get<std::string>();
  91. llama_tokens p;
  92. if (first) {
  93. p = common_tokenize(ctx, s, add_special, parse_special);
  94. first = false;
  95. } else {
  96. p = common_tokenize(ctx, s, false, parse_special);
  97. }
  98. prompt_tokens.insert(prompt_tokens.end(), p.begin(), p.end());
  99. } else {
  100. if (first) {
  101. first = false;
  102. }
  103. prompt_tokens.push_back(p.template get<llama_token>());
  104. }
  105. }
  106. } else {
  107. auto s = json_prompt.template get<std::string>();
  108. prompt_tokens = common_tokenize(ctx, s, add_special, parse_special);
  109. }
  110. return prompt_tokens;
  111. }
  112. /**
  113. * break the input "prompt" object into multiple prompt if needed, then tokenize them
  114. * this supports these cases:
  115. * - "prompt": "string"
  116. * - "prompt": [12, 34, 56]
  117. * - "prompt": [12, 34, "string", 56, 78]
  118. * and multiple prompts (multi-tasks):
  119. * - "prompt": ["string1", "string2"]
  120. * - "prompt": ["string1", [12, 34, 56]]
  121. * - "prompt": [[12, 34, 56], [78, 90, 12]]
  122. * - "prompt": [[12, 34, "string", 56, 78], [12, 34, 56]]
  123. */
  124. static std::vector<llama_tokens> tokenize_input_prompts(llama_context * ctx, const json & json_prompt, bool add_special, bool parse_special) {
  125. std::vector<llama_tokens> result;
  126. if (json_prompt.is_string() || json_is_array_of_mixed_numbers_strings(json_prompt)) {
  127. // string or mixed
  128. result.push_back(tokenize_mixed(ctx, json_prompt, add_special, parse_special));
  129. } else if (json_is_array_of_numbers(json_prompt)) {
  130. // array of tokens
  131. result.push_back(json_prompt.get<llama_tokens>());
  132. } else if (json_prompt.is_array()) {
  133. // array of prompts
  134. result.reserve(json_prompt.size());
  135. for (const auto & p : json_prompt) {
  136. if (p.is_string() || json_is_array_of_mixed_numbers_strings(p)) {
  137. result.push_back(tokenize_mixed(ctx, p, add_special, parse_special));
  138. } else if (json_is_array_of_numbers(p)) {
  139. // array of tokens
  140. result.push_back(p.get<llama_tokens>());
  141. } else {
  142. throw std::runtime_error("element of \"prompt\" must be a string, an list of tokens, or a list of mixed strings & tokens");
  143. }
  144. }
  145. } else {
  146. throw std::runtime_error("\"prompt\" must be a string, an list of tokens, a list of mixed strings & tokens, or a list of prompts");
  147. }
  148. if (result.empty()) {
  149. throw std::runtime_error("\"prompt\" must not be empty");
  150. }
  151. return result;
  152. }
  153. //
  154. // template utils
  155. //
  156. // format rerank task: [BOS]query[EOS][SEP]doc[EOS]
  157. static llama_tokens format_rerank(const struct llama_model * model, const llama_tokens & query, const llama_tokens & doc) {
  158. llama_tokens result;
  159. result.reserve(doc.size() + query.size() + 4);
  160. result.push_back(llama_token_bos(model));
  161. result.insert(result.end(), query.begin(), query.end());
  162. result.push_back(llama_token_eos(model));
  163. result.push_back(llama_token_sep(model));
  164. result.insert(result.end(), doc.begin(), doc.end());
  165. result.push_back(llama_token_eos(model));
  166. return result;
  167. }
  168. // format infill task
  169. static llama_tokens format_infill(
  170. const llama_context * ctx,
  171. const json & input_prefix,
  172. const json & input_suffix,
  173. const json & input_extra,
  174. const int n_batch,
  175. const int n_predict,
  176. const int n_ctx,
  177. const bool spm_infill,
  178. const llama_tokens & tokens_prompt
  179. ) {
  180. // TODO: optimize this block by reducing memory allocations and movement
  181. // use FIM repo-level pattern:
  182. // ref: https://arxiv.org/pdf/2409.12186
  183. //
  184. // [FIM_REP]myproject
  185. // [FIM_SEP]filename0
  186. // extra chunk 0
  187. // [FIM_SEP]filename1
  188. // extra chunk 1
  189. // ...
  190. // [FIM_SEP]filename
  191. // [FIM_PRE]prefix[FIM_SUF]suffix[FIM_MID]prompt
  192. //
  193. llama_tokens extra_tokens;
  194. extra_tokens.reserve(n_ctx);
  195. auto model = llama_get_model(ctx);
  196. auto tokens_prefix = tokenize_mixed(ctx, input_prefix, false, false);
  197. auto tokens_suffix = tokenize_mixed(ctx, input_suffix, false, false);
  198. if (llama_token_fim_rep(model) != LLAMA_TOKEN_NULL) {
  199. // TODO: make project name an input
  200. static const auto k_fim_repo = common_tokenize(ctx, "myproject\n", false, false);
  201. extra_tokens.push_back(llama_token_fim_rep(model));
  202. extra_tokens.insert(extra_tokens.end(), k_fim_repo.begin(), k_fim_repo.end());
  203. }
  204. for (const auto & chunk : input_extra) {
  205. // { "text": string, "filename": string }
  206. const std::string text = json_value(chunk, "text", std::string());
  207. const std::string filename = json_value(chunk, "filename", std::string("tmp"));
  208. if (llama_token_fim_sep(model) != LLAMA_TOKEN_NULL) {
  209. const auto k_fim_file = common_tokenize(ctx, filename + "\n", false, false);
  210. extra_tokens.insert(extra_tokens.end(), llama_token_fim_sep(model));
  211. extra_tokens.insert(extra_tokens.end(), k_fim_file.begin(), k_fim_file.end());
  212. } else {
  213. // chunk separator in binary form to avoid confusing the AI
  214. static const char k_chunk_prefix_str[] = {0x0a, 0x0a, 0x2d, 0x2d, 0x2d, 0x20, 0x73, 0x6e, 0x69, 0x70, 0x70, 0x65, 0x74, 0x20, 0x2d, 0x2d, 0x2d, 0x0a, 0x0a, 0x00};
  215. static const auto k_chunk_prefix_tokens = common_tokenize(ctx, k_chunk_prefix_str, false, false);
  216. extra_tokens.insert(extra_tokens.end(), k_chunk_prefix_tokens.begin(), k_chunk_prefix_tokens.end());
  217. }
  218. const auto chunk_tokens = common_tokenize(ctx, text, false, false);
  219. extra_tokens.insert(extra_tokens.end(), chunk_tokens.begin(), chunk_tokens.end());
  220. }
  221. if (llama_token_fim_sep(model) != LLAMA_TOKEN_NULL) {
  222. // TODO: current filename
  223. static const auto k_fim_file = common_tokenize(ctx, "filename\n", false, false);
  224. extra_tokens.insert(extra_tokens.end(), llama_token_fim_sep(model));
  225. extra_tokens.insert(extra_tokens.end(), k_fim_file.begin(), k_fim_file.end());
  226. }
  227. // for now pick FIM context to fit in a batch (ratio prefix:suffix = 3:1, TODO: configurable?)
  228. const int n_prefix_take = std::min<int>(tokens_prefix.size(), 3*(n_batch/4));
  229. const int n_suffix_take = std::min<int>(tokens_suffix.size(), std::max<int>(0, (n_batch/4) - (2 + tokens_prompt.size())));
  230. SRV_DBG("n_prefix_take = %d, n_suffix_take = %d, total = %d\n", n_prefix_take, n_suffix_take, (n_prefix_take + n_suffix_take));
  231. // fill the rest of the context with extra chunks
  232. const int n_extra_take = std::min<int>(std::max<int>(0, n_ctx - (n_batch) - 2*n_predict), extra_tokens.size());
  233. tokens_prefix.erase(tokens_prefix.begin(), tokens_prefix.begin() + tokens_prefix.size() - n_prefix_take);
  234. tokens_suffix.resize(n_suffix_take);
  235. tokens_prefix.insert(tokens_prefix.begin(), llama_token_fim_pre(model));
  236. tokens_prefix.insert(tokens_prefix.end(), tokens_prompt.begin(), tokens_prompt.end());
  237. tokens_suffix.insert(tokens_suffix.begin(), llama_token_fim_suf(model));
  238. auto embd_inp = spm_infill ? tokens_suffix : tokens_prefix;
  239. auto embd_end = spm_infill ? tokens_prefix : tokens_suffix;
  240. if (llama_add_bos_token(model)) {
  241. embd_inp.insert(embd_inp.begin(), llama_token_bos(model));
  242. }
  243. SRV_DBG("extra: n_ctx = %d, n_extra_take = %d, n_extra = %d\n", n_ctx, n_extra_take, (int) extra_tokens.size());
  244. // put the extra context before the FIM prefix
  245. embd_inp.insert(embd_inp.begin(), extra_tokens.end() - n_extra_take, extra_tokens.end());
  246. embd_inp.insert(embd_inp.end(), embd_end.begin(), embd_end.end());
  247. embd_inp.push_back(llama_token_fim_mid(model));
  248. return embd_inp;
  249. }
  250. // Format given chat. If tmpl is empty, we take the template from model metadata
  251. inline std::string format_chat(const struct llama_model * model, const std::string & tmpl, const std::vector<json> & messages) {
  252. std::vector<common_chat_msg> chat;
  253. for (size_t i = 0; i < messages.size(); ++i) {
  254. const auto & curr_msg = messages[i];
  255. std::string role = json_value(curr_msg, "role", std::string(""));
  256. std::string content;
  257. if (curr_msg.contains("content")) {
  258. if (curr_msg["content"].is_string()) {
  259. content = curr_msg["content"].get<std::string>();
  260. } else if (curr_msg["content"].is_array()) {
  261. for (const auto & part : curr_msg["content"]) {
  262. if (part.contains("text")) {
  263. content += "\n" + part["text"].get<std::string>();
  264. }
  265. }
  266. } else {
  267. throw std::runtime_error("Invalid 'content' type (ref: https://github.com/ggerganov/llama.cpp/issues/8367)");
  268. }
  269. } else {
  270. throw std::runtime_error("Missing 'content' (ref: https://github.com/ggerganov/llama.cpp/issues/8367)");
  271. }
  272. chat.push_back({role, content});
  273. }
  274. const auto formatted_chat = common_chat_apply_template(model, tmpl, chat, true);
  275. LOG_DBG("formatted_chat: '%s'\n", formatted_chat.c_str());
  276. return formatted_chat;
  277. }
  278. static std::string llama_get_chat_template(const struct llama_model * model) {
  279. std::string template_key = "tokenizer.chat_template";
  280. // call with NULL buffer to get the total size of the string
  281. int32_t res = llama_model_meta_val_str(model, template_key.c_str(), NULL, 0);
  282. if (res < 2) {
  283. return "";
  284. } else {
  285. std::vector<char> model_template(res + 1, 0);
  286. llama_model_meta_val_str(model, template_key.c_str(), model_template.data(), model_template.size());
  287. return std::string(model_template.data(), model_template.size() - 1);
  288. }
  289. }
  290. //
  291. // base64 utils (TODO: move to common in the future)
  292. //
  293. static const std::string base64_chars =
  294. "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  295. "abcdefghijklmnopqrstuvwxyz"
  296. "0123456789+/";
  297. static inline bool is_base64(uint8_t c) {
  298. return (isalnum(c) || (c == '+') || (c == '/'));
  299. }
  300. static inline std::vector<uint8_t> base64_decode(const std::string & encoded_string) {
  301. int i = 0;
  302. int j = 0;
  303. int in_ = 0;
  304. int in_len = encoded_string.size();
  305. uint8_t char_array_4[4];
  306. uint8_t char_array_3[3];
  307. std::vector<uint8_t> ret;
  308. while (in_len-- && (encoded_string[in_] != '=') && is_base64(encoded_string[in_])) {
  309. char_array_4[i++] = encoded_string[in_]; in_++;
  310. if (i == 4) {
  311. for (i = 0; i < 4; i++) {
  312. char_array_4[i] = base64_chars.find(char_array_4[i]);
  313. }
  314. char_array_3[0] = ((char_array_4[0] ) << 2) + ((char_array_4[1] & 0x30) >> 4);
  315. char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
  316. char_array_3[2] = ((char_array_4[2] & 0x3) << 6) + char_array_4[3];
  317. for (i = 0; (i < 3); i++) {
  318. ret.push_back(char_array_3[i]);
  319. }
  320. i = 0;
  321. }
  322. }
  323. if (i) {
  324. for (j = i; j < 4; j++) {
  325. char_array_4[j] = 0;
  326. }
  327. for (j = 0; j < 4; j++) {
  328. char_array_4[j] = base64_chars.find(char_array_4[j]);
  329. }
  330. char_array_3[0] = ((char_array_4[0] ) << 2) + ((char_array_4[1] & 0x30) >> 4);
  331. char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
  332. char_array_3[2] = ((char_array_4[2] & 0x3) << 6) + char_array_4[3];
  333. for (j = 0; j < i - 1; j++) {
  334. ret.push_back(char_array_3[j]);
  335. }
  336. }
  337. return ret;
  338. }
  339. //
  340. // random string / id
  341. //
  342. static std::string random_string() {
  343. static const std::string str("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz");
  344. std::random_device rd;
  345. std::mt19937 generator(rd());
  346. std::string result(32, ' ');
  347. for (int i = 0; i < 32; ++i) {
  348. result[i] = str[generator() % str.size()];
  349. }
  350. return result;
  351. }
  352. static std::string gen_chatcmplid() {
  353. return "chatcmpl-" + random_string();
  354. }
  355. //
  356. // other common utils
  357. //
  358. static bool ends_with(const std::string & str, const std::string & suffix) {
  359. return str.size() >= suffix.size() && 0 == str.compare(str.size() - suffix.size(), suffix.size(), suffix);
  360. }
  361. static size_t find_partial_stop_string(const std::string &stop, const std::string &text) {
  362. if (!text.empty() && !stop.empty()) {
  363. const char text_last_char = text.back();
  364. for (int64_t char_index = stop.size() - 1; char_index >= 0; char_index--) {
  365. if (stop[char_index] == text_last_char) {
  366. const std::string current_partial = stop.substr(0, char_index + 1);
  367. if (ends_with(text, current_partial)) {
  368. return text.size() - char_index - 1;
  369. }
  370. }
  371. }
  372. }
  373. return std::string::npos;
  374. }
  375. // TODO: reuse llama_detokenize
  376. template <class Iter>
  377. static std::string tokens_to_str(llama_context * ctx, Iter begin, Iter end) {
  378. std::string ret;
  379. for (; begin != end; ++begin) {
  380. ret += common_token_to_piece(ctx, *begin);
  381. }
  382. return ret;
  383. }
  384. // format incomplete utf-8 multibyte character for output
  385. static std::string tokens_to_output_formatted_string(const llama_context * ctx, const llama_token token) {
  386. std::string out = token == -1 ? "" : common_token_to_piece(ctx, token);
  387. // if the size is 1 and first bit is 1, meaning it's a partial character
  388. // (size > 1 meaning it's already a known token)
  389. if (out.size() == 1 && (out[0] & 0x80) == 0x80) {
  390. std::stringstream ss;
  391. ss << std::hex << (out[0] & 0xff);
  392. std::string res(ss.str());
  393. out = "byte: \\x" + res;
  394. }
  395. return out;
  396. }
  397. static bool server_sent_event(httplib::DataSink & sink, const char * event, const json & data) {
  398. const std::string str =
  399. std::string(event) + ": " +
  400. data.dump(-1, ' ', false, json::error_handler_t::replace) +
  401. "\n\n"; // required by RFC 8895 - A message is terminated by a blank line (two line terminators in a row).
  402. LOG_DBG("data stream, to_send: %s", str.c_str());
  403. return sink.write(str.c_str(), str.size());
  404. }
  405. //
  406. // OAI utils
  407. //
  408. static json oaicompat_completion_params_parse(
  409. const struct llama_model * model,
  410. const json & body, /* openai api json semantics */
  411. const std::string & chat_template) {
  412. json llama_params;
  413. // Apply chat template to the list of messages
  414. llama_params["prompt"] = format_chat(model, chat_template, body.at("messages"));
  415. // Handle "stop" field
  416. if (body.contains("stop") && body.at("stop").is_string()) {
  417. llama_params["stop"] = json::array({body.at("stop").get<std::string>()});
  418. } else {
  419. llama_params["stop"] = json_value(body, "stop", json::array());
  420. }
  421. // Handle "response_format" field
  422. if (body.contains("response_format")) {
  423. json response_format = json_value(body, "response_format", json::object());
  424. std::string response_type = json_value(response_format, "type", std::string());
  425. if (response_type == "json_object") {
  426. llama_params["json_schema"] = json_value(response_format, "schema", json::object());
  427. } else if (response_type == "json_schema") {
  428. json json_schema = json_value(response_format, "json_schema", json::object());
  429. llama_params["json_schema"] = json_value(json_schema, "schema", json::object());
  430. } else if (!response_type.empty() && response_type != "text") {
  431. throw std::runtime_error("response_format type must be one of \"text\" or \"json_object\", but got: " + response_type);
  432. }
  433. }
  434. // Handle "n" field
  435. int n_choices = json_value(body, "n", 1);
  436. if (n_choices != 1) {
  437. throw std::runtime_error("Only one completion choice is allowed");
  438. }
  439. // Handle "logprobs" field
  440. // TODO: The response format of this option is not yet OAI-compatible, but seems like no one really using it; We may need to fix it in the future
  441. if (json_value(body, "logprobs", false)) {
  442. llama_params["n_probs"] = json_value(body, "top_logprobs", 20);
  443. } else if (body.contains("top_logprobs") && !body.at("top_logprobs").is_null()) {
  444. throw std::runtime_error("top_logprobs requires logprobs to be set to true");
  445. }
  446. // Params supported by OAI but unsupported by llama.cpp
  447. static const std::vector<std::string> unsupported_params { "tools", "tool_choice" };
  448. for (const auto & param : unsupported_params) {
  449. if (body.contains(param)) {
  450. throw std::runtime_error("Unsupported param: " + param);
  451. }
  452. }
  453. // Copy remaining properties to llama_params
  454. // This allows user to use llama.cpp-specific params like "mirostat", ... via OAI endpoint.
  455. // See "launch_slot_with_task()" for a complete list of params supported by llama.cpp
  456. for (const auto & item : body.items()) {
  457. // Exception: if "n_predict" is present, we overwrite the value specified earlier by "max_tokens"
  458. if (!llama_params.contains(item.key()) || item.key() == "n_predict") {
  459. llama_params[item.key()] = item.value();
  460. }
  461. }
  462. return llama_params;
  463. }
  464. static json format_embeddings_response_oaicompat(const json & request, const json & embeddings) {
  465. json data = json::array();
  466. int32_t n_tokens = 0;
  467. int i = 0;
  468. for (const auto & elem : embeddings) {
  469. data.push_back(json{
  470. {"embedding", json_value(elem, "embedding", json::array())},
  471. {"index", i++},
  472. {"object", "embedding"}
  473. });
  474. n_tokens += json_value(elem, "tokens_evaluated", 0);
  475. }
  476. json res = json {
  477. {"model", json_value(request, "model", std::string(DEFAULT_OAICOMPAT_MODEL))},
  478. {"object", "list"},
  479. {"usage", json {
  480. {"prompt_tokens", n_tokens},
  481. {"total_tokens", n_tokens}
  482. }},
  483. {"data", data}
  484. };
  485. return res;
  486. }
  487. static json format_response_rerank(const json & request, const json & ranks) {
  488. json data = json::array();
  489. int32_t n_tokens = 0;
  490. int i = 0;
  491. for (const auto & rank : ranks) {
  492. data.push_back(json{
  493. {"index", i++},
  494. {"relevance_score", json_value(rank, "score", 0.0)},
  495. });
  496. n_tokens += json_value(rank, "tokens_evaluated", 0);
  497. }
  498. json res = json {
  499. {"model", json_value(request, "model", std::string(DEFAULT_OAICOMPAT_MODEL))},
  500. {"object", "list"},
  501. {"usage", json {
  502. {"prompt_tokens", n_tokens},
  503. {"total_tokens", n_tokens}
  504. }},
  505. {"results", data}
  506. };
  507. return res;
  508. }
  509. static bool is_valid_utf8(const std::string & str) {
  510. const unsigned char* bytes = reinterpret_cast<const unsigned char*>(str.data());
  511. const unsigned char* end = bytes + str.length();
  512. while (bytes < end) {
  513. if (*bytes <= 0x7F) {
  514. // 1-byte sequence (0xxxxxxx)
  515. bytes++;
  516. } else if ((*bytes & 0xE0) == 0xC0) {
  517. // 2-byte sequence (110xxxxx 10xxxxxx)
  518. if (end - bytes < 2 || (bytes[1] & 0xC0) != 0x80)
  519. return false;
  520. bytes += 2;
  521. } else if ((*bytes & 0xF0) == 0xE0) {
  522. // 3-byte sequence (1110xxxx 10xxxxxx 10xxxxxx)
  523. if (end - bytes < 3 || (bytes[1] & 0xC0) != 0x80 || (bytes[2] & 0xC0) != 0x80)
  524. return false;
  525. bytes += 3;
  526. } else if ((*bytes & 0xF8) == 0xF0) {
  527. // 4-byte sequence (11110xxx 10xxxxxx 10xxxxxx 10xxxxxx)
  528. if (end - bytes < 4 || (bytes[1] & 0xC0) != 0x80 ||
  529. (bytes[2] & 0xC0) != 0x80 || (bytes[3] & 0xC0) != 0x80)
  530. return false;
  531. bytes += 4;
  532. } else {
  533. // Invalid UTF-8 lead byte
  534. return false;
  535. }
  536. }
  537. return true;
  538. }
  539. static json format_tokenizer_response(const json & tokens) {
  540. return json {
  541. {"tokens", tokens}
  542. };
  543. }
  544. static json format_detokenized_response(const std::string & content) {
  545. return json {
  546. {"content", content}
  547. };
  548. }
  549. static json format_logit_bias(const std::vector<llama_logit_bias> & logit_bias) {
  550. json data = json::array();
  551. for (const auto & lb : logit_bias) {
  552. data.push_back(json{
  553. {"bias", lb.bias},
  554. {"token", lb.token},
  555. });
  556. }
  557. return data;
  558. }
  559. static std::string safe_json_to_str(json data) {
  560. return data.dump(-1, ' ', false, json::error_handler_t::replace);
  561. }