fix lint

Signed-off-by: kerthcet <kerthcet@gmail.com>

Author: kerthcet

alphatrion/server/graphql/resolvers.py

@@ -7,12 +7,12 @@
 
 from alphatrion import envs
 from alphatrion.artifact import artifact
+from alphatrion.server.graphql.types import ArtifactFile
 from alphatrion.storage import runtime
 from alphatrion.storage.sql_models import (
     FINISHED_STATUS,
     Status,
 )
-from alphatrion.server.graphql.types import ArtifactFile
 
 from .types import (
     AddUserToTeamInput,
@@ -376,9 +376,11 @@ async def list_artifact_files(
                 # Determine content type based on file extension
                 if filename.endswith(".json"):
                     content_type = "application/json"
-                elif filename.endswith(".txt") or filename.endswith(".log"):
-                    content_type = "text/plain"
-                elif filename.endswith((".py", ".js", ".ts", ".tsx", ".jsx")):
+                elif (
+                    filename.endswith(".txt")
+                    or filename.endswith(".log")
+                    or filename.endswith((".py", ".js", ".ts", ".tsx", ".jsx"))
+                ):
                     content_type = "text/plain"
                 else:
                     content_type = "text/plain"
@@ -395,7 +397,10 @@ async def list_artifact_files(
 
     @staticmethod
     async def get_artifact_content(
-        team_id: str, tag: str, repo_name: str | None = None, filename: str | None = None
+        team_id: str,
+        tag: str,
+        repo_name: str | None = None,
+        filename: str | None = None,
     ) -> ArtifactContent:
         """Get artifact content from registry."""
         try:
@@ -431,9 +436,11 @@ async def get_artifact_content(
             # Determine content type based on file extension
             if actual_filename.endswith(".json"):
                 content_type = "application/json"
-            elif actual_filename.endswith(".txt") or actual_filename.endswith(".log"):
-                content_type = "text/plain"
-            elif actual_filename.endswith((".py", ".js", ".ts", ".tsx", ".jsx")):
+            elif (
+                actual_filename.endswith(".txt")
+                or actual_filename.endswith(".log")
+                or actual_filename.endswith((".py", ".js", ".ts", ".tsx", ".jsx"))
+            ):
                 content_type = "text/plain"
             else:
                 content_type = "text/plain"

alphatrion/server/graphql/schema.py

@@ -113,7 +113,9 @@ async def artifact_content(
         repo_name: str,
         filename: str | None = None,
     ) -> ArtifactContent:
-        return await GraphQLResolvers.get_artifact_content(str(team_id), tag, repo_name, filename)
+        return await GraphQLResolvers.get_artifact_content(
+            str(team_id), tag, repo_name, filename
+        )
 
     # Dataset queries
     @strawberry.field