chunzhi/chatgpt_academic
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add test for juptyer notebook plugin

Browse Source
This commit is contained in:
Your Name 2023-04-22 16:19:36 +08:00
parent 559b3cd6bb
commit 880be21bf7
3 changed files with 19 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@ -145,3 +145,4 @@ cradle*
debug* debug*
private* private*
crazy_functions/test_project/pdf_and_word crazy_functions/test_project/pdf_and_word
crazy_functions/test_samples

12
crazy_functions/crazy_functions_test.py
View File

@ -108,6 +108,13 @@ def test_联网回答问题():
print("当前问答：", cb[-1][-1].replace("\n"," ")) print("当前问答：", cb[-1][-1].replace("\n"," "))
for i, it in enumerate(cb): print亮蓝(it[0]); print亮黄(it[1]) for i, it in enumerate(cb): print亮蓝(it[0]); print亮黄(it[1])
def test_解析ipynb文件():
from crazy_functions.解析JupyterNotebook import 解析ipynb文件
txt = "crazy_functions/test_samples"
for cookies, cb, hist, msg in 解析ipynb文件(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port):
print(cb)
# test_解析一个Python项目() # test_解析一个Python项目()
# test_Latex英文润色() # test_Latex英文润色()
# test_Markdown中译英() # test_Markdown中译英()
@ -116,9 +123,8 @@ def test_联网回答问题():
# test_总结word文档() # test_总结word文档()
# test_下载arxiv论文并翻译摘要() # test_下载arxiv论文并翻译摘要()
# test_解析一个Cpp项目() # test_解析一个Cpp项目()
# test_联网回答问题()
test_联网回答问题() test_解析ipynb文件()
input("程序完成，回车退出。") input("程序完成，回车退出。")
print("退出。") print("退出。")

13
crazy_functions/解析JupyterNotebook.py
View File

@ -1,5 +1,5 @@
from toolbox import update_ui from toolbox import update_ui
from toolbox import CatchException, report_execption from toolbox import CatchException, report_execption, write_results_to_file
fast_debug = True fast_debug = True
@ -79,9 +79,9 @@ def ipynb解释(file_manifest, project_folder, llm_kwargs, plugin_kwargs, chatbo
pfg.run_file_split(max_token_limit=1024) pfg.run_file_split(max_token_limit=1024)
n_split = len(pfg.sp_file_contents) n_split = len(pfg.sp_file_contents)
inputs_array = [f"This is a Jupyter Notebook file, tell me about Each Block in Chinese. Focus Just On Code." + inputs_array = [r"This is a Jupyter Notebook file, tell me about Each Block in Chinese. Focus Just On Code." +
f"If a block starts with `Markdown` which means it's a markdown block in ipynbipynb. " + r"If a block starts with `Markdown` which means it's a markdown block in ipynbipynb. " +
f"Start a new line for a block and block num use Chinese." + r"Start a new line for a block and block num use Chinese." +
f"\n\n{frag}" for frag in pfg.sp_file_contents] f"\n\n{frag}" for frag in pfg.sp_file_contents]
inputs_show_user_array = [f"{f}的分析如下" for f in pfg.sp_file_tag] inputs_show_user_array = [f"{f}的分析如下" for f in pfg.sp_file_tag]
sys_prompt_array = ["You are a professional programmer."] * n_split sys_prompt_array = ["You are a professional programmer."] * n_split
@ -100,8 +100,13 @@ def ipynb解释(file_manifest, project_folder, llm_kwargs, plugin_kwargs, chatbo
# <-------- 整理结果，退出 ----------> # <-------- 整理结果，退出 ---------->
block_result = " \n".join(gpt_response_collection) block_result = " \n".join(gpt_response_collection)
chatbot.append(("解析的结果如下", block_result)) chatbot.append(("解析的结果如下", block_result))
history.extend(["解析的结果如下", block_result])
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面 yield from update_ui(chatbot=chatbot, history=history) # 刷新界面
# <-------- 写入文件，退出 ---------->
res = write_results_to_file(history)
chatbot.append(("完成了吗？", res))
yield from update_ui(chatbot=chatbot, history=history) # 刷新界面
@CatchException @CatchException
def 解析ipynb文件(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port): def 解析ipynb文件(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port):