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

Update timeout and max_retries for unit tests

Browse Source
This commit is contained in:
yangdx
2025-02-06 19:56:18 +08:00
committed by ultrageopro
parent f72e4e6830
commit cec8da7f91
1 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
test_lightrag_ollama_chat.py
View File

@@ -23,8 +23,8 @@ DEFAULT_CONFIG = {
"host": "localhost",
"port": 9621,
"model": "lightrag:latest",
"timeout": 120,
"max_retries": 3,
"timeout": 300,
"max_retries": 1,
"retry_delay": 1,
},
"test_cases": {
@@ -527,14 +527,7 @@ def test_non_stream_generate() -> None:
response_json = response.json()
# Print response content
print_json_response(
{
"model": response_json["model"],
"response": response_json["response"],
"done": response_json["done"],
},
"Response content",
)
print(json.dumps(response_json, ensure_ascii=False, indent=2))
def test_stream_generate() -> None: