yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-11-04 03:39:35 +00:00
Code Issues 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
No known key found for this signature in database
1 changed files with 3 additions and 10 deletions
Show Stats 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", "host": "localhost",
"port": 9621, "port": 9621,
"model": "lightrag:latest", "model": "lightrag:latest",
"timeout": 120, "timeout": 300,
"max_retries": 3, "max_retries": 1,
"retry_delay": 1, "retry_delay": 1,
}, },
"test_cases": { "test_cases": {
@ -527,14 +527,7 @@ def test_non_stream_generate() -> None:
response_json = response.json() response_json = response.json()
# Print response content # Print response content
print_json_response( print(json.dumps(response_json, ensure_ascii=False, indent=2))
{
"model": response_json["model"],
"response": response_json["response"],
"done": response_json["done"],
},
"Response content",
)
def test_stream_generate() -> None: def test_stream_generate() -> None: