Skip to content

Commit 9588bf5

Browse files
committed
add try catch incase prgress token is not int
1 parent 26d04ac commit 9588bf5

1 file changed

Lines changed: 8 additions & 1 deletion

File tree

src/mcp/client/grpc_transport_session.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -165,7 +165,14 @@ async def call_tool(
165165
async for response in call:
166166
if response.common.HasField("progress"):
167167
progress_proto = response.common.progress
168-
progress_token = int(progress_proto.progress_token)
168+
progress = progress_proto.progress
169+
try:
170+
progress_token = int(progress_proto.progress_token)
171+
except ValueError:
172+
logger.warning(
173+
"Progress token is not an integer: %s",
174+
progress_proto.progress_token,
175+
)
169176

170177
if progress_token in self._progress_callbacks:
171178
callback = self._progress_callbacks[progress_token]

0 commit comments

Comments
 (0)