improvement: use AnyStr to keep intellisense for enums but allow forw… (

#3216)

generators/python/sdk/CHANGELOG.md

@@ -5,6 +5,22 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.12.4] - 2024-03-19
+
+- Improvement: Allow full forward compat with enums while keeping intellisense by unioning enum literals with `typing.AnyStr`.
+
+  Before:
+
+  ```python
+  Operand = typing.Union[typing.AnyStr, typing.Literal[">", "=", "less_than"]]
+  ```
+
+  After:
+
+  ```python
+  Operand = typing.Literal[">", "=", "less_than"]
+  ```
+
 ## [0.12.3] - 2024-03-18
 
 - Improvement: Allow bytes requests to take in iterators of bytes, mirroring the types allowed by HTTPX.

generators/python/sdk/VERSION

@@ -1 +1 @@
-0.12.3
+0.12.4

generators/python/src/fern_python/codegen/ast/nodes/type_hint/type_hint.py

@@ -75,6 +75,10 @@ def not_required(wrapped_type: TypeHint) -> TypeHint:
             type_parameters=[TypeParameter(wrapped_type)],
         )
 
+    @staticmethod
+    def any_str() -> TypeHint:
+        return TypeHint(type=get_reference_to_typing_import("AnyStr"))
+
     @staticmethod
     def optional(wrapped_type: TypeHint) -> TypeHint:
         return TypeHint(

generators/python/src/fern_python/generators/pydantic_model/type_declaration_handler/enum_generator.py

@@ -33,8 +33,11 @@ def generate(self) -> None:
         if self._use_str_enums:
             self._source_file.add_declaration(
                 AST.TypeAliasDeclaration(
-                    type_hint=AST.TypeHint.literal(
-                        AST.Expression(", ".join(map(lambda v: f'"{v.name.wire_value}"', self._enum.values)))
+                    type_hint=AST.TypeHint.union(
+                        AST.TypeHint.any_str(),
+                        AST.TypeHint.literal(
+                            AST.Expression(", ".join(map(lambda v: f'"{v.name.wire_value}"', self._enum.values)))
+                        ),
                     ),
                     name=self._name.name.pascal_case.safe_name,
                 ),