Skip to content

Commit

Permalink
pre-commit run
Browse files Browse the repository at this point in the history
  • Loading branch information
BWMac committed Jan 14, 2025
1 parent 389dd86 commit 15db364
Show file tree
Hide file tree
Showing 4 changed files with 9 additions and 16 deletions.
4 changes: 0 additions & 4 deletions synapseclient/api/agent_services.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,16 +2,12 @@
<https://rest-docs.synapse.org/rest/#org.sagebionetworks.repo.web.controller.AgentController>
"""

import asyncio
import json
from typing import TYPE_CHECKING, Any, Dict, Optional

if TYPE_CHECKING:
from synapseclient import Synapse

from synapseclient.core.constants.concrete_types import AGENT_CHAT_REQUEST
from synapseclient.core.exceptions import SynapseTimeoutError


async def register_agent(
cloud_agent_id: str,
Expand Down
11 changes: 5 additions & 6 deletions synapseclient/models/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,13 +13,12 @@
update_session,
)
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 AsynchronousJob
from synapseclient.models.protocols.agent_protocol import (
AgentSessionSynchronousProtocol,
AgentSynchronousProtocol,
)
from synapseclient.models.mixins.asynchronous_job import AsynchronousJob

from synapseclient.core.constants.concrete_types import AGENT_CHAT_REQUEST


class AgentType(str, Enum):
Expand Down Expand Up @@ -97,9 +96,9 @@ class AgentSession(AgentSessionSynchronousProtocol, AsynchronousJob):
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.
Expand Down
8 changes: 3 additions & 5 deletions synapseclient/models/mixins/asynchronous_job.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,13 +2,11 @@
import json
import time
from dataclasses import dataclass
from typing import TYPE_CHECKING, Optional, Dict, Any
from synapseclient.core.constants.concrete_types import AGENT_CHAT_REQUEST

from synapseclient.core.exceptions import SynapseError, SynapseTimeoutError

from enum import Enum
from typing import TYPE_CHECKING, Any, Dict, Optional

from synapseclient.core.constants.concrete_types import AGENT_CHAT_REQUEST
from synapseclient.core.exceptions import SynapseError, SynapseTimeoutError

if TYPE_CHECKING:
from synapseclient import Synapse
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from synapseclient.core.exceptions import SynapseError

if TYPE_CHECKING:
from synapseclient.models import File, Folder, Project, Table, Annotations
from synapseclient.models import Annotations, File, Folder, Project, Table


class FailureStrategy(Enum):
Expand Down

0 comments on commit 15db364

Please sign in to comment.