Merge pull request #1122 from tisnik/lcore-886-fixed-pyright-issues-in-endpoints

LCORE-886: fixed Pyright issues in endpoints

Author: tisnik

tests/integration/endpoints/test_config_integration.py

@@ -1,6 +1,8 @@
 """Integration tests for the /config endpoint."""
 
+from typing import cast
 import pytest
+
 from fastapi import HTTPException, Request, status
 
 from app.endpoints.config import config_endpoint_handler
@@ -81,6 +83,6 @@ async def test_config_endpoint_fails_without_configuration(
     # Verify error details
     assert exc_info.value.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
     assert isinstance(exc_info.value.detail, dict)
-    assert (
-        "configuration is not loaded" in exc_info.value.detail["response"].lower()
-    )  # type: ignore
+    assert "response" in exc_info.value.detail
+    detail = cast(dict[str, str], exc_info.value.detail)
+    assert "configuration is not loaded" in detail["response"].lower()

tests/integration/endpoints/test_rlsapi_v1_integration.py

@@ -9,7 +9,7 @@
 # pylint: disable=protected-access
 # pylint: disable=unused-argument
 
-from typing import Any
+from typing import Any, cast
 
 import pytest
 from fastapi import HTTPException, status
@@ -139,6 +139,7 @@ async def test_rlsapi_v1_infer_minimal_request(
 
     assert isinstance(response, RlsapiV1InferResponse)
     assert response.data.text == "Use the `ls` command to list files in a directory."
+    assert response.data.request_id is not None
     assert check_suid(response.data.request_id)
 
 
@@ -221,7 +222,9 @@ async def test_rlsapi_v1_infer_generates_unique_request_ids(
     request_ids = {r.data.request_id for r in responses}
 
     assert len(request_ids) == 3
-    assert all(check_suid(rid) for rid in request_ids)
+    for rid in request_ids:
+        assert rid is not None
+    assert all(check_suid(rid) for rid in request_ids if rid is not None)
 
 
 # ==========================================
@@ -262,7 +265,9 @@ async def test_rlsapi_v1_infer_connection_error_returns_503(
 
     assert exc_info.value.status_code == status.HTTP_503_SERVICE_UNAVAILABLE
     assert isinstance(exc_info.value.detail, dict)
-    assert "Llama Stack" in exc_info.value.detail["response"]
+    assert "response" in exc_info.value.detail
+    detail = cast(dict[str, str], exc_info.value.detail)
+    assert "Llama Stack" in detail["response"]
 
 
 @pytest.mark.asyncio