fix: treat redirects as reachable in connection check

- Enable follow_redirects in httpx client for HEAD requests
- Accept 2xx and 3xx status codes as successful connection
- Add test for redirect handling in check_connection

Generated with Ripperdoc

Co-Authored-By: Ripperdoc

Author: jstzwj

src/olah/server.py

@@ -67,13 +67,13 @@
 
 async def check_connection(url: str) -> bool:
     try:
-        async with httpx.AsyncClient() as client:
+        async with httpx.AsyncClient(follow_redirects=True) as client:
             response = await client.request(
                 method="HEAD",
                 url=url,
                 timeout=10,
             )
-        return response.status_code == 200
+        return 200 <= response.status_code < 400
     except httpx.TimeoutException:
         return False
 

tests/test_server_layers.py

@@ -418,3 +418,39 @@ async def fake_ensure_repo_visibility(app, repo, authorization):
     )
 
     assert response.status_code == 401
+
+
+@pytest.mark.asyncio
+async def test_check_connection_treats_redirects_as_reachable(monkeypatch):
+    import importlib
+
+    server_module = importlib.import_module("olah.server")
+    captured = {}
+
+    class FakeClient:
+        def __init__(self, *args, **kwargs):
+            captured["follow_redirects"] = kwargs.get("follow_redirects")
+
+        async def __aenter__(self):
+            return self
+
+        async def __aexit__(self, exc_type, exc, tb):
+            return False
+
+        async def request(self, method, url, timeout):
+            captured["request"] = (method, url, timeout)
+            return SimpleNamespace(status_code=307)
+
+    monkeypatch.setattr(server_module.httpx, "AsyncClient", FakeClient)
+
+    ok = await server_module.check_connection(
+        "https://huggingface.co/datasets/Salesforce/wikitext/resolve/main/.gitattributes"
+    )
+
+    assert ok is True
+    assert captured["follow_redirects"] is True
+    assert captured["request"] == (
+        "HEAD",
+        "https://huggingface.co/datasets/Salesforce/wikitext/resolve/main/.gitattributes",
+        10,
+    )