djangoviz/management: setup user-agent header (#12)

ariga · May 22, 2023 · d1695b5 · d1695b5
1 parent 738cf78
commit d1695b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/djangoviz/management/commands/djangoviz.py b/djangoviz/management/commands/djangoviz.py
@@ -118,7 +118,7 @@ def handle(self, *args, **options):
             self.stderr.write(self.style.ERROR("no migrations found"))
             return
         client = GraphQLClient(endpoint=API_ENDPOINT)
-        client.inject_token("user-agent", f"djangoviz/{__version__}")
+        client.inject_token(token=f"djangoviz/{__version__}", headername="user-agent")
         try:
             atlas_schema = _get_atlas_schema(client, migrations, db_driver)
             if atlas_schema is None: