add to mixins

Sage-Bionetworks · Jan 16, 2025 · 33639a6 · 33639a6
1 parent 6197f38
commit 33639a6
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/synapseclient/models/agent.py b/synapseclient/models/agent.py
@@ -14,7 +14,7 @@
 )
 from synapseclient.core.async_utils import async_to_sync, otel_trace_method
 from synapseclient.core.constants.concrete_types import AGENT_CHAT_REQUEST
-from synapseclient.models.mixins.asynchronous_job import AsynchronousCommunicator
+from synapseclient.models.mixins import AsynchronousCommunicator
 from synapseclient.models.protocols.agent_protocol import (
     AgentSessionSynchronousProtocol,
     AgentSynchronousProtocol,
@@ -193,9 +193,9 @@ class AgentSession(AgentSessionSynchronousProtocol):
     id: Optional[str] = None
     """The unique ID of the agent session. Can only be used by the user that created it."""
 
-    access_level: Optional[AgentSessionAccessLevel] = (
-        AgentSessionAccessLevel.PUBLICLY_ACCESSIBLE
-    )
+    access_level: Optional[
+        AgentSessionAccessLevel
+    ] = AgentSessionAccessLevel.PUBLICLY_ACCESSIBLE
     """The access level of the agent session.
         One of PUBLICLY_ACCESSIBLE, READ_YOUR_PRIVATE_DATA, or WRITE_YOUR_PRIVATE_DATA.
         Defaults to PUBLICLY_ACCESSIBLE.

diff --git a/synapseclient/models/mixins/__init__.py b/synapseclient/models/mixins/__init__.py
@@ -1,9 +1,11 @@
 """References to the mixins that are used in the Synapse models."""
 
 from synapseclient.models.mixins.access_control import AccessControllable
+from synapseclient.models.mixins.asynchronous_job import AsynchronousCommunicator
 from synapseclient.models.mixins.storable_container import StorableContainer
 
 __all__ = [
     "AccessControllable",
     "StorableContainer",
+    "AsynchronousCommunicator",
 ]