diff --git a/src/llama_stack_client/resources/agents/sessions.py b/src/llama_stack_client/resources/agents/sessions.py
index b8c6ae4e..8a1b3a85 100644
--- a/src/llama_stack_client/resources/agents/sessions.py
+++ b/src/llama_stack_client/resources/agents/sessions.py
@@ -120,19 +120,15 @@ def retrieve(
         }
         return self._post(
             "/agents/session/get",
-            body=maybe_transform({"turn_ids": turn_ids}, session_retrieve_params.SessionRetrieveParams),
+            body=maybe_transform({"agent_id": agent_id,
+                                  "session_id": session_id,
+                                  "turn_ids": turn_ids},
+                                 session_retrieve_params.SessionRetrieveParams),
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=maybe_transform(
-                    {
-                        "agent_id": agent_id,
-                        "session_id": session_id,
-                    },
-                    session_retrieve_params.SessionRetrieveParams,
-                ),
             ),
             cast_to=Session,
         )
@@ -273,19 +269,14 @@ async def retrieve(
         }
         return await self._post(
             "/agents/session/get",
-            body=await async_maybe_transform({"turn_ids": turn_ids}, session_retrieve_params.SessionRetrieveParams),
+            body=await async_maybe_transform({"agent_id": agent_id,
+                                              "session_id": session_id,
+                                              "turn_ids": turn_ids}, session_retrieve_params.SessionRetrieveParams),
             options=make_request_options(
                 extra_headers=extra_headers,
                 extra_query=extra_query,
                 extra_body=extra_body,
                 timeout=timeout,
-                query=await async_maybe_transform(
-                    {
-                        "agent_id": agent_id,
-                        "session_id": session_id,
-                    },
-                    session_retrieve_params.SessionRetrieveParams,
-                ),
             ),
             cast_to=Session,
         )