diff --git a/literalai/api/gql.py b/literalai/api/gql.py index ea0b677..f65bd08 100644 --- a/literalai/api/gql.py +++ b/literalai/api/gql.py @@ -766,7 +766,7 @@ CREATE_GENERATION = """ mutation CreateGeneration($generation: GenerationPayloadInput!) { createGeneration(generation: $generation) { - id, + id type } }