diff --git a/features/steps/langgraph_platform_remote_graph_steps.py b/features/steps/langgraph_platform_remote_graph_steps.py index 1034ab035..b31962dc2 100644 --- a/features/steps/langgraph_platform_remote_graph_steps.py +++ b/features/steps/langgraph_platform_remote_graph_steps.py @@ -275,9 +275,7 @@ def step_register_two_graphs( @then('list_graphs should return "{graph_id_a}" and "{graph_id_b}" in sorted order') -def step_list_graphs_sorted( - context: Context, graph_id_a: str, graph_id_b: str -) -> None: +def step_list_graphs_sorted(context: Context, graph_id_a: str, graph_id_b: str) -> None: graphs = context.manager.list_graphs() expected = sorted([graph_id_a, graph_id_b]) assert graphs == expected, f"Expected {expected!r}, got {graphs!r}" @@ -457,17 +455,13 @@ def step_pg_config_ssl_mode(context: Context, expected: str) -> None: @then('the pg config async URL should start with "{prefix}"') def step_pg_config_async_url(context: Context, prefix: str) -> None: url = context.pg_config.to_url(async_driver=True) - assert url.startswith(prefix), ( - f"Expected URL to start with {prefix!r}, got {url!r}" - ) + assert url.startswith(prefix), f"Expected URL to start with {prefix!r}, got {url!r}" @then('the pg config sync URL should start with "{prefix}"') def step_pg_config_sync_url(context: Context, prefix: str) -> None: url = context.pg_config.to_url(async_driver=False) - assert url.startswith(prefix), ( - f"Expected URL to start with {prefix!r}, got {url!r}" - ) + assert url.startswith(prefix), f"Expected URL to start with {prefix!r}, got {url!r}" @when("I try to create a PostgreSQLConnectionConfig with empty host") diff --git a/repo b/repo new file mode 160000 index 000000000..bcf66f270 --- /dev/null +++ b/repo @@ -0,0 +1 @@ +Subproject commit bcf66f270832fe71738b48f79a00c49e6b695749 diff --git a/robot/helper_langgraph_platform_integration.py b/robot/helper_langgraph_platform_integration.py index 96b04fb19..9d731bc61 100644 --- a/robot/helper_langgraph_platform_integration.py +++ b/robot/helper_langgraph_platform_integration.py @@ -62,17 +62,13 @@ def manager_not_configured() -> None: try: manager.list_graphs() - raise AssertionError( - "Expected RemoteGraphNotAvailableError from list_graphs" - ) + raise AssertionError("Expected RemoteGraphNotAvailableError from list_graphs") except RemoteGraphNotAvailableError: pass try: manager.invoke("test-actor", {}) - raise AssertionError( - "Expected RemoteGraphNotAvailableError from invoke" - ) + raise AssertionError("Expected RemoteGraphNotAvailableError from invoke") except RemoteGraphNotAvailableError: pass @@ -234,9 +230,7 @@ def postgresql_config() -> None: assert config.database == "cleveragents", ( f"Expected database='cleveragents', got {config.database!r}" ) - assert config.username == "app", ( - f"Expected username='app', got {config.username!r}" - ) + assert config.username == "app", f"Expected username='app', got {config.username!r}" assert config.port == 5432, f"Expected port=5432, got {config.port}" assert config.ssl_mode == "prefer", ( f"Expected ssl_mode='prefer', got {config.ssl_mode!r}" @@ -251,12 +245,8 @@ def postgresql_config() -> None: assert async_url.startswith("postgresql+asyncpg://"), ( f"Expected async URL to start with 'postgresql+asyncpg://', got {async_url!r}" ) - assert "db.example.com" in async_url, ( - f"Expected host in URL, got {async_url!r}" - ) - assert "cleveragents" in async_url, ( - f"Expected database in URL, got {async_url!r}" - ) + assert "db.example.com" in async_url, f"Expected host in URL, got {async_url!r}" + assert "cleveragents" in async_url, f"Expected database in URL, got {async_url!r}" # Sync URL sync_url = config.to_url(async_driver=False)