fix hard code for Query

Author: Yamashou

clientgen/source.go

@@ -161,6 +161,11 @@ func (s *Source) Query() (*Query, error) {
 		return nil, xerrors.Errorf("generate failed for query struct type : %w", err)
 	}
 
+	s.sourceGenerator.cfg.Models.Add(
+		s.schema.Query.Name,
+		fmt.Sprintf("%s.%s", s.sourceGenerator.client.Pkg(), templates.ToGo(s.schema.Query.Name)),
+	)
+
 	return &Query{
 		Name: s.schema.Query.Name,
 		Type: fields.StructType(),
@@ -178,6 +183,11 @@ func (s *Source) Mutation() (*Mutation, error) {
 		return nil, xerrors.Errorf("generate failed for mutation struct type : %w", err)
 	}
 
+	s.sourceGenerator.cfg.Models.Add(
+		s.schema.Mutation.Name,
+		fmt.Sprintf("%s.%s", s.sourceGenerator.client.Pkg(), templates.ToGo(s.schema.Mutation.Name)),
+	)
+
 	return &Mutation{
 		Name: s.schema.Mutation.Name,
 		Type: fields.StructType(),

clientgen/source_generator.go

@@ -95,7 +95,7 @@ func (r *SourceGenerator) NewResponseFieldsByDefinition(definition *ast.Definiti
 
 		var typ types.Type
 		if field.Type.Name() == "Query" || field.Type.Name() == "Mutation" {
-			baseType, err := r.binder.FindType(r.client.Pkg().Path(), "Query")
+			baseType, err := r.binder.FindType(r.client.Pkg().Path(), field.Type.Name())
 			if err != nil {
 				return nil, xerrors.Errorf("not found type: %w", err)
 			}