yangdx commited on
Commit
2dd7565
·
1 Parent(s): beee286

Update timeout and max_retries for unit tests

Browse files
Files changed (1) hide show
  1. test_lightrag_ollama_chat.py +3 -10
test_lightrag_ollama_chat.py CHANGED
@@ -23,8 +23,8 @@ DEFAULT_CONFIG = {
23
  "host": "localhost",
24
  "port": 9621,
25
  "model": "lightrag:latest",
26
- "timeout": 120,
27
- "max_retries": 3,
28
  "retry_delay": 1,
29
  },
30
  "test_cases": {
@@ -527,14 +527,7 @@ def test_non_stream_generate() -> None:
527
  response_json = response.json()
528
 
529
  # Print response content
530
- print_json_response(
531
- {
532
- "model": response_json["model"],
533
- "response": response_json["response"],
534
- "done": response_json["done"],
535
- },
536
- "Response content",
537
- )
538
 
539
 
540
  def test_stream_generate() -> None:
 
23
  "host": "localhost",
24
  "port": 9621,
25
  "model": "lightrag:latest",
26
+ "timeout": 300,
27
+ "max_retries": 1,
28
  "retry_delay": 1,
29
  },
30
  "test_cases": {
 
527
  response_json = response.json()
528
 
529
  # Print response content
530
+ print(json.dumps(response_json, ensure_ascii=False, indent=2))
 
 
 
 
 
 
 
531
 
532
 
533
  def test_stream_generate() -> None: