Как установить тайм-аут для конкретного запроса в gremlin_python для AWS Neptune?

Я использую библиотеку gremlin_python для выполнения запроса к базе данных AWS Neptune, и у меня возникают таймауты, несмотря на установку определенного порога таймаута с помощью параметра AssessmentTimeout. Вот конкретный запрос и сообщение об ошибке, которое я получил:

count_companies_with_representatives = (
    self.reader_g.V()
    .with_("evaluationTimeout", MAX_QUERY_TIMEOUT)
    .hasLabel("company")
    .where(__.bothE("director").count().is_(P.gt(0)))
    .count()
    .next()
)

Сообщение об ошибке:

{"detailedMessage":"A timeout occurred within the script [RequestMessage{, requestId=xxxxx, op='bytecode', processor='traversal', args = {gremlin=[[], [V(), with(evaluationTimeout, 1000000), hasLabel(company), where([[], [bothE(director), count(), is(gt(0))]]), count()]], aliases = {g=g}}}]","code":"TimeLimitExceededException","requestId":"xxxxx","message":"A timeout occurred within the script [RequestMessage{, requestId=xxxxx, op='bytecode', processor='traversal', args = {gremlin=[[], [V(), with(evaluationTimeout, 1000000), hasLabel(company), where([[], [bothE(director), count(), is(gt(0))]]), count()]], aliases = {g=g}}}]"}

Несмотря на то, что для параметра AssessmentTimeout установлено значение MAX_QUERY_TIMEOUT (в данном случае 1000000 мс), время запроса истекает преждевременно. Мне нужно убедиться, что каждый запрос соблюдает этот установленный тайм-аут, чтобы эффективно управлять производительностью и избежать преждевременного завершения.

Я попытался установить тайм-аут непосредственно в запросе, используя параметр AssessmentTimeout, ожидая, что это позволит запросу выполняться до MAX_QUERY_TIMEOUT миллисекунд до истечения времени ожидания. Однако ошибка указывает на то, что тайм-аут наступает задолго до достижения этого предела. Я ожидал, что настройка тайм-аута предотвратит досрочное завершение запроса и предоставит ему достаточно времени для завершения в нормальных рабочих условиях. Мне нужен совет о том, как правильно применять тайм-ауты к отдельным запросам в этой среде, или чтобы понять, существует ли лучший подход к управлению временем выполнения запросов в Neptune.