vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

ChatON:LoadJSON: ChatTemplates - global/system/user/assistant

Browse source 
Manually iterate the json object items using begin-end explicitly,
because the implicit iteration for loop related helpers for the
used json lib gives only the values and not a key-value pair.
This commit is contained in:
HanishKVC 2024-05-12 01:15:04 +05:30
parent 2efc09f2d0
commit 444d2ccf9c
1 changed files with 47 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

54
common/chaton.hpp
View file

@ -315,14 +315,54 @@ inline bool chaton_meta_load(const std::string &fname) {
std::ifstream f(fname); std::ifstream f(fname);
json conMeta = json::parse(f); json conMeta = json::parse(f);
for(auto it=conMeta.begin(); it != conMeta.end(); ++it) { for(auto it=conMeta.begin(); it != conMeta.end(); ++it) {
std::cout << it.key() << std::endl;
std::cout << it.value() << std::endl; auto group = it.key();
auto curMeta = conMeta[it.key()]; auto curTmpl = conMeta[group];
std::cout << curMeta.dump(4) << std::endl;
} std::string globalBegin = curTmpl[K_GLOBAL][K_BEGIN];
for(auto curTmpl: conMeta) { gCT.set_value<std::string>(group, { K_GLOBAL, K_BEGIN }, globalBegin);
std::cout << curTmpl.dump(4) << std::endl;
std::string globalEnd = curTmpl[K_GLOBAL][K_END];
gCT.set_value<std::string>(group, { K_GLOBAL, K_END }, globalEnd);
std::string systemBegin = curTmpl[K_SYSTEM][K_BEGIN];
gCT.set_value<std::string>(group, { K_SYSTEM, K_BEGIN }, systemBegin);
std::string systemPrefix = curTmpl[K_SYSTEM][K_PREFIX];
gCT.set_value<std::string>(group, { K_SYSTEM, K_PREFIX }, systemPrefix);
std::string systemSuffix = curTmpl[K_SYSTEM][K_SUFFIX];
gCT.set_value<std::string>(group, { K_SYSTEM, K_SUFFIX }, systemSuffix);
std::string systemEnd = curTmpl[K_SYSTEM][K_END];
gCT.set_value<std::string>(group, { K_SYSTEM, K_END }, systemEnd);
std::string userBegin = curTmpl[K_USER][K_BEGIN];
gCT.set_value<std::string>(group, { K_USER, K_BEGIN }, userBegin);
std::string userPrefix = curTmpl[K_USER][K_PREFIX];
gCT.set_value<std::string>(group, { K_USER, K_PREFIX }, userPrefix);
std::string userSuffix = curTmpl[K_USER][K_SUFFIX];
gCT.set_value<std::string>(group, { K_USER, K_SUFFIX }, userSuffix);
std::string userEnd = curTmpl[K_USER][K_END];
gCT.set_value<std::string>(group, { K_USER, K_END }, userEnd);
std::string assistantBegin = curTmpl[K_ASSISTANT][K_BEGIN];
gCT.set_value<std::string>(group, { K_ASSISTANT, K_BEGIN }, assistantBegin);
std::string assistantPrefix = curTmpl[K_ASSISTANT][K_PREFIX];
gCT.set_value<std::string>(group, { K_ASSISTANT, K_PREFIX }, assistantPrefix);
std::string assistantSuffix = curTmpl[K_ASSISTANT][K_SUFFIX];
gCT.set_value<std::string>(group, { K_ASSISTANT, K_SUFFIX }, assistantSuffix);
std::string assistantEnd = curTmpl[K_ASSISTANT][K_END];
gCT.set_value<std::string>(group, { K_ASSISTANT, K_END }, assistantEnd);
} }
LOGXLN("%s", gCT.dump("", "DBUG:ChatONMetaLoad:ChatTemplates:").c_str());
return true; return true;
} }