添加测试

This commit is contained in:
Your Name 2023-04-19 14:49:34 +08:00
parent 966af4d4c5
commit 6bf89dfa2d
2 changed files with 8 additions and 1 deletions

View File

@ -44,7 +44,7 @@ WEB_PORT = -1
MAX_RETRY = 2 MAX_RETRY = 2
# OpenAI模型选择是gpt4现在只对申请成功的人开放 # OpenAI模型选择是gpt4现在只对申请成功的人开放
LLM_MODEL = "gpt-3.5-turbo" # 可选 "chatglm" LLM_MODEL = "gpt-3.5-turbo" # 可选 ↓↓↓
AVAIL_LLM_MODELS = ["gpt-3.5-turbo", "api2d-gpt-3.5-turbo", "gpt-4", "api2d-gpt-4", "chatglm"] AVAIL_LLM_MODELS = ["gpt-3.5-turbo", "api2d-gpt-3.5-turbo", "gpt-4", "api2d-gpt-4", "chatglm"]
# 本地LLM模型如ChatGLM的执行方式 CPU/GPU # 本地LLM模型如ChatGLM的执行方式 CPU/GPU

View File

@ -37,6 +37,12 @@ def test_解析一个Python项目():
for cookies, cb, hist, msg in 解析一个Python项目(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port): for cookies, cb, hist, msg in 解析一个Python项目(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port):
print(cb) print(cb)
def test_解析一个Cpp项目():
from crazy_functions.解析项目源代码 import 解析一个C项目
txt = "crazy_functions/test_project/cpp/cppipc"
for cookies, cb, hist, msg in 解析一个C项目(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port):
print(cb)
def test_Latex英文润色(): def test_Latex英文润色():
from crazy_functions.Latex全文润色 import Latex英文润色 from crazy_functions.Latex全文润色 import Latex英文润色
txt = "crazy_functions/test_project/latex/attention" txt = "crazy_functions/test_project/latex/attention"
@ -80,6 +86,7 @@ test_批量翻译PDF文档()
test_谷歌检索小助手() test_谷歌检索小助手()
test_总结word文档() test_总结word文档()
test_下载arxiv论文并翻译摘要() test_下载arxiv论文并翻译摘要()
test_解析一个Cpp项目()
input("程序完成,回车退出。") input("程序完成,回车退出。")
print("退出。") print("退出。")