compat: Add check for Graphene 2.x type argument

Author: Rami Chowdhury

graphene_pydantic/converters.py

@@ -20,6 +20,7 @@
     String,
     Union,
 )
+import graphene
 from graphene.types.base import BaseType
 from graphene.types.datetime import Date, DateTime, Time
 from pydantic import BaseModel
@@ -31,6 +32,8 @@
 
 from pydantic import fields
 
+GRAPHENE2 = graphene.VERSION[0] < 3
+
 SHAPE_SINGLETON = (fields.SHAPE_SINGLETON,)
 SHAPE_SEQUENTIAL = (
     fields.SHAPE_LIST,
@@ -89,7 +92,7 @@ def convert_pydantic_input_field(
     """
     declared_type = getattr(field, "type_", None)
     field_kwargs.setdefault(
-        "type_",
+        "type" if GRAPHENE2 else "type_",
         convert_pydantic_type(
             declared_type, field, registry, parent_type=parent_type, model=model
         ),
@@ -118,7 +121,7 @@ def convert_pydantic_field(
     """
     declared_type = getattr(field, "type_", None)
     field_kwargs.setdefault(
-        "type",
+        "type" if GRAPHENE2 else "type_",
         convert_pydantic_type(
             declared_type, field, registry, parent_type=parent_type, model=model
         ),