yangdx
commited on
Commit
·
2dd7565
1
Parent(s):
beee286
Update timeout and max_retries for unit tests
Browse files- 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":
|
27 |
-
"max_retries":
|
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 |
-
|
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:
|