Fixed the MCP error handling for large pulls.

src/fgai/graylog_source.py

@@ -58,13 +58,19 @@ class GraylogStreamSource:
         events: list[LogEvent] = []
         page_size = 1_000
         pages = 0
+        partial_error = ""
         while len(events) < max_events:
             request_limit = min(page_size, max_events - len(events))
-            result = self.client.call_tool("search_messages", {**arguments, "limit": request_limit, "offset": len(events)})
+            try:
+                result = self.client.call_tool("search_messages", {**arguments, "limit": request_limit, "offset": len(events)})
+            except RuntimeError as exc:
+                partial_error = str(exc)
+                break
             content = result.get("result", {}).get("content", []) if isinstance(result.get("result"), dict) else []
             if isinstance(result.get("result"), dict) and result["result"].get("isError"):
                 detail = next((str(item.get("text")) for item in content if isinstance(item, dict) and item.get("type") == "text"), "Graylog search failed")
-                raise RuntimeError(f"graylog_search_error: {detail}")
+                partial_error = f"graylog_search_error: {detail}"
+                break
             records: list[dict[str, object]] = []
             for item in content if isinstance(content, list) else []:
                 if isinstance(item, dict) and item.get("type") == "text":
@@ -77,7 +83,15 @@ class GraylogStreamSource:
             if len(records) < request_limit:
                 break
         latest = max((event.fields.get("eventtime", "") for event in events), default="")
-        status.update({"source": "graylog_mcp", "events_fetched": len(events), "pages": pages, "truncated": len(events) >= max_events, "latest_event_time": latest})
+        status.update({
+            "source": "graylog_mcp",
+            "events_fetched": len(events),
+            "pages": pages,
+            "partial": bool(partial_error),
+            "error": partial_error,
+            "truncated": len(events) >= max_events,
+            "latest_event_time": latest,
+        })
         return events, status
 
     def _event(self, record: dict[str, object]) -> LogEvent: