Fix multipart import (#2729)

* fix import for multipart * changelog
Azure · Jul 30, 2024 · fbe8a5a · fbe8a5a
1 parent df8fef5
commit fbe8a5a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/.chronus/changes/fix-multipart-import-2024-6-29-16-22-46.md b/.chronus/changes/fix-multipart-import-2024-6-29-16-22-46.md
@@ -0,0 +1,7 @@
+---
+changeKind: fix
+packages:
+  - "@azure-tools/typespec-python"
+---
+
+Fix import for multipart
diff --git a/packages/typespec-python/generator/pygen/codegen/models/operation.py b/packages/typespec-python/generator/pygen/codegen/models/operation.py
@@ -495,7 +495,9 @@ def imports(self, async_mode: bool, **kwargs: Any) -> FileImport:
         relative_path = "..." if async_mode else ".."
         if self.code_model.options["models_mode"] == "dpg":
             if self.parameters.has_body:
-                if not self.has_form_data_body:
+                if self.has_form_data_body:
+                    file_import.add_submodule_import(relative_path, "_model_base", ImportType.LOCAL)
+                else:
                     file_import.add_submodule_import(
                         f"{relative_path}_model_base",
                         "SdkJSONEncoder",