diff --git a/tests/e2e/models.py b/tests/e2e/models.py
index 0dd8e554c8f..9cc28b38d27 100644
--- a/tests/e2e/models.py
+++ b/tests/e2e/models.py
@@ -8,7 +8,7 @@ from __future__ import annotations
 
 from collections.abc import Sequence
 from datetime import datetime
-from typing import Final, Literal
+from typing import Final, Literal, TypeAlias
 
 from e2e_http import PartialBody
 from pydantic import (
@@ -204,7 +204,7 @@ class FileObject(BaseModel):
 class TextContentPart(BaseModel):
     type: str = "text"
     text: str
-    cache_control: "CacheControl | None" = None
+    cache_control: CacheControl | None = None
 
 
 class ImageContentPart(BaseModel):
@@ -304,7 +304,7 @@ class ChatToolResultTurn(BaseModel):
     content: str
 
 
-type ChatTurn = ChatMessage | ChatAssistantTurn | ChatToolResultTurn
+ChatTurn: TypeAlias = ChatMessage | ChatAssistantTurn | ChatToolResultTurn
 
 
 class HostedWebSearchTool(BaseModel):
@@ -532,7 +532,7 @@ class AnthropicCustomTool(BaseModel):
     input_schema: ToolInputSchema
 
 
-type AnthropicTool = AnthropicToolSearchTool | AnthropicWebSearchTool | AnthropicCustomTool
+AnthropicTool: TypeAlias = AnthropicToolSearchTool | AnthropicWebSearchTool | AnthropicCustomTool
 
 
 class AnthropicContentBlock(BaseModel):
@@ -570,7 +570,7 @@ class AnthropicToolResultTurn(BaseModel):
     content: list[AnthropicToolResultBlock]
 
 
-type AnthropicMessage = ChatMessage | AnthropicAssistantTurn | AnthropicToolResultTurn
+AnthropicMessage: TypeAlias = ChatMessage | AnthropicAssistantTurn | AnthropicToolResultTurn
 
 
 class AnthropicToolChoice(BaseModel):