fix: update error message

bifrostlab · Apr 6, 2024 · 113cf1a · 113cf1a
1 parent d3d8b1e
commit 113cf1a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_llm.py b/tests/test_llm.py
@@ -67,7 +67,7 @@ async def test_review_resume__valid_url() -> None:
 
   for url in valid_urls:
     response = await llm.review_resume(MODEL, url, AI_SERVER_URL)
-    assert not response[0].startswith("Invalid URL")
+    assert not response[0].startswith("Error: Invalid URL")
 
 
 @pytest.mark.asyncio
@@ -76,7 +76,7 @@ async def test_review_resume__invalid_url() -> None:
 
   for url in invalid_urls:
     response = await llm.review_resume(MODEL, url, AI_SERVER_URL)
-    assert response[0].startswith("Invalid URL")
+    assert response[0].startswith("Error: Invalid URL")
 
 
 @pytest.mark.asyncio