From 4ed3ed7e1445d382be11f97d96d21505d0d11822 Mon Sep 17 00:00:00 2001 From: lmiranda Date: Mon, 26 Jan 2026 17:25:37 -0500 Subject: [PATCH] fix(data-platform): reset index after filter to prevent extra column The filter tool was adding an __index_level_0__ column to results because pandas query() preserves the original index, which gets converted to a column when storing the DataFrame. Added .reset_index(drop=True) after query() to drop the preserved index and create a clean sequential index. Fixes #203 Co-Authored-By: Claude Opus 4.5 --- mcp-servers/data-platform/mcp_server/pandas_tools.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mcp-servers/data-platform/mcp_server/pandas_tools.py b/mcp-servers/data-platform/mcp_server/pandas_tools.py index e632c7e..f3d8af0 100644 --- a/mcp-servers/data-platform/mcp_server/pandas_tools.py +++ b/mcp-servers/data-platform/mcp_server/pandas_tools.py @@ -330,7 +330,7 @@ class PandasTools: return {'error': f'DataFrame not found: {data_ref}'} try: - filtered = df.query(condition) + filtered = df.query(condition).reset_index(drop=True) result_name = name or f"{data_ref}_filtered" return self._check_and_store( filtered, -- 2.49.1