openai - move client attribute definition

pgrayy · pgrayy · commit 8b61d5b64c94 · 2025-05-21T14:10:30.000Z
diff --git a/src/strands/models/openai.py b/src/strands/models/openai.py
@@ -4,7 +4,7 @@
 """
 
 import logging
-from typing import Any, Iterable, Optional, TypedDict, cast
+from typing import Any, Iterable, Optional, Protocol, TypedDict, cast
 
 import openai
 from typing_extensions import Unpack, override
@@ -14,9 +14,17 @@
 logger = logging.getLogger(__name__)
 
 
+class Client(Protocol):
+    """Protocol defining the OpenAI-compatible interface for the underlying provider client."""
+
+    chat: Any
+
+
 class OpenAIModel(SAOpenAIModel):
     """OpenAI model provider implementation."""
 
+    client: Client
+
     class OpenAIConfig(TypedDict, total=False):
         """Configuration options for OpenAI models.
 
diff --git a/src/strands/types/models/openai.py b/src/strands/types/models/openai.py
@@ -10,7 +10,7 @@
 import json
 import logging
 import mimetypes
-from typing import Any, Optional, Protocol
+from typing import Any, Optional
 
 from typing_extensions import override
 
@@ -22,19 +22,12 @@
 logger = logging.getLogger(__name__)
 
 
-class Client(Protocol):
-    """Protocol defining the OpenAI-compatible interface for the underlying provider client."""
-
-    chat: Any
-
-
 class OpenAIModel(Model, abc.ABC):
     """Base OpenAI model provider implementation.
 
     Implements shared logic for formatting requests and responses to and from the OpenAI specification.
     """
 
-    client: Client
     config: dict[str, Any]
 
     @staticmethod
