fixed some code

Author: Code-Hex

src/index.ts

@@ -8,11 +8,14 @@ import {
 } from "@graphql-codegen/plugin-helpers";
 import { GraphQLSchema } from "graphql";
 
-export const plugin: PluginFunction<ValidationSchemaPluginConfig> = (
+export const plugin: PluginFunction<
+  ValidationSchemaPluginConfig,
+  Types.ComplexPluginOutput
+> = (
   schema: GraphQLSchema,
   _documents: Types.DocumentFile[],
   config: ValidationSchemaPluginConfig
-): Types.PluginOutput => {
+): Types.ComplexPluginOutput => {
   const { schema: _schema, ast } = transformSchemaAST(schema, config);
   const { buildImports, ...visitor } = YupSchemaVisitor(_schema, config);
 

src/yup/index.ts

@@ -29,7 +29,7 @@ export const YupSchemaVisitor = (
       if (config.importFrom && importTypes.length > 0) {
         return [
           importYup,
-          `import { ${importTypes.join(", ")} } from "${config.importFrom}";`,
+          `import { ${importTypes.join(", ")} } from '${config.importFrom}'`,
         ];
       }
       return [importYup];
@@ -64,7 +64,7 @@ export const YupSchemaVisitor = (
           .withContent(
             `yup.mixed().oneOf([${node.values
               ?.map(
-                (enumOption) => `'${tsVisitor.convertName(enumOption.name)}'`
+                (enumOption) => `'${enumOption.name.value}'`
               )
               .join(", ")}])`
           ).string;