ChatOn: Capture the idea

This commit is contained in:
HanishKVC 2024-04-22 19:59:45 +05:30
parent 628b299106
commit 2146a253e8
2 changed files with 23 additions and 0 deletions

View file

@ -65,6 +65,7 @@ add_library(${TARGET} STATIC
train.cpp
ngram-cache.h
ngram-cache.cpp
chaton.hpp
)
if (BUILD_SHARED_LIBS)

22
common/chaton.hpp Normal file
View file

@ -0,0 +1,22 @@
#pragma once
/***
* Keep chatting with model and needed role tagging using special tokens simple and flexible, while building on existing interactive flow
* 1. Use a json file to configure the needed tags for each of the supported chat-handshake-template-standard
* a. system-prefix, system-suffix,
* b. user-prefix, user-suffix,
* c. reverse-prompt
* d. global-begin-marker, global-end-marker
* e. per-msg-begin-marker, per-msg-end-marker
* f. is per-msg-begin-marker used for system+user combo
* 2. Give the below option to user wrt system prompt, this should give the flexibility to either keep system prompt simple or complex in a flexible yet simple way.
* a. the system prompt they specify using -f, is used as is with parse_special when tokenising or
* b. whether the system prefix and suffix is added, but without parse_special tokenisation of system-prompt provided by user.
* 3. chat-apply-template uses the json file, which was loaded, to decide on how to generate the tagged messages for tokenisation
* a. input: [ { role: message }, { role: message}, ....]
* b. output: [ {flag: data}, { flag: data}, {flag: data}, ....]
* * flag is whether to do parse_special for this data, during tokenization or not
*
*/