refactor: Lint agents-api (CI)

agents-api/agents_api/queries/agents/create_agent.py

@@ -10,14 +10,13 @@
 from sqlglot import parse_one
 from uuid_extensions import uuid7
 
-from ...metrics.counters import increase_counter
-
 from ...autogen.openapi_model import Agent, CreateAgentRequest
+from ...metrics.counters import increase_counter
 from ..utils import (
     generate_canonical_name,
     pg_query,
-    wrap_in_class,
     rewrap_exceptions,
+    wrap_in_class,
 )
 
 ModelT = TypeVar("ModelT", bound=Any)

agents-api/agents_api/queries/agents/create_or_update_agent.py

@@ -14,8 +14,8 @@
 from ..utils import (
     generate_canonical_name,
     pg_query,
-    wrap_in_class,
     rewrap_exceptions,
+    wrap_in_class,
 )
 
 ModelT = TypeVar("ModelT", bound=Any)

agents-api/agents_api/queries/agents/delete_agent.py

@@ -10,12 +10,12 @@
 from sqlglot import parse_one
 
 from ...autogen.openapi_model import ResourceDeletedResponse
-from ...metrics.counters import increase_counter
 from ...common.utils.datetime import utcnow
+from ...metrics.counters import increase_counter
 from ..utils import (
     pg_query,
-    wrap_in_class,
     rewrap_exceptions,
+    wrap_in_class,
 )
 
 ModelT = TypeVar("ModelT", bound=Any)

agents-api/agents_api/queries/agents/get_agent.py

@@ -13,8 +13,8 @@
 from ...metrics.counters import increase_counter
 from ..utils import (
     pg_query,
-    wrap_in_class,
     rewrap_exceptions,
+    wrap_in_class,
 )
 
 raw_query = """

agents-api/agents_api/queries/agents/list_agents.py

@@ -13,8 +13,8 @@
 from ...metrics.counters import increase_counter
 from ..utils import (
     pg_query,
-    wrap_in_class,
     rewrap_exceptions,
+    wrap_in_class,
 )
 
 ModelT = TypeVar("ModelT", bound=Any)

agents-api/agents_api/queries/agents/patch_agent.py

@@ -13,8 +13,8 @@
 from ...metrics.counters import increase_counter
 from ..utils import (
     pg_query,
-    wrap_in_class,
     rewrap_exceptions,
+    wrap_in_class,
 )
 
 ModelT = TypeVar("ModelT", bound=Any)

agents-api/agents_api/queries/agents/update_agent.py

@@ -13,8 +13,8 @@
 from ...metrics.counters import increase_counter
 from ..utils import (
     pg_query,
-    wrap_in_class,
     rewrap_exceptions,
+    wrap_in_class,
 )
 
 ModelT = TypeVar("ModelT", bound=Any)

agents-api/agents_api/queries/developers/get_developer.py

@@ -12,8 +12,8 @@
 from ..utils import (
     partialclass,
     pg_query,
-    wrap_in_class,
     rewrap_exceptions,
+    wrap_in_class,
 )
 
 # TODO: Add verify_developer

agents-api/agents_api/queries/entries/create_entries.py

@@ -166,7 +166,7 @@ async def add_entry_relations(
                 item.get("is_leaf", False),  # $5
             ]
         )
-        
+
     return [
         (
             session_exists_query,

agents-api/agents_api/queries/entries/list_entries.py

@@ -108,8 +108,5 @@ async def list_entries(
             [session_id, developer_id],
             "fetchrow",
         ),
-        (
-            query,
-            entry_params
-        ),
+        (query, entry_params),
     ]

agents-api/agents_api/queries/utils.py

@@ -106,7 +106,9 @@ def prepare_pg_query_args(
                 batch.append(
                     (
                         "fetchrow",
-                        AsyncPGFetchArgs(query=query, args=variables, timeout=query_timeout),
+                        AsyncPGFetchArgs(
+                            query=query, args=variables, timeout=query_timeout
+                        ),
                     )
                 )
             case _:
@@ -173,7 +175,9 @@ async def wrapper(
                             print(*args)
                             print("%" * 100)
 
-                            if method_name == "fetchrow" and (len(results) == 0 or results.get("bool") is None):
+                            if method_name == "fetchrow" and (
+                                len(results) == 0 or results.get("bool") is None
+                            ):
                                 raise asyncpg.NoDataFoundError
 
                         end = timeit and time.perf_counter()

agents-api/tests/test_entry_queries.py

@@ -3,9 +3,8 @@
 It verifies the functionality of adding, retrieving, and processing entries as defined in the schema.
 """
 
-from uuid_extensions import uuid7
-
 from fastapi import HTTPException
+from uuid_extensions import uuid7
 from ward import raises, test
 
 from agents_api.autogen.openapi_model import CreateEntryRequest

agents-api/tests/test_messages_truncation.py

@@ -1,4 +1,3 @@
-
 # from uuid_extensions import uuid7
 
 # from ward import raises, test

agents-api/tests/test_session_queries.py

@@ -7,32 +7,32 @@
 from ward import raises, test
 
 from agents_api.autogen.openapi_model import (
-    Session,
-    CreateSessionRequest,
     CreateOrUpdateSessionRequest,
-    UpdateSessionRequest,
+    CreateSessionRequest,
     PatchSessionRequest,
-    ResourceUpdatedResponse,
     ResourceDeletedResponse,
+    ResourceUpdatedResponse,
+    Session,
+    UpdateSessionRequest,
 )
 from agents_api.clients.pg import create_db_pool
 from agents_api.queries.sessions import (
     count_sessions,
+    create_or_update_session,
+    create_session,
+    delete_session,
     get_session,
     list_sessions,
-    create_session,
-    create_or_update_session,
-    update_session,
     patch_session,
-    delete_session,
+    update_session,
 )
 from tests.fixtures import (
     pg_dsn,
-    test_developer_id,
-    test_developer,
-    test_user,
     test_agent,
+    test_developer,
+    test_developer_id,
     test_session,
+    test_user,
 )