robust-json-parser 0.1.6

Robust JSON extraction and repair utilities for LLM-generated content.

🛠️ robust-json

Robust JSON extraction and repair utilities for LLM-generated content.

Parse JSON from messy LLM outputs with confidence. robust-json extracts and repairs JSON even when models mix commentary with structured data, use incorrect quotes, add trailing commas, include comments, or truncate responses mid-object.

---

✨ Why robust-json?

Large Language Models are powerful but inconsistent when generating JSON. They might:

• 📝 Mix text and JSON: Embed JSON inside markdown code blocks or conversational responses
• 💬 Add comments: Include // or # comments that break standard JSON parsers
• 🔤 Use wrong quotes: Generate single quotes (') instead of double quotes (")
• 🔚 Add trailing commas: Place commas after the last item in arrays/objects
• ✂️ Truncate output: Stop mid-JSON due to token limits or errors

robust-json handles all these cases automatically, so you can focus on using the data instead of fighting with parser errors.

---

🚀 Features

• 🔍 Smart extraction: Automatically finds JSON objects and arrays within free-form text
• 🔧 Auto-repair: Fixes common LLM errors including:
  • Single-quoted strings → double quotes
  • Mixed quote types (e.g., 'text" → 'text')
  • Inline comments (// and #)
  • Trailing commas
  • Unclosed braces and brackets
• 🎯 Multiple parsers: Falls back through json → ast.literal_eval for maximum compatibility
• ⚡ Performance: Optional speedups with regex (enhanced regex engine) and numba (JIT-compiled bracket scanning)
• 🌍 Unicode support: Handles international characters and emoji seamlessly

---

📦 Installation

Basic installation:

pip install robust-json-parser

With performance optimizations (numba JIT):

pip install robust-json-parser[speedups]

With regex (enhanced regex engine with better Unicode support):

pip install robust-json-parser[regex]

All extras:

pip install robust-json-parser[speedups,regex]

Requirements: Python 3.9+

---

🎯 Quick Start

Basic Usage

from robust_json import loads

# LLM output with mixed formatting
llm_response = """
Sure! Here's the data you requested:
```json
{
  "name": "Alice",
  "age": 30,
  "hobbies": ["reading", "coding",],  // trailing comma
  "active": true,  # Python-style comment
}

Hope this helps!
"""

data = loads(llm_response)
print(data)
# {'name': 'Alice', 'age': 30, 'hobbies': ['reading', 'coding'], 'active': True}

Handling Malformed JSON

from robust_json import loads

# Mixed quotes, comments, and multilingual text
message = """
Hello, I'm a recruitment consultant. Here's the job description for your matching assessment:
```json
{"id": "algo", "position": "Large Language Model Algorithm Engineer",
# this is the keywords list used to analyze the candidate
 "keywords": {"positive": ["PEFT", "RLHF"], "negative": ["CNN", "RNN"]}, # negative keywords is supported
 "summary": 'The candidate has some AI background, but lacks experience."
 }
"""

data = loads(message)
print(data["keywords"]["positive"])
# ['PEFT', 'RLHF']

Truncated/Partial JSON

from robust_json import loads

# JSON cut off mid-object
incomplete = '{"user": {"name": "Bob", "email": "[e-mail]"'

data = loads(incomplete)
print(data)
# {'user': {'name': 'Bob', 'email': '[e-mail]'}}

Extract Multiple JSON Objects

from robust_json import extract_all, RobustJSONParser

text = """
First result: {"a": 1, "b": 2}
Some text in between...
Second result: {"x": 10, "y": 20}
"""

# Get all extractions with metadata
extractions = extract_all(text)
for extraction in extractions:
    print(f"Found at position {extraction.start}: {extraction.text}")

# Or just get the parsed objects
parser = RobustJSONParser()
objects = parser.parse_all(text)
print(objects)
# [{'a': 1, 'b': 2}, {'x': 10, 'y': 20}]

---

📚 API Reference

loads(source, *, allow_partial=True, default=None, strict=False)

Parse the first JSON object found in the source text.

Parameters:

• source (str): Text containing JSON
• allow_partial (bool): If True, auto-complete truncated JSON (default: True)
• default (Optional): Return this value if no JSON found (default: None raises error)
• strict (bool): If True, only extract from code blocks and brace-delimited content (default: False)

Returns: Parsed Python object (dict, list, etc.)

Raises: ValueError if no JSON found and no default provided

---

extract(source, *, allow_partial=True)

Extract the first JSON-like fragment with metadata.

Returns: Extraction object or None

---

extract_all(source, *, allow_partial=True)

Extract all JSON-like fragments from text.

Returns: List of Extraction objects

---

RobustJSONParser

Main parser class for advanced usage.

Methods:

• extract(source, limit=None): Find JSON fragments (returns list of Extraction objects)
• parse_first(source): Parse first JSON object (returns parsed object or None)
• parse_all(source): Parse all JSON objects (returns list of parsed objects)

Parameters:

• allow_partial (bool): Auto-complete truncated JSON (default: True)
• strict (bool): Only extract from explicit JSON contexts (default: False)

---

Extraction

Dataclass representing an extracted JSON candidate.

Attributes:

• text (str): The extracted text
• start (int): Starting position in source
• end (int): Ending position in source
• is_partial (bool): Whether the extraction appears truncated
• repaired (Optional[str]): The repaired version after processing

---

🔧 How It Works

1. 🔎 Extraction: Scans text for JSON patterns using:

   • Markdown code blocks (```json ... ```)
   • Brace-balanced regions ({...}, [...])

2. 🛠️ Repair: Applies fixes in order:

   • Strip // and # comments
   • Fix mixed quote types (e.g., 'text" → 'text')
   • Normalize single quotes to double quotes
   • Remove trailing commas
   • Balance unclosed braces (if allow_partial=True)

3. ✅ Parse: Attempts parsing with:

   • json.loads() (standard JSON)
   • ast.literal_eval() (Python literals)

4. 📊 Return: Returns first successful parse or continues to next candidate

---

🎨 Use Cases

• 🤖 LLM Integration: Parse structured output from ChatGPT, Claude, Llama, etc.
• 📊 Data Extraction: Extract JSON from logs, documentation, or mixed-format files
• 🔄 API Responses: Handle malformed API responses gracefully
• 🧪 Testing: Validate and repair JSON in test fixtures
• 📝 Data Migration: Clean up inconsistent JSON during migrations

---

⚡ Performance Tips

1. Install speedups for large-scale processing:

   pip install robust-json-parser[speedups]  # numba JIT compilation
   pip install robust-json-parser[regex]  # enhanced regex engine with better Unicode support
   

2. Use strict mode when JSON is always in code blocks:

   loads(text, strict=True)  # Faster, skips fallback attempts
   

3. Disable partial completion if you know JSON is complete:

   loads(text, allow_partial=False)  # Skips brace-balancing step
   

4. Reuse parser instance for multiple parses:

   parser = RobustJSONParser()
   for text in texts:
       data = parser.parse_first(text)
   

---

🧪 Test Status

Overall Test Coverage: 97.9% (139/142 tests passing)

Category	Test File	Passed	Failed	Total	Pass Rate	Status
Core Functionality	test_parser.py	5	0	5	100.0%	✅
Comprehensive Tests	test_comprehensive.py	49	2	51	96.1%	✅
Edge Cases	test_edge_cases.py	38	1	39	97.4%	✅
LLM Scenarios	test_llm_scenarios.py	30	1	31	96.8%	✅
Performance	test_performance.py	11	0	11	100.0%	✅
Batch Processing	test_batch_performance.py	5	0	5	100.0%	✅

Test Categories Breakdown

• ✅ Core Functionality (100%): Basic parsing, extraction, and repair features
• ✅ Comprehensive Tests (96.1%): Real-world scenarios, complex nested structures, multilingual content
• ✅ Edge Cases (97.4%): Unicode handling, malformed JSON, bracket matching, error recovery
• ✅ LLM Scenarios (96.8%): ChatGPT/Claude-style outputs, conversational text extraction
• ✅ Performance (100%): Large datasets, memory usage, parsing speed benchmarks
• ✅ Batch Processing (100%): Parallel processing, multiprocessing, error handling

Known Issues (3 failing tests)

• Complex Incomplete JSON: Token-limited LLM outputs with deeply nested incomplete structures
• Extraction Order: extract_all function needs to preserve proper ordering
• Deep Nesting: Complex nested structures with mismatched brackets need enhanced repair

---

🤝 Contributing

We welcome contributions from developers of all skill levels! Whether you're fixing bugs, adding features, or improving documentation, your help makes this project better for everyone.

🎯 How to Contribute

1. 🐛 Bug Reports: Found an issue? Open a GitHub issue with:

   • Clear description of the problem
   • Minimal reproducible example
   • Expected vs actual behavior

2. ✨ Feature Requests: Have an idea? We'd love to hear it! Open an issue to discuss:

   • Use case and motivation
   • Proposed implementation approach
   • Any breaking changes

3. 🔧 Code Contributions: Ready to code? Here's how:

   # Fork and clone the repository
   git clone https://github.com/your-username/robust-json.git
   cd robust-json
   
   # Install in development mode
   pip install -e ".[speedups,regex,dev]"
   
   # Run tests to ensure everything works
   pytest tests/
   
   # Make your changes and test them
   pytest tests/ -v
   
   # Submit a pull request
   

🧪 Testing Your Changes

# Run all tests
pytest tests/

# Run specific test categories
pytest tests/test_parser.py          # Core functionality
pytest tests/test_comprehensive.py   # Comprehensive scenarios
pytest tests/test_llm_scenarios.py   # LLM-specific cases
pytest tests/test_edge_cases.py      # Edge cases and error handling
pytest tests/test_performance.py     # Performance benchmarks

# Run with coverage
pytest tests/ --cov=robust_json --cov-report=html

🎨 Areas We'd Love Help With

• 🌍 Internationalization: Better support for non-Latin scripts and RTL languages
• ⚡ Performance: Optimize parsing speed for very large JSON objects
• 🔍 LLM Integration: Improve extraction from more LLM output formats
• 📚 Documentation: Examples, tutorials, and API documentation
• 🧪 Test Coverage: Add more edge cases and real-world scenarios
• 🐛 Bug Fixes: Help us get to 100% test pass rate!

📋 Development Guidelines

• Code Style: Follow PEP 8, use type hints, and add docstrings
• Testing: Add tests for new features and bug fixes
• Documentation: Update README and docstrings as needed
• Performance: Consider performance impact of changes
• Compatibility: Maintain Python 3.9+ compatibility

🏆 Recognition

Contributors will be recognized in our README and release notes. We appreciate every contribution, no matter how small!

Ready to get started? Check out our open issues or start with the failing tests above!

---

📝 License

This project is licensed under the MIT License - see the LICENSE file for details.

---

🙏 Acknowledgments

Built for developers working with LLM-generated content who need reliability without sacrificing flexibility.

---

Made with ❤️ for the AI/LLM community