Flexible Key Value Extracting Mcp Server

by KunihiroS

316 downloads
Not rated
GitHub

Description

# Flexible Key-Value Extracting MCP Server Version: 0.1.0 This MCP server extracts key-value pairs from arbitrary, noisy, or unstructured text using LLMs (GPT-4.1-mini) and pydantic-ai. It ensures type safety and supports multiple output formats (JSON, YAML, TOML). The server is…

About

# Flexible Key-Value Extracting MCP Server Version: 0.1.0 This MCP server extracts key-value pairs from arbitrary, noisy, or unstructured text using LLMs (GPT-4.1-mini) and pydantic-ai. It ensures type safety and supports multiple output formats (JSON, YAML, TOML). The server is robust to any input and always attempts…

Details

Author
KunihiroS
Downloads
316
Categories
Other

- Extracts key-value pairs from noisy or broken input text.
- Supports Japanese, English, and Chinese (automatic language detection).
- Uses type-safe Pydantic output validation.
- Returns results in JSON, YAML, or TOML format.
- Always returns a well-formed response, even on extraction failure.
- Employs GPT-4.1-mini for extraction, type annotation, and type evaluation.

Setting up with Highlight

This MCP is not yet compatible with Highlight’s one-click setup. However, you can still use it with Highlight by following these steps:

  1. Download and install Highlight from highlightai.com/download
  2. Navigate to the plugins tab and select "Add Custom Plugin"
  3. Configure the plugin with the settings below
    Plugin Name Flexible Key Value Extracting Mcp Server
    Command (node, npx, python, etc.)

    Please refer to the README for specific instructions on how to obtain API keys or other required environment variables.

  4. Enable "Start Automatically" if you want the plugin to start when Highlight launches

From the repository

Requires Python 3.9+ and an OpenAI API key set in settings.json. Run python server.py. The server exposes three tools: extract_json, extract_yaml, and extract_toml. When configuring the MCP host you must specify log mode via --log=off or --log=on --logfile=<absolute path>.

Claude Desktop / Cursor

Paste into your MCP client config file to install this server.

{
    "mcpServers": {
        "flexible key value extracting mcp server": {
            "kv-extractor-mcp-server": {
                "command": "pipx",
                "args": [
                    "run",
                    "kv-extractor-mcp-server",
                    "--log=off"
                ],
                "env": {
                    "OPENAI_API_KEY": "<your-api-key>"
                }
            }
        }
    }
}

McpServers

{
    "kv-extractor-mcp-server": {
        "command": "pipx",
        "args": [
            "run",
            "kv-extractor-mcp-server",
            "--log=off"
        ],
        "env": {
            "OPENAI_API_KEY": "<your-api-key>"
        }
    }
}

Flexible Key-Value Extracting MCP Server

Version: 0.1.0

This MCP server extracts key-value pairs from arbitrary, noisy, or unstructured text using LLMs (GPT-4.1-mini) and pydantic-ai.
It ensures type safety and supports multiple output formats (JSON, YAML, TOML). The server is robust to any input and always attempts to structure data as much as possible, however, perfect extraction is not guaranteed.

---

Release Notes

v0.1.0

- Initial release

Tools

- /extract_json : Extracts type-safe key-value pairs in JSON format from input text.
- /extract_yaml : Extracts type-safe key-value pairs in YAML format from input text.
- /extract_toml : Extracts type-safe key-value pairs in TOML format from input text.
- Note: Due to TOML specifications, arrays of objects (dicts) or deeply nested structures cannot be directly represented. See "Note on TOML Output Limitations" below for details.

Note:
- Supported languages: Japanese, English, and Chinese (Simplified/Traditional).
- Extraction relies on pydantic-ai and LLMs. Perfect extraction is not guaranteed.
- Longer input sentences will take more time to process. Please be patient.
- On first launch, the server will download spaCy models, so the process will take longer initially.

Estimated Processing Time Sample

| Input Tokens | Input Characters (approx.) | Measured Processing Time (sec) | Model Configuration |
|:-----------:|:--------------------------:|:------------------------------:|:-------------------|
| 200 | ~400 | ~15 | gpt-4.1-mini |

Actual processing time may vary significantly depending on API response, network conditions, and model load. Even short texts may take 15 seconds or more.

Features

- Flexible extraction: Handles any input, including noisy or broken data. - JP / EN / ZH full support: Preprocessing with spaCy NER by automatic language detection (Japanese, English, Chinese [Simplified/Traditional] supported; others are rejected with error). - Type-safe output: Uses Pydantic for output validation. - Multiple formats: Returns results as JSON, YAML, or TOML. - Robust error handling: Always returns a well-formed response, even on failure. - High accuracy: Uses GPT-4.1-mini for both extraction/annotation and type evaluation, with Pydantic for final structuring.

Tested Scenarios

The server has been tested with various inputs, including: - Simple key-value pairs - Noisy or unstructured text with important information buried within - Different data formats (JSON, YAML, TOML) for output

Processing Flow

Below is a flowchart representing the processing flow of the key-value extraction pipeline as implemented in server.py:
flowchart TD
    A[Input Text] --> B[Step 0: Preprocessing with spaCy Lang Detect then NER]
    B --> C[Step 1: Key-Value Extraction - LLM]
    C --> D[Step 2: Type Annotation - LLM]
    D --> E[Step 3: Type Evaluation - LLM]
    E --> F[Step 4: Type Normalization - Static Rules + LLM]
    F --> G[Step 5: Final Structuring with Pydantic]
    G --> H[Output in JSON/YAML/TOML]

Preprocessing with spaCy (Multilingual NER)

This server uses spaCy with automatic language detection to extract named entities from the input text before passing it to the LLM. Supported languages are Japanese (ja_core_news_md), English (en_core_web_sm), and Chinese (Simplified/Traditional, zh_core_web_sm).

- The language of the input text is automatically detected using langdetect.
- If the detected language is not Japanese, English, or Chinese, the server returns an error: Unsupported lang detected.
- The appropriate spaCy model is automatically downloaded and loaded as needed. No manual installation is required.
- The extracted phrase list is included in the LLM prompt as follows:

> [Preprocessing Candidate Phrases (spaCy NER)]
> The following is a list of phrases automatically extracted from the input text using spaCy's detected language model.
> These phrases represent detected entities such as names, dates, organizations, locations, numbers, etc.
> This list is for reference only and may contain irrelevant or incorrect items. The LLM uses its own judgment and considers the entire input text to flexibly infer the most appropriate key-value pairs.

Step Details

This project's key-value extraction pipeline consists of multiple steps. Each step's details are as follows:

Step 0: Preprocessing with spaCy (Language Detection → Named Entity Recognition)

- Purpose: Automatically detect the language of the input text and use the appropriate spaCy model (e.g., ja_core_news_md, en_core_web_sm, zh_core_web_sm) to extract named entities. - Output: The extracted phrase list, which is included in the LLM prompt as a hint to improve key-value pair extraction accuracy.

Step 1: Key-Value Extraction (LLM)

- Purpose: Use GPT-4.1-mini to extract key-value pairs from the input text and the extracted phrase list. - Details: - The prompt includes instructions to return list-formatted values when the same key appears multiple times. - Few-shot examples are designed to include list-formatted outputs. - Output: Example: key: person, value: ["Tanaka", "Sato"]

Step 2: Type Annotation (LLM)

- Purpose: Use GPT-4.1-mini to infer the data type (int, str, bool, list, etc.) of each key-value pair extracted in Step 1. - Details: - The type annotation prompt includes instructions for list and multiple value support. - Output: Example: key: person, value: ["Tanaka", "Sato"] -> list[str]

Step 3: Type Evaluation (LLM)

- Purpose: Use GPT-4.1-mini to evaluate and correct the type annotations from Step 2. - Details: - For each key-value pair, GPT-4.1-mini re-evaluates the type annotation's validity and context. - If type errors or ambiguities are detected, GPT-4.1-mini automatically corrects or supplements the type. - Example: Correcting a value extracted as a number but should be a string, or determining whether a value is a list or a single value. - Output: The type-evaluated key-value pair list.

Step 4: Type Normalization (Static Rules + LLM Fallback)

- Purpose: Convert the type-evaluated data into Python's standard types (int, float, bool, str, list, None, etc.). - Details: - Apply static normalization rules (regular expressions or type conversion functions) to convert values into Python's standard types. - Example: Converting comma-separated values to lists, "true"/"false" to bool, or date expressions to standard formats. - If static rules cannot convert a value, use LLM-based type conversion fallback. - Unconvertible values are safely handled as None or str. - Output: The Python-type-normalized key-value pair list.

Step 5: Final Structuring with Pydantic

- Purpose: Validate and structure the type-normalized data using Pydantic models (KVOut/KVPayload). - Details: - Map each key-value pair to Pydantic models, ensuring type safety and data integrity. - Validate single values, lists, null, and composite types according to the schema. - If validation fails, attach error information while preserving as much data as possible. - The final output is returned in the specified format (JSON, YAML, or TOML). - Output: The type-safe and validated dict or specified format (JSON/YAML/TOML) output.

---

This pipeline is designed to accommodate future list format support and Pydantic schema extensions.

Note on TOML Output Limitations

No reviews yet — be the first

Sign in to leave a review

Use Google, GitHub, or an email account so ratings stay tied to real people.

Email sign in

No reviews posted yet.