Upload 2 files

mods/btn_rag.py

@@ -30,7 +30,7 @@ def init(cfg):
             # ========== 需要临时注入的内容 ==========
             model.venv_create('rag')
             t_rag = chat_template('system', _rag)
-            model.eval_t(t_rag, _n_keep, _n_discard)
+            model.eval_t(t_rag, _n_keep, _n_discard, chat_template.im_start_token)
             yield model.venv_info
 
     cfg['btn_rag_fn'] = {

mods/btn_submit.py

@@ -36,7 +36,7 @@ def init(cfg):
             model.venv_create('usr')
             t_msg = history[-1][0]
             t_msg = chat_template(_usr, t_msg)
-            model.eval_t(t_msg, _n_keep, _n_discard)
+            model.eval_t(t_msg, _n_keep, _n_discard, chat_template.im_start_token)
             yield history, model.venv_info
             # ========== 模型输出 ==========
             if cfg['btn_stop_status']: