Generated Enum types are not valid as parameters

[asfaltboy]

Describe the bug

When my schema contains a reference to an Enum type, then the code generator creates an enum.Enum subclass. However, I can pass an instance of this Enum type into a parameter function, and mypy is happy. However, when executing the code, I see the error:

>   ???
E   TypeError: a str or edgedb.EnumValue(__tid__=59ec73b8-4261-11ed-a3eb-b1b44f04f696) is expected as a valid enum argument, got Color

edgedb/protocol/./codecs/./enum.pyx:27: TypeError

Changing the passed argument to the Enum value (e.g Color.RED.value) results in mypy complaining that str is not a valid type.

Full trace back - click to expand!

test.py:62: in insert_shirt
    return await executor.query_single(
../../Library/Caches/pypoetry/virtualenvs/test-edgedb-bug/lib/python3.10/site-packages/edgedb/abstract.py:238: in query_single
    return await self._query(QueryContext(
../../Library/Caches/pypoetry/virtualenvs/test-edgedb-bug/lib/python3.10/site-packages/edgedb/base_client.py:739: in _query
    return await con.raw_query(query_context)
../../Library/Caches/pypoetry/virtualenvs/test-edgedb-bug/lib/python3.10/site-packages/edgedb/base_client.py:239: in raw_query
    raise e
../../Library/Caches/pypoetry/virtualenvs/test-edgedb-bug/lib/python3.10/site-packages/edgedb/base_client.py:234: in raw_query
    return await self._protocol.query(**args)
edgedb/protocol/protocol.pyx:594: in query
    ???
edgedb/protocol/protocol.pyx:559: in execute
    ???
edgedb/protocol/protocol.pyx:379: in _execute
    ???
edgedb/protocol/protocol.pyx:1125: in edgedb.protocol.protocol.SansIOProtocol.encode_args
    ???
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _

>   ???
E   edgedb.errors.InvalidArgumentError: invalid input for query argument $color: <Color.RED: 'Red'> (a str or edgedb.EnumValue(__tid__=59ec73b8-4261-11ed-a3eb-b1b44f04f696) is expected as a valid enum argument, got Color)

edgedb/protocol/./codecs/./object.pyx:114: InvalidArgumentError

Reproduction

Given the schema:

scalar type Color extending enum<Red, Green, Blue>;

type Shirt {
  property color -> Color;
}

When we have a query:

select (
    insert Shirt {
        color := <Color>$color
    }
) { id, color };

Run edgedb-py, and execute:

insert_shirt(color=Color.RED)

Expected behavior

I expect to be able to pass the Enum value directly as a parameter, as it's the type declared by the code generator.

Versions (please complete the following information):

• OS: macOS Ventura 13.2.1 (22D68)
• EdgeDB version: 2.12+a5dfd9f
• EdgeDB CLI version: 2.3.1+5d93f42
• edgedb-python version: 1.3.0
• Python version: 3.10.6

Additional context

Note: I raised another issue/question in discord regarding optional parameters: https://discord.com/channels/841451783728529451/849374935209607199/1081881229676265592