gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove error handling tests from "all" mode

Browse Source
This commit is contained in:
yangdx
2025-02-02 21:59:56 +08:00
parent c07b5522fe
commit 9f82ba4970
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test_lightrag_ollama_chat.py
View File

@@ -742,7 +742,7 @@ Configuration file (config.json):
nargs="+", nargs="+",
choices=list(get_test_cases().keys()) + ["all"], choices=list(get_test_cases().keys()) + ["all"],
default=["all"], default=["all"],
help="Test cases to run, options: %(choices)s. Use 'all' to run all tests", help="Test cases to run, options: %(choices)s. Use 'all' to run all tests （except error tests)",
) )
return parser.parse_args() return parser.parse_args()
@@ -766,21 +766,18 @@ if __name__ == "__main__":
try: try:
if "all" in args.tests: if "all" in args.tests:
# Run all tests # Run all tests except error handling tests
if OutputControl.is_verbose(): if OutputControl.is_verbose():
print("\n【Chat API Tests】") print("\n【Chat API Tests】")
run_test(test_non_stream_chat, "Non-streaming Chat Test") run_test(test_non_stream_chat, "Non-streaming Chat Test")
run_test(test_stream_chat, "Streaming Chat Test") run_test(test_stream_chat, "Streaming Chat Test")
run_test(test_query_modes, "Chat Query Mode Test") run_test(test_query_modes, "Chat Query Mode Test")
run_test(test_error_handling, "Chat Error Handling Test")
run_test(test_stream_error_handling, "Chat Streaming Error Test")
if OutputControl.is_verbose(): if OutputControl.is_verbose():
print("\n【Generate API Tests】") print("\n【Generate API Tests】")
run_test(test_non_stream_generate, "Non-streaming Generate Test") run_test(test_non_stream_generate, "Non-streaming Generate Test")
run_test(test_stream_generate, "Streaming Generate Test") run_test(test_stream_generate, "Streaming Generate Test")
run_test(test_generate_with_system, "Generate with System Prompt Test") run_test(test_generate_with_system, "Generate with System Prompt Test")
run_test(test_generate_error_handling, "Generate Error Handling Test")
run_test(test_generate_concurrent, "Generate Concurrent Test") run_test(test_generate_concurrent, "Generate Concurrent Test")
else: else:
# Run specified tests # Run specified tests