添加测试
Browse files- config.py +1 -1
- crazy_functions/crazy_functions_test.py +7 -0
config.py
CHANGED
@@ -44,7 +44,7 @@ WEB_PORT = -1
|
|
44 |
MAX_RETRY = 2
|
45 |
|
46 |
# OpenAI模型选择是(gpt4现在只对申请成功的人开放)
|
47 |
-
LLM_MODEL = "gpt-3.5-turbo" # 可选
|
48 |
AVAIL_LLM_MODELS = ["gpt-3.5-turbo", "api2d-gpt-3.5-turbo", "gpt-4", "api2d-gpt-4", "chatglm"]
|
49 |
|
50 |
# 本地LLM模型如ChatGLM的执行方式 CPU/GPU
|
|
|
44 |
MAX_RETRY = 2
|
45 |
|
46 |
# OpenAI模型选择是(gpt4现在只对申请成功的人开放)
|
47 |
+
LLM_MODEL = "gpt-3.5-turbo" # 可选 ↓↓↓
|
48 |
AVAIL_LLM_MODELS = ["gpt-3.5-turbo", "api2d-gpt-3.5-turbo", "gpt-4", "api2d-gpt-4", "chatglm"]
|
49 |
|
50 |
# 本地LLM模型如ChatGLM的执行方式 CPU/GPU
|
crazy_functions/crazy_functions_test.py
CHANGED
@@ -37,6 +37,12 @@ def test_解析一个Python项目():
|
|
37 |
for cookies, cb, hist, msg in 解析一个Python项目(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port):
|
38 |
print(cb)
|
39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
def test_Latex英文润色():
|
41 |
from crazy_functions.Latex全文润色 import Latex英文润色
|
42 |
txt = "crazy_functions/test_project/latex/attention"
|
@@ -80,6 +86,7 @@ test_批量翻译PDF文档()
|
|
80 |
test_谷歌检索小助手()
|
81 |
test_总结word文档()
|
82 |
test_下载arxiv论文并翻译摘要()
|
|
|
83 |
|
84 |
input("程序完成,回车退出。")
|
85 |
print("退出。")
|
|
|
37 |
for cookies, cb, hist, msg in 解析一个Python项目(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port):
|
38 |
print(cb)
|
39 |
|
40 |
+
def test_解析一个Cpp项目():
|
41 |
+
from crazy_functions.解析项目源代码 import 解析一个C项目
|
42 |
+
txt = "crazy_functions/test_project/cpp/cppipc"
|
43 |
+
for cookies, cb, hist, msg in 解析一个C项目(txt, llm_kwargs, plugin_kwargs, chatbot, history, system_prompt, web_port):
|
44 |
+
print(cb)
|
45 |
+
|
46 |
def test_Latex英文润色():
|
47 |
from crazy_functions.Latex全文润色 import Latex英文润色
|
48 |
txt = "crazy_functions/test_project/latex/attention"
|
|
|
86 |
test_谷歌检索小助手()
|
87 |
test_总结word文档()
|
88 |
test_下载arxiv论文并翻译摘要()
|
89 |
+
test_解析一个Cpp项目()
|
90 |
|
91 |
input("程序完成,回车退出。")
|
92 |
print("退出。")
|