Skip to content

Commit 9d306a9

Browse files
authored
chore(kagent-adk/session_sevice): add logging for Kagent API errors
Signed-off-by: Huang Huang <[email protected]>
1 parent 1634791 commit 9d306a9

File tree

1 file changed

+21
-4
lines changed

1 file changed

+21
-4
lines changed

python/packages/kagent-adk/src/kagent/adk/_session_service.py

Lines changed: 21 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,11 @@ async def create_session(
4747
json=request_data,
4848
headers={"X-User-ID": user_id},
4949
)
50-
response.raise_for_status()
50+
try:
51+
response.raise_for_status()
52+
except httpx.HTTPStatusError:
53+
logger.error(f"Failed to create session: {response.text}")
54+
raise
5155

5256
data = response.json()
5357
if not data.get("data"):
@@ -116,13 +120,18 @@ async def get_session(
116120
except httpx.HTTPStatusError as e:
117121
if e.response.status_code == 404:
118122
return None
123+
logger.error(f"Failed to get session: {response.text}")
119124
raise
120125

121126
@override
122127
async def list_sessions(self, *, app_name: str, user_id: str) -> ListSessionsResponse:
123128
# Make API call to list sessions
124129
response = await self.client.get(f"/api/sessions?user_id={user_id}", headers={"X-User-ID": user_id})
125-
response.raise_for_status()
130+
try:
131+
response.raise_for_status()
132+
except httpx.HTTPStatusError:
133+
logger.error(f"Failed to list sessions: {response.text}")
134+
raise
126135

127136
data = response.json()
128137
sessions_data = data.get("data", [])
@@ -145,7 +154,11 @@ async def delete_session(self, *, app_name: str, user_id: str, session_id: str)
145154
f"/api/sessions/{session_id}?user_id={user_id}",
146155
headers={"X-User-ID": user_id},
147156
)
148-
response.raise_for_status()
157+
try:
158+
response.raise_for_status()
159+
except httpx.HTTPStatusError:
160+
logger.error(f"Failed to delete session: {response.text}")
161+
raise
149162

150163
@override
151164
async def append_event(self, session: Session, event: Event) -> Event:
@@ -161,7 +174,11 @@ async def append_event(self, session: Session, event: Event) -> Event:
161174
json=event_data,
162175
headers={"X-User-ID": session.user_id},
163176
)
164-
response.raise_for_status()
177+
try:
178+
response.raise_for_status()
179+
except httpx.HTTPStatusError:
180+
logger.error(f"Failed to append event: {response.text}")
181+
raise
165182

166183
# TODO: potentially pull and update the session from the server
167184
# Update the in-memory session.

0 commit comments

Comments
 (0)