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

ChatON: Cleanup/Refine initial go at tmpl_apply_ex_capi

Browse source
This commit is contained in:
HanishKVC 2024-05-07 11:44:25 +05:30
parent 7c288d3dfc
commit 43a3a91b03
1 changed files with 11 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

15
common/chaton.hpp
View file

@ -608,7 +608,10 @@ inline int32_t chaton_tmpl_apply_ex_capi(
std::string taggedMsgs;
std::string types;
std::vector<int> lens;
int32_t taggedLength = chaton_tmpl_apply_ex(tmpl, vMsgs, taggedMsgs, types, lens, alertAssistantAtEnd);
if (!chaton_tmpl_apply_ex(tmpl, vMsgs, taggedMsgs, types, lens, alertAssistantAtEnd)) {
return -1;
}
int32_t taggedLength = taggedMsgs.size();
if (taggedLength <= 0) {
return taggedLength;
}
@ -616,9 +619,13 @@ inline int32_t chaton_tmpl_apply_ex_capi(
strlcpy(dest, taggedMsgs.c_str(), destLength);
}
if (*pNumParts > 0) {
strlcpy(partTypes, types.c_str(), *pNumParts);
for(int i=0; i < *pNumParts; i++) {
partLengths[i] = lens[i];
if (partTypes != nullptr) {
strlcpy(partTypes, types.c_str(), *pNumParts);
}
if (partLengths != nullptr) {
for(int i=0; i < *pNumParts; i++) {
partLengths[i] = lens[i];
}
}
}
*pNumParts = types.length();