fix(api): remove agent_id and task_id parameters from states update method

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 64
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp/agentex-sdk-550d6a4f6985887b66c29240e495469cc0a74e2c966175364321ae2eb15595d8.yml
-openapi_spec_hash: a93e0b4ee276d29ebd1c2f2a4c798f36
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp/agentex-sdk-e48cc9d460e306acf0f98efad4b5cccb793af107d0aaf50af455954d4d943e2d.yml
+openapi_spec_hash: 2d85e7026b3f21e81c533d51486c0182
 config_hash: 82cb83ac175dbf40265128506294218b

src/agentex/resources/states.py

@@ -122,9 +122,7 @@ def update(
         self,
         state_id: str,
         *,
-        agent_id: str,
         state: Dict[str, object],
-        task_id: str,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -148,14 +146,7 @@ def update(
             raise ValueError(f"Expected a non-empty value for `state_id` but received {state_id!r}")
         return self._put(
             path_template("/states/{state_id}", state_id=state_id),
-            body=maybe_transform(
-                {
-                    "agent_id": agent_id,
-                    "state": state,
-                    "task_id": task_id,
-                },
-                state_update_params.StateUpdateParams,
-            ),
+            body=maybe_transform({"state": state}, state_update_params.StateUpdateParams),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
@@ -356,9 +347,7 @@ async def update(
         self,
         state_id: str,
         *,
-        agent_id: str,
         state: Dict[str, object],
-        task_id: str,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -382,14 +371,7 @@ async def update(
             raise ValueError(f"Expected a non-empty value for `state_id` but received {state_id!r}")
         return await self._put(
             path_template("/states/{state_id}", state_id=state_id),
-            body=await async_maybe_transform(
-                {
-                    "agent_id": agent_id,
-                    "state": state,
-                    "task_id": task_id,
-                },
-                state_update_params.StateUpdateParams,
-            ),
+            body=await async_maybe_transform({"state": state}, state_update_params.StateUpdateParams),
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),

src/agentex/types/state_update_params.py

@@ -9,8 +9,4 @@
 
 
 class StateUpdateParams(TypedDict, total=False):
-    agent_id: Required[str]
-
     state: Required[Dict[str, object]]
-
-    task_id: Required[str]

tests/api_resources/test_states.py

@@ -105,9 +105,7 @@ def test_path_params_retrieve(self, client: Agentex) -> None:
     def test_method_update(self, client: Agentex) -> None:
         state = client.states.update(
             state_id="state_id",
-            agent_id="agent_id",
             state={"foo": "bar"},
-            task_id="task_id",
         )
         assert_matches_type(State, state, path=["response"])
 
@@ -116,9 +114,7 @@ def test_method_update(self, client: Agentex) -> None:
     def test_raw_response_update(self, client: Agentex) -> None:
         response = client.states.with_raw_response.update(
             state_id="state_id",
-            agent_id="agent_id",
             state={"foo": "bar"},
-            task_id="task_id",
         )
 
         assert response.is_closed is True
@@ -131,9 +127,7 @@ def test_raw_response_update(self, client: Agentex) -> None:
     def test_streaming_response_update(self, client: Agentex) -> None:
         with client.states.with_streaming_response.update(
             state_id="state_id",
-            agent_id="agent_id",
             state={"foo": "bar"},
-            task_id="task_id",
         ) as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -149,9 +143,7 @@ def test_path_params_update(self, client: Agentex) -> None:
         with pytest.raises(ValueError, match=r"Expected a non-empty value for `state_id` but received ''"):
             client.states.with_raw_response.update(
                 state_id="",
-                agent_id="agent_id",
                 state={"foo": "bar"},
-                task_id="task_id",
             )
 
     @pytest.mark.skip(reason="Mock server tests are disabled")
@@ -330,9 +322,7 @@ async def test_path_params_retrieve(self, async_client: AsyncAgentex) -> None:
     async def test_method_update(self, async_client: AsyncAgentex) -> None:
         state = await async_client.states.update(
             state_id="state_id",
-            agent_id="agent_id",
             state={"foo": "bar"},
-            task_id="task_id",
         )
         assert_matches_type(State, state, path=["response"])
 
@@ -341,9 +331,7 @@ async def test_method_update(self, async_client: AsyncAgentex) -> None:
     async def test_raw_response_update(self, async_client: AsyncAgentex) -> None:
         response = await async_client.states.with_raw_response.update(
             state_id="state_id",
-            agent_id="agent_id",
             state={"foo": "bar"},
-            task_id="task_id",
         )
 
         assert response.is_closed is True
@@ -356,9 +344,7 @@ async def test_raw_response_update(self, async_client: AsyncAgentex) -> None:
     async def test_streaming_response_update(self, async_client: AsyncAgentex) -> None:
         async with async_client.states.with_streaming_response.update(
             state_id="state_id",
-            agent_id="agent_id",
             state={"foo": "bar"},
-            task_id="task_id",
         ) as response:
             assert not response.is_closed
             assert response.http_request.headers.get("X-Stainless-Lang") == "python"
@@ -374,9 +360,7 @@ async def test_path_params_update(self, async_client: AsyncAgentex) -> None:
         with pytest.raises(ValueError, match=r"Expected a non-empty value for `state_id` but received ''"):
             await async_client.states.with_raw_response.update(
                 state_id="",
-                agent_id="agent_id",
                 state={"foo": "bar"},
-                task_id="task_id",
             )
 
     @pytest.mark.skip(reason="Mock server tests are disabled")