precommit

.gitignore

@@ -57,4 +57,8 @@ packages/**/generated
 
 **/.idea/*
 
-.python-version
+.python-version
+
+# This file is too large to be included in the repository without LFS
+seed/python-sdk/trace/snippet-templates.json
+seed/python-sdk/**/output.prof

generators/python/sdk/Dockerfile

@@ -30,14 +30,14 @@ COPY ./src ./src
 
 RUN poetry install
 
-# ENTRYPOINT ["python", "-m", "src.fern_python.generators.sdk.cli"]
+ENTRYPOINT ["python", "-m", "src.fern_python.generators.sdk.cli"]
 
 ## NOTE: Uncomment the below to generate a flame graph for the python generator. 
 ##       To visualize the flamegraph you can run: 
 ##         - poetry add snakeviz
 ##         - poetry run snakeviz output.prof
-RUN mkdir -p /fern/output
+# RUN mkdir -p /fern/output
 ## For a time-limited run, uncomment the below line and comment the next line.
-ENTRYPOINT ["timeout", "-s", "INT", "10m", "python", "-m", "cProfile", "-o", "/fern/output/output.prof", "/src/fern_python/generators/sdk/cli.py"]
+# ENTRYPOINT ["timeout", "-s", "INT", "5m", "python", "-m", "cProfile", "-o", "/fern/output/output.prof", "/src/fern_python/generators/sdk/cli.py"]
 ## For a regularly profiled run, uncomment the below line and comment the previous line.
 # ENTRYPOINT ["python", "-m", "cProfile", "-o", "/fern/output/output.prof", "/src/fern_python/generators/sdk/cli.py"]

generators/python/sdk/VERSION

@@ -1 +1 @@
-3.6.0
+3.7.0

generators/python/src/fern_python/generators/pydantic_model/fern_aware_pydantic_model.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from types import TracebackType
-from typing import List, Optional, Sequence, Set, Tuple, Type
+from typing import List, Optional, Sequence, Tuple, Type
 
 import fern.ir.resources as ir_types
 
@@ -181,7 +181,9 @@ def _must_import_after_current_declaration(self, type_name: ir_types.DeclaredTyp
         type_id_to_reference = self._type_id_for_forward_ref()
         is_circular_reference = False
         if type_id_to_reference is not None:
-            is_circular_reference = self._context.does_type_reference_other_type(type_id=type_name.type_id, other_type_id=type_id_to_reference)
+            is_circular_reference = self._context.does_type_reference_other_type(
+                type_id=type_name.type_id, other_type_id=type_id_to_reference
+            )
 
         if is_circular_reference:
             self._model_contains_forward_refs = True

generators/python/src/fern_python/generators/pydantic_model/type_declaration_handler/discriminated_union/discriminated_union_with_utils_generator.py

@@ -278,7 +278,9 @@ def generate(self) -> None:
                     forward_refed_types = [
                         referenced_type_id
                         for referenced_type_id in referenced_type_ids
-                        if self._context.does_type_reference_other_type(type_id=referenced_type_id, other_type_id=self._name.type_id)
+                        if self._context.does_type_reference_other_type(
+                            type_id=referenced_type_id, other_type_id=self._name.type_id
+                        )
                     ]
 
                     if len(forward_refed_types) > 0:

generators/python/src/fern_python/generators/pydantic_model/type_declaration_handler/pydantic_models/pydantic_model_simple_discriminated_union_generator.py

@@ -159,7 +159,7 @@ def _finish(self, type_alias_declaration: AST.TypeAliasDeclaration) -> None:
             # There are some types that will not cause their dependent types to rebuild
             # for example other unions, and so to make sure those types are rebuilt
             # we import them all here.
-            for type_id in (self._context.maybe_get_type_ids_for_type_reference(referenced_type) or []):
+            for type_id in self._context.maybe_get_type_ids_for_type_reference(referenced_type) or []:
                 type_alias_declaration.add_ghost_reference(
                     self._context.get_class_reference_for_type_id(
                         type_id,
@@ -184,9 +184,9 @@ def _update_forward_refs_for_single_property_member(
     ) -> None:
         referenced_type_ids: Set[ir_types.TypeId] = single_union_type.shape.visit(
             same_properties_as_object=lambda _: set(),
-            single_property=lambda single_property: set(self._context.get_referenced_types_of_type_reference(
-                single_property.type
-            ) or []),
+            single_property=lambda single_property: set(
+                self._context.get_referenced_types_of_type_reference(single_property.type) or []
+            ),
             no_properties=lambda: set(),
         )