Upload btn_retry.py
Browse files- 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 |
-
|
|
|
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
|