chore(internal): use TypeAlias marker for type assignments (#621)

anthropics · Aug 5, 2024 · a4bff9c · a4bff9c
1 parent 0a3f3fa
commit a4bff9c
Show file tree

Hide file tree

Showing 9 changed files with 18 additions and 18 deletions.
diff --git a/src/anthropic/types/content_block.py b/src/anthropic/types/content_block.py
@@ -1,12 +1,12 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 from typing import Union
-from typing_extensions import Annotated
+from typing_extensions import Annotated, TypeAlias
 
 from .._utils import PropertyInfo
 from .text_block import TextBlock
 from .tool_use_block import ToolUseBlock
 
 __all__ = ["ContentBlock"]
 
-ContentBlock = Annotated[Union[TextBlock, ToolUseBlock], PropertyInfo(discriminator="type")]
+ContentBlock: TypeAlias = Annotated[Union[TextBlock, ToolUseBlock], PropertyInfo(discriminator="type")]
diff --git a/src/anthropic/types/message_create_params.py b/src/anthropic/types/message_create_params.py
@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from typing import List, Union, Iterable, Optional
-from typing_extensions import Literal, Required, TypedDict
+from typing_extensions import Literal, Required, TypeAlias, TypedDict
 
 from .tool_param import ToolParam
 from .model_param import ModelParam
@@ -287,7 +287,7 @@ class ToolChoiceToolChoiceTool(TypedDict, total=False):
     type: Required[Literal["tool"]]
 
 
-ToolChoice = Union[ToolChoiceToolChoiceAuto, ToolChoiceToolChoiceAny, ToolChoiceToolChoiceTool]
+ToolChoice: TypeAlias = Union[ToolChoiceToolChoiceAuto, ToolChoiceToolChoiceAny, ToolChoiceToolChoiceTool]
 
 
 class MessageCreateParamsNonStreaming(MessageCreateParamsBase):

diff --git a/src/anthropic/types/model.py b/src/anthropic/types/model.py
@@ -1,11 +1,11 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 from typing import Union
-from typing_extensions import Literal
+from typing_extensions import Literal, TypeAlias
 
 __all__ = ["Model"]
 
-Model = Union[
+Model: TypeAlias = Union[
     str,
     Literal[
         "claude-3-5-sonnet-20240620",

diff --git a/src/anthropic/types/model_param.py b/src/anthropic/types/model_param.py
@@ -3,11 +3,11 @@
 from __future__ import annotations
 
 from typing import Union
-from typing_extensions import Literal
+from typing_extensions import Literal, TypeAlias
 
 __all__ = ["ModelParam"]
 
-ModelParam = Union[
+ModelParam: TypeAlias = Union[
     str,
     Literal[
         "claude-3-5-sonnet-20240620",

diff --git a/src/anthropic/types/raw_content_block_delta_event.py b/src/anthropic/types/raw_content_block_delta_event.py
@@ -1,7 +1,7 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 from typing import Union
-from typing_extensions import Literal, Annotated
+from typing_extensions import Literal, Annotated, TypeAlias
 
 from .._utils import PropertyInfo
 from .._models import BaseModel
@@ -10,7 +10,7 @@
 
 __all__ = ["RawContentBlockDeltaEvent", "Delta"]
 
-Delta = Annotated[Union[TextDelta, InputJSONDelta], PropertyInfo(discriminator="type")]
+Delta: TypeAlias = Annotated[Union[TextDelta, InputJSONDelta], PropertyInfo(discriminator="type")]
 
 
 class RawContentBlockDeltaEvent(BaseModel):

diff --git a/src/anthropic/types/raw_content_block_start_event.py b/src/anthropic/types/raw_content_block_start_event.py
@@ -1,7 +1,7 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 from typing import Union
-from typing_extensions import Literal, Annotated
+from typing_extensions import Literal, Annotated, TypeAlias
 
 from .._utils import PropertyInfo
 from .._models import BaseModel
@@ -10,7 +10,7 @@
 
 __all__ = ["RawContentBlockStartEvent", "ContentBlock"]
 
-ContentBlock = Annotated[Union[TextBlock, ToolUseBlock], PropertyInfo(discriminator="type")]
+ContentBlock: TypeAlias = Annotated[Union[TextBlock, ToolUseBlock], PropertyInfo(discriminator="type")]
 
 
 class RawContentBlockStartEvent(BaseModel):

diff --git a/src/anthropic/types/raw_message_stream_event.py b/src/anthropic/types/raw_message_stream_event.py
@@ -1,7 +1,7 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 from typing import Union
-from typing_extensions import Annotated
+from typing_extensions import Annotated, TypeAlias
 
 from .._utils import PropertyInfo
 from .raw_message_stop_event import RawMessageStopEvent
@@ -13,7 +13,7 @@
 
 __all__ = ["RawMessageStreamEvent"]
 
-RawMessageStreamEvent = Annotated[
+RawMessageStreamEvent: TypeAlias = Annotated[
     Union[
         RawMessageStartEvent,
         RawMessageDeltaEvent,

diff --git a/src/anthropic/types/tool_param.py b/src/anthropic/types/tool_param.py
@@ -3,7 +3,7 @@
 from __future__ import annotations
 
 from typing import Dict, Union, Optional
-from typing_extensions import Literal, Required, TypedDict
+from typing_extensions import Literal, Required, TypeAlias, TypedDict
 
 __all__ = ["ToolParam", "InputSchema"]
 
@@ -14,7 +14,7 @@ class InputSchemaTyped(TypedDict, total=False):
     properties: Optional[object]
 
 
-InputSchema = Union[InputSchemaTyped, Dict[str, object]]
+InputSchema: TypeAlias = Union[InputSchemaTyped, Dict[str, object]]
 
 
 class ToolParam(TypedDict, total=False):

diff --git a/src/anthropic/types/tool_result_block_param.py b/src/anthropic/types/tool_result_block_param.py
@@ -3,14 +3,14 @@
 from __future__ import annotations
 
 from typing import Union, Iterable
-from typing_extensions import Literal, Required, TypedDict
+from typing_extensions import Literal, Required, TypeAlias, TypedDict
 
 from .text_block_param import TextBlockParam
 from .image_block_param import ImageBlockParam
 
 __all__ = ["ToolResultBlockParam", "Content"]
 
-Content = Union[TextBlockParam, ImageBlockParam]
+Content: TypeAlias = Union[TextBlockParam, ImageBlockParam]
 
 
 class ToolResultBlockParam(TypedDict, total=False):