From 357fb0c7c33ddd25095f80cd69bd89a826600a38 Mon Sep 17 00:00:00 2001
From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
Date: Sat, 18 Jul 2026 19:17:20 +0000
Subject: [PATCH] fix(e2e): use renamed proxy client in load tests

Co-Authored-By: Ishaan Jaffer <155045088+ishaan-berri@users.noreply.github.com>
---
 tests/e2e/load/conftest.py    | 4 ++--
 tests/e2e/load/load_client.py | 6 +++---
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/tests/e2e/load/conftest.py b/tests/e2e/load/conftest.py
index e2d135092fb..09fbdd06f0f 100644
--- a/tests/e2e/load/conftest.py
+++ b/tests/e2e/load/conftest.py
@@ -5,7 +5,7 @@ from collections.abc import Iterator
 import pytest
 from requests import RequestException
 
-from e2e_gateway import Gateway
+from proxy_client import ProxyClient
 from e2e_http import NoBody, Success
 from load_client import LoadClient, build_client
 from load_constants import LOAD_MODEL
@@ -23,7 +23,7 @@ def client() -> LoadClient:
     return build_client()
 
 
-def _model_is_servable(gateway: Gateway, model_name: str) -> bool:
+def _model_is_servable(gateway: ProxyClient, model_name: str) -> bool:
     result = gateway.transport.get(
         "/v1/models",
         headers=gateway.transport.master,
diff --git a/tests/e2e/load/load_client.py b/tests/e2e/load/load_client.py
index df7c91fadf9..ab2ffb8040b 100644
--- a/tests/e2e/load/load_client.py
+++ b/tests/e2e/load/load_client.py
@@ -2,13 +2,13 @@ from __future__ import annotations
 
 from dataclasses import dataclass
 
-from e2e_gateway import Gateway, build_gateway
+from proxy_client import ProxyClient, build_proxy_client
 
 
 @dataclass(frozen=True, slots=True)
 class LoadClient:
-    gateway: Gateway
+    gateway: ProxyClient
 
 
 def build_client() -> LoadClient:
-    return LoadClient(gateway=build_gateway())
+    return LoadClient(gateway=build_proxy_client())