Add missing setBody import #726 (#730)

Foso · Nov 9, 2024 · ae5a523 · ae5a523
2 parents 194fbe3 + 3c0d8a5
commit ae5a523
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/ktorfit-ksp/src/main/kotlin/de/jensklingenberg/ktorfit/model/FunctionData.kt b/ktorfit-ksp/src/main/kotlin/de/jensklingenberg/ktorfit/model/FunctionData.kt
@@ -207,7 +207,12 @@ fun KSFunctionDeclaration.toFunctionData(
                 addImport("io.ktor.util.AttributeKey")
             }
 
-            if (it is Body || it is ParameterAnnotation.PartMap || it is ParameterAnnotation.Part) {
+            if (it is Body ||
+                it is ParameterAnnotation.PartMap ||
+                it is ParameterAnnotation.Part ||
+                it is FieldMap ||
+                it is Field
+            ) {
                 addImport("io.ktor.client.request.setBody")
             }