Limour commited on
Commit
89cf6c5
1 Parent(s): 670a6e9

Upload btn_retry.py

Browse files
Files changed (1) hide show
  1. mods/btn_retry.py +2 -1
mods/btn_retry.py CHANGED
@@ -22,7 +22,8 @@ def init(cfg):
22
  raise RuntimeError
23
  # ========== 回滚到上一次用户输入 ==========
24
  if not model.venv_revision('usr'):
25
- return history, model.venv_info
 
26
  # ========== 需要临时注入的内容 ==========
27
  if len(_rag) > 0:
28
  model.venv_create('rag') # 记录 venv_idx
 
22
  raise RuntimeError
23
  # ========== 回滚到上一次用户输入 ==========
24
  if not model.venv_revision('usr'):
25
+ yield history, model.venv_info
26
+ return
27
  # ========== 需要临时注入的内容 ==========
28
  if len(_rag) > 0:
29
  model.venv_create('rag') # 记录 venv_idx