You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
749 B
22 lines
749 B
3 weeks ago
|
#pragma once
|
||
|
|
|
||
|
|
#include "ggml.h"
|
||
|
|
// Change JSON_ASSERT from assert() to GGML_ASSERT:
|
||
|
|
#define JSON_ASSERT GGML_ASSERT
|
||
|
|
#include "json.hpp"
|
||
|
|
|
||
|
|
std::string json_schema_to_grammar(const nlohmann::ordered_json & schema,
|
||
|
|
bool force_gbnf = false);
|
||
|
|
|
||
|
|
struct common_grammar_builder {
|
||
|
|
std::function<std::string(const std::string &, const std::string &)> add_rule;
|
||
|
|
std::function<std::string(const std::string &, const nlohmann::ordered_json &)> add_schema;
|
||
|
|
std::function<void(nlohmann::ordered_json &)> resolve_refs;
|
||
|
|
};
|
||
|
|
|
||
|
|
struct common_grammar_options {
|
||
|
|
bool dotall = false;
|
||
|
|
};
|
||
|
|
|
||
|
|
std::string build_grammar(const std::function<void(const common_grammar_builder &)> & cb, const common_grammar_options & options = {});
|