From 206c3539dc117823091880c48ac93212bf75562d Mon Sep 17 00:00:00 2001
From: wangzy <wangziyi@pjlab.org.cn>
Date: Tue, 6 Feb 2024 10:58:50 +0800
Subject: [PATCH] reset IPython

---
 agent/streaming_inference.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/agent/streaming_inference.py b/agent/streaming_inference.py
index 37ccf4f..e5532b7 100644
--- a/agent/streaming_inference.py
+++ b/agent/streaming_inference.py
@@ -54,7 +54,7 @@ DEFAULT_PROMPT = (
 
 
 def parse_args():
-    parser = argparse.ArgumentParser()
+    parser = argparse.ArgumentParser(description='Math Code Interpreter')
     parser.add_argument('--backend',
                         type=str,
                         default='lmdeploy',
@@ -572,7 +572,8 @@ def predict(args):
                 ]
                 item['pred'].append('😭' if not lang else
                                     extract_answer(lang[-1]['content']) or '😭')
-
+                agent._interpreter_executor.actions[
+                    'IPythonInteractive'].reset()
             except Exception as e:
                 err = str(traceback.format_exc())
                 print(f'Error processing index {item["idx"]}: {e}\n{err}')