pre-commit fixes

agent_test.py

@@ -1,18 +1,18 @@
 import asyncio
-from synapseclient import Synapse
 import json
+
+from synapseclient import Synapse
 from synapseclient.api import (
-    register_agent,
     get_agent,
-    start_session,
-    get_session,
-    update_session,
-    send_prompt,
     get_response,
+    get_session,
     get_trace,
+    register_agent,
+    send_prompt,
+    start_session,
+    update_session,
 )
 
-
 AWS_AGENT_ID = "APLZVUZ4HR"
 
 

synapseclient/api/__init__.py

@@ -1,4 +1,14 @@
 # These are all of the models that are used by the Synapse client.
+from .agent_services import (
+    get_agent,
+    get_response,
+    get_session,
+    get_trace,
+    register_agent,
+    send_prompt,
+    start_session,
+    update_session,
+)
 from .annotations import set_annotations, set_annotations_async
 from .configuration_services import (
     get_client_authenticated_s3_profile,
@@ -38,16 +48,6 @@
     put_file_multipart_add,
     put_file_multipart_complete,
 )
-from .agent_services import (
-    register_agent,
-    get_agent,
-    start_session,
-    get_session,
-    update_session,
-    send_prompt,
-    get_response,
-    get_trace,
-)
 
 __all__ = [
     # annotations

synapseclient/api/agent_services.py

@@ -2,8 +2,8 @@
 <https://rest-docs.synapse.org/rest/#org.sagebionetworks.repo.web.controller.AgentController>
 """
 
-import json
 import asyncio
+import json
 from typing import TYPE_CHECKING, Any, Dict, Optional
 
 if TYPE_CHECKING: