Fixed serialization of typing.Literal (#867)

Fixes #852

Author: maciejdudko

temporalio/converter.py

@@ -23,6 +23,7 @@
     ClassVar,
     Dict,
     List,
+    Literal,
     Mapping,
     NewType,
     Optional,
@@ -39,7 +40,7 @@
 import google.protobuf.json_format
 import google.protobuf.message
 import google.protobuf.symbol_database
-from typing_extensions import Literal
+import typing_extensions
 
 import temporalio.api.common.v1
 import temporalio.api.enums.v1
@@ -1424,7 +1425,7 @@ def value_to_type(
     type_args: Tuple = getattr(hint, "__args__", ())
 
     # Literal
-    if origin is Literal:
+    if origin is Literal or origin is typing_extensions.Literal:
         if value not in type_args:
             raise TypeError(f"Value {value} not in literal values {type_args}")
         return value

tests/test_converter.py

@@ -16,6 +16,7 @@
     Dict,
     Iterable,
     List,
+    Literal,
     Mapping,
     MutableMapping,
     NewType,
@@ -31,12 +32,12 @@
 
 import pydantic
 import pytest
-from typing_extensions import Literal, TypedDict
+import typing_extensions
+from typing_extensions import TypedDict
 
 import temporalio.api.common.v1
 import temporalio.common
 from temporalio.api.common.v1 import Payload, Payloads
-from temporalio.api.common.v1 import Payload as AnotherNameForPayload
 from temporalio.api.failure.v1 import Failure
 from temporalio.common import RawValue
 from temporalio.converter import (
@@ -311,6 +312,9 @@ def fail(hint: Any, value: Any) -> None:
     ok(Literal["foo"], "foo")
     ok(Literal["foo", False], False)
     fail(Literal["foo", "bar"], "baz")
+    ok(typing_extensions.Literal["foo"], "foo")
+    ok(typing_extensions.Literal["foo", False], False)
+    fail(typing_extensions.Literal["foo", "bar"], "baz")
 
     # Dataclass
     ok(MyDataClass, MyDataClass("foo", 5, SerializableEnum.FOO))