Fix docker issues, TTL cleanup

[abrookins]

This PR fixes an issue with local testing and Docker where redis containers would fail to come up. Previously, each pytest xdist worker would attempt to connect to its own redis instance. If the instance failed to come up in time, the configuration defaulted to localhost: 6379. As long as a redis instance is running locally, the problem would have been masked, but this causes the tests to fail when a local redis instance is unavailable.

We attempt to resolve the issue by waiting longer for individual redis containers to become available. In the previous configuration, it was also possible for multiple xdist workers to start redis instances mapped to the same port (causing conflicts). We resolve this by giving each xdist worker a random, non-conflicting host port for their redis instance.

[Copilot]

Pull Request Overview

This PR addresses Docker issues encountered during local testing by updating Redis container port mapping and improves TTL handling in Redis-based checkpoints. Key changes include:

• Updating the Docker Compose Redis service configuration to use an ephemeral host port.
• Refactoring TTL logic in checkpoint functions to delegate TTL application to a helper function.
• Reordering and consolidating import statements in tests and checkpoint modules.

Reviewed Changes

Copilot reviewed 7 out of 7 changed files in this pull request and generated no comments.

Show a summary per file

File	Description
tests/test_cluster_mode.py	Reorganized import order for clarity.
tests/test_async_cluster_mode.py	Adjusted import order to correctly locate AsyncRedisStore.
tests/docker-compose.yml	Updated Redis service port mapping to use a published ephemeral port.
tests/conftest.py	Added connection wait logic to ensure the Docker container becomes ready.
langgraph/checkpoint/redis/types.py	Updated import order for RedisCluster to match expected usage.
langgraph/checkpoint/redis/aio.py	Refactored TTL handling in async functions to use a helper function.
langgraph/checkpoint/redis/init.py	Minor import reordering for consistency.

Comments suppressed due to low confidence (3)

langgraph/checkpoint/redis/aio.py:598

• Ensure that _apply_ttl_to_keys is implemented to properly handle a null second parameter and correctly apply TTL when no blob keys are provided.

await self._apply_ttl_to_keys(checkpoint_key, [key for key, _ in blobs] if blobs else None,)

langgraph/checkpoint/redis/aio.py:768

• Verify that _apply_ttl_to_keys handles cases when only a single key is present to avoid inadvertently skipping TTL assignment for created keys.

await self._apply_ttl_to_keys(created_keys[0], created_keys[1:] if len(created_keys) > 1 else None,)

tests/docker-compose.yml:7

• Include a comment explaining that 'published: 0' instructs Docker to dynamically assign an ephemeral port, which supports non-conflicting port allocation during parallel testing.

published: 0