Reimplement mcp server using mcp instead of fastmcp. Provide action payload schema as tool schema and description from docstrings.

Author: Aksem

docs/wm-er-protocol.md

@@ -96,6 +96,14 @@ The protocol is LSP-shaped with a small set of custom commands.
     - Note: `result_by_format` is a JSON string (not a JSON object) due to
       LSP serialization constraints in the runner.
 
+  - `actions/getPayloadSchemas`
+    - Arguments: none
+    - Result: `{ action_name: JSON Schema fragment | null }`
+    - Returns a payload schema for every action currently known to the runner.
+      Each schema has `properties` (field name → JSON Schema type object) and
+      `required` (list of field names without defaults).
+      `null` means the action class could not be imported.
+
   - `actions/mergeResults`
     - Arguments: `[action_name, results]`
     - Result: `{ "merged": ... }` or `{ "error": "..." }`

docs/wm-protocol.md

@@ -310,6 +310,38 @@ All fields optional. If `project` is omitted, returns actions from all projects.
 
 ---
 
+#### `actions/getPayloadSchemas`
+
+Return payload schemas for the specified actions in a project. Used by the MCP
+server to build accurate `inputSchema` entries for each tool.
+
+- **Type:** request
+- **Clients:** MCP
+- **Status:** implemented
+
+**Params:**
+
+```json
+{ "project": "/abs/path/to/project", "action_names": ["lint", "format"] }
+```
+
+**Result:**
+
+```json
+{
+  "schemas": {
+    "lint":   { "properties": { "file_paths": {"type": "array", "items": {"type": "string"}}, "target": {"type": "string", "enum": ["project", "files"]} }, "required": [] },
+    "format": { "properties": { "save": {"type": "boolean"}, "target": {"type": "string"}, "file_paths": {"type": "array", "items": {"type": "string"}} }, "required": [] }
+  }
+}
+```
+
+Each schema value is `null` for actions whose class cannot be imported in any
+Extension Runner. Schemas are cached per project in the WM and invalidated
+whenever runner config is updated.
+
+---
+
 #### `actions/getTree`
 
 Get the hierarchical action tree for IDE sidebar display.

finecode_extension_api/src/finecode_extension_api/actions/clean_finecode_logs.py

@@ -40,6 +40,8 @@ class CleanFinecodeLogsAction(
         CleanFinecodeLogsRunResult,
     ]
 ):
+    """Remove FineCode log files."""
+
     PAYLOAD_TYPE = CleanFinecodeLogsRunPayload
     RUN_CONTEXT_TYPE = CleanFinecodeLogsRunContext
     RESULT_TYPE = CleanFinecodeLogsRunResult

finecode_extension_api/src/finecode_extension_api/actions/create_env.py

@@ -19,6 +19,8 @@ class CreateEnvAction(
         CreateEnvRunPayload, CreateEnvRunContext, CreateEnvsRunResult
     ]
 ):
+    """Create a single environment(without installing dependencies, only environment)."""
+
     PAYLOAD_TYPE = CreateEnvRunPayload
     RUN_CONTEXT_TYPE = CreateEnvRunContext
     RESULT_TYPE = CreateEnvsRunResult

finecode_extension_api/src/finecode_extension_api/actions/create_envs.py

@@ -19,11 +19,10 @@ class EnvInfo:
 
 @dataclasses.dataclass
 class CreateEnvsRunPayload(code_action.RunActionPayload):
-    # Explicit env list. When empty, handlers should discover envs.
     envs: list[EnvInfo] = dataclasses.field(default_factory=list)
-    # Remove old env and create a new one from scratch even if the current one
-    # is valid.
+    """Explicit list of environments to create. Empty means handlers discover envs."""
     recreate: bool = False
+    """Remove and recreate existing environments from scratch even if they are already valid."""
 
 
 class CreateEnvsRunContext(code_action.RunActionContext[CreateEnvsRunPayload]):
@@ -74,6 +73,8 @@ class CreateEnvsAction(
         CreateEnvsRunPayload, CreateEnvsRunContext, CreateEnvsRunResult
     ]
 ):
+    """Create environments for the workspace(without installing dependencies, only environment)."""
+
     PAYLOAD_TYPE = CreateEnvsRunPayload
     RUN_CONTEXT_TYPE = CreateEnvsRunContext
     RESULT_TYPE = CreateEnvsRunResult

finecode_extension_api/src/finecode_extension_api/actions/dump_config.py

@@ -63,6 +63,8 @@ def to_text(self) -> str | textstyler.StyledText:
 class DumpConfigAction(
     code_action.Action[DumpConfigRunPayload, DumpConfigRunContext, DumpConfigRunResult]
 ):
+    """Resolve and dump the merged project configuration."""
+
     PAYLOAD_TYPE = DumpConfigRunPayload
     RUN_CONTEXT_TYPE = DumpConfigRunContext
     RESULT_TYPE = DumpConfigRunResult

finecode_extension_api/src/finecode_extension_api/actions/format.py

@@ -15,9 +15,11 @@ class FormatTarget(enum.StrEnum):
 @dataclasses.dataclass
 class FormatRunPayload(code_action.RunActionPayload):
     save: bool = True
+    """Whether to write formatted content back to disk."""
     target: FormatTarget = FormatTarget.PROJECT
-    # optional, expected only with `target == FormatTarget.FILES`
+    """Scope of formatting: 'project' (default) formats the whole project, 'files' formats only file_paths."""
     file_paths: list[Path] = dataclasses.field(default_factory=list)
+    """Files to format. Only used when target is 'files'."""
 
 
 class FormatRunContext(code_action.RunActionContext[FormatRunPayload]): ...
@@ -30,6 +32,8 @@ class FormatRunResult(format_files_action.FormatFilesRunResult): ...
 class FormatAction(
     code_action.Action[FormatRunPayload, FormatRunContext, FormatRunResult]
 ):
+    """Format source code in a project or specific files."""
+
     PAYLOAD_TYPE = FormatRunPayload
     RUN_CONTEXT_TYPE = FormatRunContext
     RESULT_TYPE = FormatRunResult

finecode_extension_api/src/finecode_extension_api/actions/format_files.py

@@ -108,6 +108,8 @@ class FormatFilesAction(
         FormatFilesRunPayload, FormatFilesRunContext, FormatFilesRunResult
     ]
 ):
+    """Format specific files. Internal action dispatched by format."""
+
     PAYLOAD_TYPE = FormatFilesRunPayload
     RUN_CONTEXT_TYPE = FormatFilesRunContext
     RESULT_TYPE = FormatFilesRunResult

finecode_extension_api/src/finecode_extension_api/actions/get_dist_artifact_version.py

@@ -40,6 +40,8 @@ class GetDistArtifactVersionAction(
         GetDistArtifactVersionRunResult,
     ]
 ):
+    """Read the version from a distribution artifact."""
+
     PAYLOAD_TYPE = GetDistArtifactVersionRunPayload
     RUN_CONTEXT_TYPE = GetDistArtifactVersionRunContext
     RESULT_TYPE = GetDistArtifactVersionRunResult

finecode_extension_api/src/finecode_extension_api/actions/get_src_artifact_registries.py

@@ -52,6 +52,8 @@ class GetSrcArtifactRegistriesAction(
         GetSrcArtifactRegistriesRunResult,
     ]
 ):
+    """List the registries configured for an artifact."""
+
     PAYLOAD_TYPE = GetSrcArtifactRegistriesRunPayload
     RUN_CONTEXT_TYPE = GetSrcArtifactRegistriesRunContext
     RESULT_TYPE = GetSrcArtifactRegistriesRunResult