monolith.py

from graphene import ObjectType, Int, List, String, Field, NonNull
from graphene_federation import build_schema, extend, external


@extend(fields='id')
class FileNode(ObjectType):
    id = external(Int(required=True))


class MessageNode(ObjectType):
    id = Int(required=True)
    thread_id = Int(required=True)
    author_id = Int()
    files = List(NonNull(FileNode))
    body = String(required=True)


class Query(ObjectType):
    message = Field(lambda: MessageNode, id=Int())


schema = build_schema(Query)