Add enums to from part (#425)

MatthewZMSU · Matvey Zvantsov · web-flow · commit bfe6e4c9860f · 2026-04-14T12:15:23.000+02:00
Co-authored-by: Matvey Zvantsov &lt;zvancov.mu@ispras.ru&gt;
diff --git a/ariadne_codegen/client_generators/custom_arguments.py b/ariadne_codegen/client_generators/custom_arguments.py
@@ -228,7 +228,7 @@ def _parse_graphql_type_name(
                 generate_import_from(names=[name], from_="input_types", level=1)
             )
         elif isinstance(type_, GraphQLEnumType):
-            self._add_import(generate_import_from(names=[name], level=1))
+            self._add_import(generate_import_from(names=[name], from_="enums", level=1))
         elif isinstance(type_, GraphQLScalarType):
             if name not in self.custom_scalars:
                 name = INPUT_SCALARS_MAP.get(name, ANY)

Original file line number	Diff line number	Diff line change
`@@ -228,7 +228,7 @@ def _parse_graphql_type_name(`
`228`	`228`	`generate_import_from(names=[name], from_="input_types", level=1)`
`229`	`229`	`)`
`230`	`230`	`elif isinstance(type_, GraphQLEnumType):`
`231`		`- self._add_import(generate_import_from(names=[name], level=1))`
	`231`	`+ self._add_import(generate_import_from(names=[name], from_="enums", level=1))`
`232`	`232`	`elif isinstance(type_, GraphQLScalarType):`
`233`	`233`	`if name not in self.custom_scalars:`
`234`	`234`	`name = INPUT_SCALARS_MAP.get(name, ANY)`