From f288e91fd981bdd538f461e06d2c6be5a0456d9d Mon Sep 17 00:00:00 2001 From: Chirag Jain Date: Mon, 13 Jul 2026 22:06:17 +0530 Subject: [PATCH] Revert support for draft session --- src/truefoundry_gateway_sdk/agents/agent_session.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/src/truefoundry_gateway_sdk/agents/agent_session.py b/src/truefoundry_gateway_sdk/agents/agent_session.py index faa9c75..792967d 100644 --- a/src/truefoundry_gateway_sdk/agents/agent_session.py +++ b/src/truefoundry_gateway_sdk/agents/agent_session.py @@ -9,7 +9,6 @@ if typing.TYPE_CHECKING: from ..client import AsyncTrueFoundryGateway, TrueFoundryGateway from ..core.request_options import RequestOptions - from ..types.draft_session import DraftSession as RawDraftSession from ..types.list_session_events_response import ListSessionEventsResponse from ..types.list_turns_response import ListTurnsResponse from ..types.previous_turn_id_input import PreviousTurnIdInput @@ -71,9 +70,9 @@ class AgentSession: A session enriched with convenience methods: prepare_turn, list_turns, get_turn, list_events, cancel. """ - def __init__(self, session: typing.Union[RawSession, RawDraftSession], client: TrueFoundryGateway) -> None: + def __init__(self, session: RawSession, client: TrueFoundryGateway) -> None: self._id: str = session.id - self._agent_name: typing.Optional[str] = session.agent_name + self._agent_name: str = session.agent_name self._title: typing.Optional[str] = session.title self._created_by_subject: Subject = session.created_by_subject self._created_at: str = session.created_at @@ -94,7 +93,7 @@ def id(self) -> str: return self._id @property - def agent_name(self) -> typing.Optional[str]: + def agent_name(self) -> str: """ Returns ------- @@ -280,9 +279,9 @@ class AsyncAgentSession: Async version of AgentSession. """ - def __init__(self, session: typing.Union[RawSession, RawDraftSession], client: AsyncTrueFoundryGateway) -> None: + def __init__(self, session: RawSession, client: AsyncTrueFoundryGateway) -> None: self._id: str = session.id - self._agent_name: typing.Optional[str] = session.agent_name + self._agent_name: str = session.agent_name self._title: typing.Optional[str] = session.title self._created_by_subject: Subject = session.created_by_subject self._created_at: str = session.created_at @@ -303,7 +302,7 @@ def id(self) -> str: return self._id @property - def agent_name(self) -> typing.Optional[str]: + def agent_name(self) -> str: """ Returns -------