plugging

Signed-off-by: rakdutta <[email protected]>

Author: rakdutta

mcpgateway/services/tool_service.py

@@ -1433,7 +1433,16 @@ async def connect_to_streamablehttp_server(server_url: str, headers: dict = head
                         # Reconstruct ToolResult from modified result
                         modified_result = post_result.modified_payload.result
                         if isinstance(modified_result, dict) and "content" in modified_result:
-                            tool_result = ToolResult(content=modified_result["content"], structured_content=modified_result["structuredContent"] if "structuredContent" in modified_result else modified_result["structured_content"])
+                            # Safely obtain structured content using .get() to avoid KeyError when
+                            # plugins provide only the content without structured content fields.
+                            structured = None
+                            if isinstance(modified_result, dict):
+                                structured = modified_result.get("structuredContent") if "structuredContent" in modified_result else modified_result.get("structured_content")
+
+                            tool_result = ToolResult(
+                                content=modified_result["content"],
+                                structured_content=structured
+                            )
                         else:
                             # If result is not in expected format, convert it to text content
                             tool_result = ToolResult(content=[TextContent(type="text", text=str(modified_result))])