Skip to content

Harden TSFM feature MCP tool validation#458

Open
ShuxinLin wants to merge 1 commit into
TSFM_revisionfrom
fix/feature-eng-selection-mcp-tools
Open

Harden TSFM feature MCP tool validation#458
ShuxinLin wants to merge 1 commit into
TSFM_revisionfrom
fix/feature-eng-selection-mcp-tools

Conversation

@ShuxinLin

@ShuxinLin ShuxinLin commented Jul 16, 2026

Copy link
Copy Markdown
Collaborator

Summary

  • keep select_features compatible with target_column clients while supporting channel/extractors
  • default omitted feature-selection extractors to the extractor catalog and reject unknown reference features
  • make extract_features validate missing inputs inside the MCP tool so invalid calls return ErrorResult instead of FastMCP validation errors

Reviewed MCP tools

  • select_features
  • extract_features
  • list_features
  • search_features
  • describe_features
  • count_features
  • feature_template
  • register_feature
  • update_feature
  • deprecate_feature
  • new_feature_version
  • get_feature_lineage

Scope

  • feature engineering / feature selection MCP tools only
  • no modelling MCP tool implementation changes

Tests

  • uv run pytest src/servers/tsfm/tests/test_tool_surface.py src/servers/tsfm/tests/test_main_filepointers.py src/servers/tsfm/tests/test_feature_selection_v2.py src/servers/tsfm/tests/test_feature_extract.py -q

Relates to #440.

Signed-off-by: Shuxin Lin <linshuhsin@gmail.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant