Skip to content

Commit ec43295

Browse files
monadoidmonadoid
committed
Added empty body logic for /end endpoint
1 parent 798c057 commit ec43295

File tree

1 file changed

+24
-19
lines changed

1 file changed

+24
-19
lines changed

‎src/stagehand/resources/sessions.py‎

Lines changed: 24 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,6 @@
1010

1111
from ..types import (
1212
session_act_params,
13-
session_end_params,
1413
session_start_params,
1514
session_execute_params,
1615
session_extract_params,
@@ -41,6 +40,14 @@
4140
__all__ = ["SessionsResource", "AsyncSessionsResource"]
4241

4342

43+
def _format_x_sent_at(value: Union[str, datetime] | Omit) -> str | NotGiven:
44+
if isinstance(value, datetime):
45+
return value.isoformat()
46+
if isinstance(value, Omit):
47+
return not_given
48+
return value
49+
50+
4451
class SessionsResource(SyncAPIResource):
4552
@cached_property
4653
def with_raw_response(self) -> SessionsResourceWithRawResponse:
@@ -241,7 +248,7 @@ def act(
241248
{
242249
"x-language": str(x_language) if is_given(x_language) else not_given,
243250
"x-sdk-version": x_sdk_version,
244-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
251+
"x-sent-at": _format_x_sent_at(x_sent_at),
245252
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
246253
}
247254
),
@@ -272,7 +279,6 @@ def end(
272279
self,
273280
id: str,
274281
*,
275-
_force_body: object | Omit = omit,
276282
x_language: Literal["typescript", "python", "playground"] | Omit = omit,
277283
x_sdk_version: str | Omit = omit,
278284
x_sent_at: Union[str, datetime] | Omit = omit,
@@ -313,15 +319,15 @@ def end(
313319
{
314320
"x-language": str(x_language) if is_given(x_language) else not_given,
315321
"x-sdk-version": x_sdk_version,
316-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
322+
"x-sent-at": _format_x_sent_at(x_sent_at),
317323
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
318324
}
319325
),
320326
**(extra_headers or {}),
321327
}
322328
return self._post(
323329
f"/v1/sessions/{id}/end",
324-
body=maybe_transform({"_force_body": _force_body}, session_end_params.SessionEndParams),
330+
body={}, # Empty object to satisfy Content-Type requirement
325331
options=make_request_options(
326332
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
327333
),
@@ -499,7 +505,7 @@ def execute(
499505
{
500506
"x-language": str(x_language) if is_given(x_language) else not_given,
501507
"x-sdk-version": x_sdk_version,
502-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
508+
"x-sent-at": _format_x_sent_at(x_sent_at),
503509
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
504510
}
505511
),
@@ -712,7 +718,7 @@ def extract(
712718
{
713719
"x-language": str(x_language) if is_given(x_language) else not_given,
714720
"x-sdk-version": x_sdk_version,
715-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
721+
"x-sent-at": _format_x_sent_at(x_sent_at),
716722
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
717723
}
718724
),
@@ -794,7 +800,7 @@ def navigate(
794800
{
795801
"x-language": str(x_language) if is_given(x_language) else not_given,
796802
"x-sdk-version": x_sdk_version,
797-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
803+
"x-sent-at": _format_x_sent_at(x_sent_at),
798804
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
799805
}
800806
),
@@ -996,7 +1002,7 @@ def observe(
9961002
{
9971003
"x-language": str(x_language) if is_given(x_language) else not_given,
9981004
"x-sdk-version": x_sdk_version,
999-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
1005+
"x-sent-at": _format_x_sent_at(x_sent_at),
10001006
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
10011007
}
10021008
),
@@ -1091,7 +1097,7 @@ def start(
10911097
{
10921098
"x-language": str(x_language) if is_given(x_language) else not_given,
10931099
"x-sdk-version": x_sdk_version,
1094-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
1100+
"x-sent-at": _format_x_sent_at(x_sent_at),
10951101
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
10961102
}
10971103
),
@@ -1322,7 +1328,7 @@ async def act(
13221328
{
13231329
"x-language": str(x_language) if is_given(x_language) else not_given,
13241330
"x-sdk-version": x_sdk_version,
1325-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
1331+
"x-sent-at": _format_x_sent_at(x_sent_at),
13261332
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
13271333
}
13281334
),
@@ -1353,7 +1359,6 @@ async def end(
13531359
self,
13541360
id: str,
13551361
*,
1356-
_force_body: object | Omit = omit,
13571362
x_language: Literal["typescript", "python", "playground"] | Omit = omit,
13581363
x_sdk_version: str | Omit = omit,
13591364
x_sent_at: Union[str, datetime] | Omit = omit,
@@ -1394,15 +1399,15 @@ async def end(
13941399
{
13951400
"x-language": str(x_language) if is_given(x_language) else not_given,
13961401
"x-sdk-version": x_sdk_version,
1397-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
1402+
"x-sent-at": _format_x_sent_at(x_sent_at),
13981403
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
13991404
}
14001405
),
14011406
**(extra_headers or {}),
14021407
}
14031408
return await self._post(
14041409
f"/v1/sessions/{id}/end",
1405-
body=await async_maybe_transform({"_force_body": _force_body}, session_end_params.SessionEndParams),
1410+
body={}, # Empty object to satisfy Content-Type requirement
14061411
options=make_request_options(
14071412
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
14081413
),
@@ -1580,7 +1585,7 @@ async def execute(
15801585
{
15811586
"x-language": str(x_language) if is_given(x_language) else not_given,
15821587
"x-sdk-version": x_sdk_version,
1583-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
1588+
"x-sent-at": _format_x_sent_at(x_sent_at),
15841589
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
15851590
}
15861591
),
@@ -1793,7 +1798,7 @@ async def extract(
17931798
{
17941799
"x-language": str(x_language) if is_given(x_language) else not_given,
17951800
"x-sdk-version": x_sdk_version,
1796-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
1801+
"x-sent-at": _format_x_sent_at(x_sent_at),
17971802
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
17981803
}
17991804
),
@@ -1875,7 +1880,7 @@ async def navigate(
18751880
{
18761881
"x-language": str(x_language) if is_given(x_language) else not_given,
18771882
"x-sdk-version": x_sdk_version,
1878-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
1883+
"x-sent-at": _format_x_sent_at(x_sent_at),
18791884
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
18801885
}
18811886
),
@@ -2077,7 +2082,7 @@ async def observe(
20772082
{
20782083
"x-language": str(x_language) if is_given(x_language) else not_given,
20792084
"x-sdk-version": x_sdk_version,
2080-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
2085+
"x-sent-at": _format_x_sent_at(x_sent_at),
20812086
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
20822087
}
20832088
),
@@ -2172,7 +2177,7 @@ async def start(
21722177
{
21732178
"x-language": str(x_language) if is_given(x_language) else not_given,
21742179
"x-sdk-version": x_sdk_version,
2175-
"x-sent-at": x_sent_at.isoformat() if is_given(x_sent_at) else not_given,
2180+
"x-sent-at": _format_x_sent_at(x_sent_at),
21762181
"x-stream-response": str(x_stream_response) if is_given(x_stream_response) else not_given,
21772182
}
21782183
),

0 commit comments

Comments
 (0)