From fbcfcef506f43142b809e7135af52e4a8160c1df Mon Sep 17 00:00:00 2001 From: bwmac Date: Wed, 8 Jan 2025 14:41:47 -0500 Subject: [PATCH] pre-commit fixes --- agent_test.py | 14 +++++++------- synapseclient/api/__init__.py | 20 ++++++++++---------- synapseclient/api/agent_services.py | 2 +- 3 files changed, 18 insertions(+), 18 deletions(-) diff --git a/agent_test.py b/agent_test.py index 9ea91ca48..3bf56bf2b 100644 --- a/agent_test.py +++ b/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" diff --git a/synapseclient/api/__init__.py b/synapseclient/api/__init__.py index e42d2aab3..fe94e0de5 100644 --- a/synapseclient/api/__init__.py +++ b/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 diff --git a/synapseclient/api/agent_services.py b/synapseclient/api/agent_services.py index b1fdd28c4..821541a93 100644 --- a/synapseclient/api/agent_services.py +++ b/synapseclient/api/agent_services.py @@ -2,8 +2,8 @@ """ -import json import asyncio +import json from typing import TYPE_CHECKING, Any, Dict, Optional if TYPE_CHECKING: