如何在石墨烯mongo python中更新EmbeddedDocument字段

from graphene import ObjectType, String, Schema
from graphene_mongo import MongoengineObjectType
from mongoengine import Document, EmbeddedDocument, fields
from pymongo import MongoClient

class Address(EmbeddedDocument):
    street = fields.StringField()
    city = fields.StringField()
    state = fields.StringField()

class User(Document):
    name = fields.StringField()
    address = fields.EmbeddedDocumentField(Address)

class AddressType(ObjectType):
    street = String()
    city = String()
    state = String()

class UserType(MongoengineObjectType):
    class Meta:
        model = User

class Query(ObjectType):
    user = String(name=String())

    def resolve_user(self, info, name):
        user = User.objects(name=name).first()
        return user

class UpdateUser(ObjectType):
    update_user = String(name=String(), street=String(), city=String(), state=String())

    def resolve_update_user(self, info, name, street, city, state):
        user = User.objects(name=name).first()
        user.address.street = street
        user.address.city = city
        user.address.state = state
        user.save()
        return "User updated successfully"

class Mutation(ObjectType):
    update_user = UpdateUser.Field()

schema = Schema(query=Query, mutation=Mutation)

client = MongoClient('mongodb://localhost:27017/')
db = client['your_database_name']

result = schema.execute('''
    mutation {
        update_user(name: "John", street: "123 Main St", city: "New York", state: "NY")
    }
''', context_value={'db': db})

print(result.data['update_user'])