fix

YunCheng66 · Mar 20, 2023 · 8ca2ea6 · 8ca2ea6
1 parent 43d813c
commit 8ca2ea6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/modules/context.py b/modules/context.py
@@ -36,9 +36,10 @@ def append(self, query, output) -> str:
 
     def update_last(self, query, output) -> None:
         self.history[-1] = (query, output)
+        self.rh[-1] = (query, output)
 
     def refresh_last(self) -> None:
-        query, output = self.history[-1]
+        query, output = self.rh[-1]
         self.rh[-1] = (query, parse_codeblock(output))
 
     def clear(self):

diff --git a/modules/ui.py b/modules/ui.py
@@ -26,7 +26,7 @@ def predict(query, max_length, top_p, temperature):
             flag = False
         else:
             ctx.update_last(query, output)
-        yield ctx.history, ""
+        yield ctx.rh, ""
     ctx.refresh_last()
     yield ctx.rh, ""