vantage-cli 0.1.5__tar.gz → 0.1.7__tar.gz

{vantage_cli-0.1.5 → vantage_cli-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vantage-cli
-Version: 0.1.5
+Version: 0.1.7
 Summary: The Vantage Compute CLI.
 Author-email: jamesbeedy <james@vantagecompute.ai>
 License: MIT
@@ -39,6 +39,8 @@ Description-Content-Type: text/markdown
 
 # Vantage CLI
 
+A modern Python CLI tool to interface to Vantage Compute.
+
 [![License](https://img.shields.io/badge/license-Apache%202.0-green.svg)](LICENSE)
 [![Python](https://img.shields.io/badge/python-3.12+-blue.svg)](https://python.org)
 [![PyPI](https://img.shields.io/pypi/v/vantage-cli.svg)](https://pypi.org/project/vantage-cli/)
@@ -48,12 +50,13 @@ Description-Content-Type: text/markdown
 ![Pull Requests](https://img.shields.io/github/issues-pr/vantagecompute/vantage-cli?label=pull-requests&logo=github&style=plastic)
 ![GitHub Contributors](https://img.shields.io/github/contributors/vantagecompute/vantage-cli?logo=github&style=plastic)
 
-</div>
-
-A modern Python CLI tool to interface to Vantage Compute.
+</br>
 
 ## 🚀 Quick Start
 
+</div>
+
+
 ### Option 1: Install from PyPI (Recommended)
 
 ```bash

{vantage_cli-0.1.5 → vantage_cli-0.1.7}/README.md

@@ -7,6 +7,8 @@
 
 # Vantage CLI
 
+A modern Python CLI tool to interface to Vantage Compute.
+
 [![License](https://img.shields.io/badge/license-Apache%202.0-green.svg)](LICENSE)
 [![Python](https://img.shields.io/badge/python-3.12+-blue.svg)](https://python.org)
 [![PyPI](https://img.shields.io/pypi/v/vantage-cli.svg)](https://pypi.org/project/vantage-cli/)
@@ -16,12 +18,13 @@
 ![Pull Requests](https://img.shields.io/github/issues-pr/vantagecompute/vantage-cli?label=pull-requests&logo=github&style=plastic)
 ![GitHub Contributors](https://img.shields.io/github/contributors/vantagecompute/vantage-cli?logo=github&style=plastic)
 
-</div>
-
-A modern Python CLI tool to interface to Vantage Compute.
+</br>
 
 ## 🚀 Quick Start
 
+</div>
+
+
 ### Option 1: Install from PyPI (Recommended)
 
 ```bash

{vantage_cli-0.1.5 → vantage_cli-0.1.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "vantage-cli"
-version = "0.1.5"
+version = "0.1.7"
 description = "The Vantage Compute CLI."
 authors = [
     {name = "jamesbeedy", email = "james@vantagecompute.ai"}

{vantage_cli-0.1.5 → vantage_cli-0.1.7}/uv.lock

@@ -1747,7 +1747,7 @@ wheels = [
 
 [[package]]
 name = "vantage-cli"
-version = "0.1.5"
+version = "0.1.7"
 source = { editable = "." }
 dependencies = [
     { name = "aiohttp" },

{vantage_cli-0.1.5 → vantage_cli-0.1.7}/vantage_cli/__init__.py

@@ -55,8 +55,28 @@ class AsyncTyper(typer.Typer):
     def maybe_run_async(func: Callable, *args: Any, **kwargs: Any) -> Any:
         """Run function asynchronously if it's a coroutine, otherwise run normally."""
         if inspect.iscoroutinefunction(func):
-            return asyncio.run(func(*args, **kwargs))
-        return func(*args, **kwargs)
+            # Check if we're already in an event loop
+            try:
+                asyncio.get_running_loop()
+                # We're in an event loop, cannot use asyncio.run()
+                # This typically happens in tests, return the coroutine
+                return func(*args, **kwargs)
+            except RuntimeError:
+                # No event loop running, safe to use asyncio.run()
+                return asyncio.run(func(*args, **kwargs))
+        else:
+            # Check if the function call returns a coroutine
+            result = func(*args, **kwargs)
+            if inspect.iscoroutine(result):
+                # Function returned a coroutine, need to run it
+                try:
+                    asyncio.get_running_loop()
+                    # We're in an event loop, return the coroutine
+                    return result
+                except RuntimeError:
+                    # No event loop running, safe to use asyncio.run()
+                    return asyncio.run(result)
+            return result
 
     def command(
         self,

{vantage_cli-0.1.5 → vantage_cli-0.1.7}/vantage_cli/commands/alias/__init__.py

@@ -6,6 +6,7 @@ from .clusters import clusters_command
 from .federations import federations_command
 from .networks import networks_command
 from .notebooks import notebooks_command
+from .profiles import profiles_command
 from .teams import teams_command
 
 __all__ = [
@@ -15,5 +16,6 @@ __all__ = [
     "federations_command",
     "networks_command",
     "notebooks_command",
+    "profiles_command",
     "teams_command",
 ]

vantage_cli-0.1.7/vantage_cli/commands/alias/profiles.py

@@ -0,0 +1,17 @@
+# © 2025 Vantage Compute, Inc. All rights reserved.
+# Confidential and proprietary. Unauthorized use prohibited.
+"""Alias command for profiles -> profile list."""
+
+import typer
+
+from vantage_cli.commands.profile.crud import list_profiles
+
+
+def profiles_command(
+    ctx: typer.Context,
+):
+    """List all profiles (alias for 'vantage profile list')."""
+    # Extract flags from the AsyncTyper context
+    json_output = getattr(ctx.obj, "json_output", False) if ctx.obj else False
+    verbose = getattr(ctx.obj, "verbose", False) if ctx.obj else False
+    list_profiles(ctx, json_output=json_output, verbose=verbose)

{vantage_cli-0.1.5 → vantage_cli-0.1.7}/vantage_cli/commands/profile/crud.py

@@ -47,11 +47,15 @@ def create_profile(
     activate: Annotated[
         bool, typer.Option("--activate", help="Activate this profile after creation")
     ] = False,
+    json_output: Annotated[
+        bool, typer.Option("--json", "-j", help="Output in JSON format")
+    ] = False,
+    verbose: Annotated[
+        bool, typer.Option("--verbose", "-v", help="Enable verbose terminal output")
+    ] = False,
 ):
     """Create a new Vantage CLI profile."""
     # Get the effective JSON output preference
-    # Get JSON flag from context (automatically set by AsyncTyper)
-    json_output = getattr(ctx.obj, "json_output", False) if ctx.obj else False
     effective_json = get_effective_json_output(ctx, json_output)
 
     # Check if profile already exists
@@ -138,11 +142,16 @@ def delete_profile(
     ctx: typer.Context,
     profile_name: Annotated[str, typer.Argument(help="Name of the profile to delete")],
     force: Annotated[bool, typer.Option("--force", "-f", help="Skip confirmation prompt")] = False,
+    json_output: Annotated[
+        bool, typer.Option("--json", "-j", help="Output in JSON format")
+    ] = False,
+    verbose: Annotated[
+        bool, typer.Option("--verbose", "-v", help="Enable verbose terminal output")
+    ] = False,
 ):
     """Delete a Vantage CLI profile."""
     # Get the effective JSON output preference
-    # Get JSON flag from context (automatically set by AsyncTyper)
-    json_output = getattr(ctx.obj, "json_output", False) if ctx.obj else False
+    effective_json = get_effective_json_output(ctx, json_output)
     effective_json = get_effective_json_output(ctx, json_output)
 
     # Check if profile exists
@@ -250,11 +259,15 @@ def delete_profile(
 def get_profile(
     ctx: typer.Context,
     profile_name: Annotated[str, typer.Argument(help="Name of the profile to get details for")],
+    json_output: Annotated[
+        bool, typer.Option("--json", "-j", help="Output in JSON format")
+    ] = False,
+    verbose: Annotated[
+        bool, typer.Option("--verbose", "-v", help="Enable verbose terminal output")
+    ] = False,
 ):
     """Get details of a specific Vantage CLI profile."""
     # Get the effective JSON output preference
-    # Get JSON flag from context (automatically set by AsyncTyper)
-    json_output = getattr(ctx.obj, "json_output", False) if ctx.obj else False
     effective_json = get_effective_json_output(ctx, json_output)
 
     # Check if profile exists
@@ -305,10 +318,16 @@ def get_profile(
             )
 
 
-def list_profiles(ctx: typer.Context):
+def list_profiles(
+    ctx: typer.Context,
+    json_output: Annotated[
+        bool, typer.Option("--json", "-j", help="Output in JSON format")
+    ] = False,
+    verbose: Annotated[
+        bool, typer.Option("--verbose", "-v", help="Enable verbose terminal output")
+    ] = False,
+):
     """List all Vantage CLI profiles."""
-    # Get JSON flag from context (automatically set by AsyncTyper)
-    json_output = getattr(ctx.obj, "json_output", False) if ctx.obj else False
     # Get the effective JSON output preference
     effective_json = get_effective_json_output(ctx, json_output)
 
@@ -366,11 +385,15 @@ def list_profiles(ctx: typer.Context):
 def use_profile(
     ctx: typer.Context,
     profile_name: Annotated[str, typer.Argument(help="Name of the profile to activate")],
+    json_output: Annotated[
+        bool, typer.Option("--json", "-j", help="Output in JSON format")
+    ] = False,
+    verbose: Annotated[
+        bool, typer.Option("--verbose", "-v", help="Enable verbose terminal output")
+    ] = False,
 ):
     """Activate a profile for use in the current session."""
     # Get the effective JSON output preference
-    # Get JSON flag from context (automatically set by AsyncTyper)
-    json_output = getattr(ctx.obj, "json_output", False) if ctx.obj else False
     effective_json = get_effective_json_output(ctx, json_output)
 
     # Check if profile exists

{vantage_cli-0.1.5 → vantage_cli-0.1.7}/vantage_cli/main.py

@@ -26,6 +26,7 @@ from vantage_cli.commands.alias import (
     federations_command,
     networks_command,
     notebooks_command,
+    profiles_command,
     teams_command,
 )
 from vantage_cli.commands.cloud import clouds_app
@@ -202,6 +203,7 @@ async def logout(ctx: typer.Context):
 
 
 @app.command()
+@handle_abort
 @with_cache
 @attach_settings
 async def whoami(ctx: typer.Context):
@@ -325,6 +327,7 @@ app.command("clusters", hidden=True)(clusters_command)
 app.command("federations", hidden=True)(federations_command)
 app.command("networks", hidden=True)(networks_command)
 app.command("notebooks", hidden=True)(notebooks_command)
+app.command("profiles", hidden=True)(profiles_command)
 app.command("teams", hidden=True)(teams_command)
 
 

vantage_cli-0.1.5/debug_annotations.py

@@ -1,38 +0,0 @@
-#!/usr/bin/env python3
-"""Debug script to check if JsonOption annotation is the issue."""
-
-import inspect
-import typer
-import functools
-from typing_extensions import Annotated
-
-JsonOption = Annotated[bool, typer.Option("--json", "-j", help="Output in JSON format")]
-
-def test_wrapper1(ctx: typer.Context, json_output: bool = False):
-    """Wrapper with plain bool type."""
-    return "test"
-
-def test_wrapper2(ctx: typer.Context, json_output: JsonOption = False):
-    """Wrapper with JsonOption type.""" 
-    return "test"
-
-print("test_wrapper1 signature (plain bool):")
-print(inspect.signature(test_wrapper1))
-
-print("\ntest_wrapper2 signature (JsonOption):")
-print(inspect.signature(test_wrapper2))
-
-# Test parameter inspection
-sig1 = inspect.signature(test_wrapper1)
-sig2 = inspect.signature(test_wrapper2)
-
-print(f"\ntest_wrapper1 parameters: {list(sig1.parameters.keys())}")
-print(f"test_wrapper2 parameters: {list(sig2.parameters.keys())}")
-
-print(f"\njson_output param in wrapper1: {'json_output' in sig1.parameters}")
-print(f"json_output param in wrapper2: {'json_output' in sig2.parameters}")
-
-if 'json_output' in sig2.parameters:
-    param = sig2.parameters['json_output']
-    print(f"json_output annotation: {param.annotation}")
-    print(f"json_output default: {param.default}")

vantage_cli-0.1.5/debug_decorators.py

@@ -1,46 +0,0 @@
-#!/usr/bin/env python3
-"""Debug script to understand how the decorators work."""
-
-import inspect
-import typer
-from vantage_cli.decorators import vantage_command, JsonOption
-
-app = typer.Typer()
-
-# Test the decorator manually
-def test_func(ctx: typer.Context):
-    return "test"
-
-print("Original function signature:")
-print(inspect.signature(test_func))
-
-# Apply the decorator manually to see what happens
-decorator = vantage_command(app, "test")
-wrapped = decorator(test_func)
-
-print("\nAfter applying vantage_command:")
-print(f"wrapped is: {wrapped}")
-print(f"wrapped type: {type(wrapped)}")
-print(f"wrapped signature: {inspect.signature(wrapped)}")
-
-print(f"\nRegistered commands: {len(app.registered_commands)}")
-if app.registered_commands:
-    cmd = app.registered_commands[0]
-    print(f"Command name: {cmd.name}")
-    print(f"Command callback: {cmd.callback}")
-    print(f"Command callback type: {type(cmd.callback)}")
-    if cmd.callback:
-        print(f"Command callback signature: {inspect.signature(cmd.callback)}")
-
-# Let's also test what the wrapper function looks like in isolation
-import functools
-
-def manual_wrapper(func):
-    @functools.wraps(func)
-    def wrapper(ctx: typer.Context, *args, json_output: JsonOption = False, **kwargs):
-        print(f"Wrapper called with json_output={json_output}")
-        return func(ctx, *args, **kwargs)
-    return wrapper
-
-wrapped_manual = manual_wrapper(test_func)
-print(f"\nManual wrapper signature: {inspect.signature(wrapped_manual)}")

vantage_cli-0.1.5/debug_typer.py

@@ -1,41 +0,0 @@
-#!/usr/bin/env python3
-"""Debug script to understand how typer.command works."""
-
-import inspect
-import typer
-import functools
-from typing_extensions import Annotated
-
-app = typer.Typer()
-
-JsonOption = Annotated[bool, typer.Option("--json", "-j", help="Output in JSON format")]
-
-def original_func(ctx: typer.Context):
-    return "original"
-
-print("Original function signature:")
-print(inspect.signature(original_func))
-
-# Create a wrapper manually  
-@functools.wraps(original_func)
-def wrapper(ctx: typer.Context, *args, json_output: JsonOption = False, **kwargs):
-    return original_func(ctx, *args, **kwargs)
-
-print(f"\nWrapper signature: {inspect.signature(wrapper)}")
-
-# Test what app.command returns
-result = app.command("test")(wrapper)
-
-print(f"\napp.command result: {result}")
-print(f"result is wrapper: {result is wrapper}")
-print(f"result is original_func: {result is original_func}")
-print(f"result signature: {inspect.signature(result)}")
-
-# Check what's registered
-print(f"\nRegistered commands: {len(app.registered_commands)}")
-if app.registered_commands:
-    cmd = app.registered_commands[0]
-    print(f"Command callback: {cmd.callback}")
-    print(f"Callback is wrapper: {cmd.callback is wrapper}")
-    print(f"Callback is result: {cmd.callback is result}")
-    print(f"Callback signature: {inspect.signature(cmd.callback)}")

vantage_cli-0.1.5/debug_wraps.py

@@ -1,57 +0,0 @@
-#!/usr/bin/env python3
-"""Debug script to test functools.wraps behavior."""
-
-import inspect
-import typer
-import functools
-from typing_extensions import Annotated
-
-JsonOption = Annotated[bool, typer.Option("--json", "-j", help="Output in JSON format")]
-
-def original_func(ctx: typer.Context):
-    """Original function."""
-    return "original"
-
-print("Original function signature:")
-print(inspect.signature(original_func))
-
-# Test wrapper WITHOUT functools.wraps
-def wrapper_no_wraps(ctx: typer.Context, json_output: JsonOption = False):
-    """Wrapper without functools.wraps."""
-    return original_func(ctx)
-
-print(f"\nWrapper without @functools.wraps signature:")
-print(inspect.signature(wrapper_no_wraps))
-
-# Test wrapper WITH functools.wraps
-@functools.wraps(original_func)
-def wrapper_with_wraps(ctx: typer.Context, json_output: JsonOption = False):
-    """Wrapper with functools.wraps."""
-    return original_func(ctx)
-
-print(f"\nWrapper with @functools.wraps signature:")
-print(inspect.signature(wrapper_with_wraps))
-
-print(f"\nWrapper without wraps parameters: {list(inspect.signature(wrapper_no_wraps).parameters.keys())}")
-print(f"Wrapper with wraps parameters: {list(inspect.signature(wrapper_with_wraps).parameters.keys())}")
-
-# Check what __wrapped__ attribute contains
-if hasattr(wrapper_with_wraps, '__wrapped__'):
-    print(f"\n__wrapped__ signature: {inspect.signature(wrapper_with_wraps.__wrapped__)}")
-    
-# Test manually setting signature
-wrapper_manual = lambda ctx, json_output=False: original_func(ctx)
-wrapper_manual.__name__ = original_func.__name__
-wrapper_manual.__doc__ = original_func.__doc__
-
-print(f"\nManual wrapper (before signature): {inspect.signature(wrapper_manual)}")
-
-# Manually set the signature
-from inspect import Parameter, Signature
-params = [
-    Parameter('ctx', Parameter.POSITIONAL_OR_KEYWORD, annotation=typer.Context),
-    Parameter('json_output', Parameter.KEYWORD_ONLY, default=False, annotation=JsonOption)
-]
-wrapper_manual.__signature__ = Signature(params)
-
-print(f"Manual wrapper (after signature): {inspect.signature(wrapper_manual)}")

vantage_cli-0.1.5/fix_cluster_tests.py

@@ -1,57 +0,0 @@
-#!/usr/bin/env python3
-"""Script to fix cluster tests by applying json_output parameter removal pattern."""
-
-import re
-
-def fix_cluster_tests():
-    file_path = "tests/unit/test_cluster_commands.py"
-    
-    with open(file_path, 'r') as f:
-        content = f.read()
-    
-    # Add SimpleNamespace import if not present
-    if "from types import SimpleNamespace" not in content:
-        content = content.replace(
-            "from unittest.mock import Mock, patch",
-            "from types import SimpleNamespace\nfrom unittest.mock import Mock, patch"
-        )
-    
-    # Fix mock_context fixtures
-    fixture_pattern = r'(ctx\.obj = Mock\(\)\s+ctx\.obj\.settings = Mock\(\)\s+ctx\.obj\.profile = "default")'
-    fixture_replacement = 'ctx.obj = SimpleNamespace(\n            profile="default",\n            verbose=False,\n            json_output=False\n        )'
-    
-    content = re.sub(fixture_pattern, fixture_replacement, content, flags=re.MULTILINE)
-    
-    # Fix function calls - remove json_output parameters
-    patterns_to_fix = [
-        (r'list_clusters\(([^,]+), json_output=\w+\)', r'list_clusters(\1)'),
-        (r'delete_cluster\(([^,]+), ([^,]+), force=(\w+), json_output=\w+\)', r'delete_cluster(\1, \2, force=\3)'),
-        (r'get_cluster\(([^,]+), ([^,]+), json_output=\w+\)', r'get_cluster(\1, \2)'),
-    ]
-    
-    for pattern, replacement in patterns_to_fix:
-        content = re.sub(pattern, replacement, content)
-    
-    # Add context setting before function calls for json_output tests
-    content = re.sub(
-        r'(mock_get_effective_json_output\.return_value = True)\s+(# Call the function)',
-        r'\1\n        mock_ctx.obj.json_output = True\n        \2',
-        content,
-        flags=re.MULTILINE
-    )
-    
-    # Handle cases where json_output is False
-    content = re.sub(
-        r'(await list_clusters\(mock_ctx\))\s+(# Check that render_json was called)',
-        r'mock_ctx.obj.json_output = True\n        \1\n        \2',
-        content,
-        flags=re.MULTILINE
-    )
-    
-    with open(file_path, 'w') as f:
-        f.write(content)
-    
-    print("Cluster tests fixed!")
-
-if __name__ == "__main__":
-    fix_cluster_tests()

vantage_cli-0.1.5/fix_config_tests.py

@@ -1,51 +0,0 @@
-#!/usr/bin/env python3
-"""Script to fix config tests by applying json_output parameter removal pattern."""
-
-import re
-
-def fix_config_tests():
-    file_path = "tests/unit/test_config_commands.py"
-    
-    with open(file_path, 'r') as f:
-        content = f.read()
-    
-    # Add SimpleNamespace import if not present
-    if "from types import SimpleNamespace" not in content:
-        content = content.replace(
-            "from unittest.mock import AsyncMock, Mock, patch",
-            "from types import SimpleNamespace\nfrom unittest.mock import AsyncMock, Mock, patch"
-        )
-    
-    # Fix mock_context fixtures - replace Mock objects with SimpleNamespace
-    fixture_pattern = r'(ctx\.obj = Mock\(\)[^}]+ctx\.obj\.profile = "default"[^}]*return ctx)'
-    fixture_replacement = '''ctx.obj = SimpleNamespace(
-            profile="default",
-            verbose=False,
-            json_output=False
-        )
-        return ctx'''
-    
-    content = re.sub(fixture_pattern, fixture_replacement, content, flags=re.MULTILINE | re.DOTALL)
-    
-    # Fix function calls - remove json_output parameters and add context setting
-    patterns_to_fix = [
-        (r'(await clear_config\([^,]+), json_output=(\w+)\)', r'mock_ctx.obj.json_output = \2\n        \1)'),
-    ]
-    
-    for pattern, replacement in patterns_to_fix:
-        content = re.sub(pattern, replacement, content)
-    
-    # Also fix any direct call patterns
-    content = re.sub(
-        r'await clear_config\(([^,]+), force=(\w+), json_output=(\w+)\)',
-        r'mock_ctx.obj.json_output = \3\n        await clear_config(\1, force=\2)',
-        content
-    )
-    
-    with open(file_path, 'w') as f:
-        f.write(content)
-    
-    print("Config tests fixed!")
-
-if __name__ == "__main__":
-    fix_config_tests()

vantage_cli-0.1.5/fix_federation_tests.py

@@ -1,47 +0,0 @@
-#!/usr/bin/env python3
-"""Script to fix federation tests by applying json_output parameter removal pattern."""
-
-import re
-
-def fix_federation_tests():
-    file_path = "tests/unit/test_federation_commands.py"
-    
-    with open(file_path, 'r') as f:
-        content = f.read()
-    
-    # Fix mock_context fixtures
-    fixture_pattern = r'(ctx\.obj = Mock\(\)\s+ctx\.obj\.settings = Mock\(\)\s+ctx\.obj\.profile = "default")'
-    fixture_replacement = 'ctx.obj = SimpleNamespace(\n            profile="default",\n            verbose=False,\n            json_output=False\n        )'
-    
-    content = re.sub(fixture_pattern, fixture_replacement, content, flags=re.MULTILINE)
-    
-    # Fix function calls - remove json_output parameters
-    patterns_to_fix = [
-        (r'list_federations\(([^,]+), json_output=\w+\)', r'list_federations(\1)'),
-        (r'create_federation\(([^,]+), ([^,]+), ([^,]+), json_output=\w+\)', r'create_federation(\1, \2, \3)'),
-        (r'delete_federation\(([^,]+), ([^,]+), force=(\w+), json_output=\w+\)', r'delete_federation(\1, \2, force=\3)'),
-        (r'get_federation\(([^,]+), ([^,]+), json_output=\w+\)', r'get_federation(\1, \2)'),
-        (r'update_federation\(([^,]+), ([^,]+), json_output=\w+\)', r'update_federation(\1, \2)'),
-    ]
-    
-    for pattern, replacement in patterns_to_fix:
-        content = re.sub(pattern, replacement, content)
-    
-    # Add context setting before function calls
-    json_output_patterns = [
-        (r'(mock_get_json_output\.return_value = True)\s+# Run the command\s+import asyncio\s+(asyncio\.run\(list_federations\(mock_context\)\))', 
-         r'\1\n        mock_context.obj.json_output = True\n        # Run the command\n        import asyncio\n        \2'),
-        (r'(mock_get_json_output\.return_value = False)\s+mock_console = Mock\(\)\s+mock_console_class\.return_value = mock_console\s+# Run the command\s+import asyncio\s+(asyncio\.run\(list_federations\(mock_context\)\))',
-         r'\1\n        mock_console = Mock()\n        mock_console_class.return_value = mock_console\n        mock_context.obj.json_output = False\n        # Run the command\n        import asyncio\n        \2'),
-    ]
-    
-    for pattern, replacement in json_output_patterns:
-        content = re.sub(pattern, replacement, content, flags=re.MULTILINE | re.DOTALL)
-    
-    with open(file_path, 'w') as f:
-        f.write(content)
-    
-    print("Federation tests fixed!")
-
-if __name__ == "__main__":
-    fix_federation_tests()