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
parent 506e39e14e
commit 6ca7487cac
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", "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: