Spaces:
Running
Running
| import re | |
| import json | |
| import aiofiles | |
| from typing import Dict, Any | |
| def extract_json_from_text(text: str) -> Dict[str, Any]: | |
| """Extract JSON from text that might contain markdown or other content""" | |
| text = text.strip() | |
| # Try to parse as complete JSON | |
| if text.startswith("{") and text.endswith("}") or text.startswith("[") and text.endswith("]"): | |
| try: | |
| return json.loads(text) | |
| except json.JSONDecodeError: | |
| pass | |
| # Look for JSON within Markdown code blocks | |
| json_pattern = r"```(?:json)?\s*([\s\S]*?)\s*```" | |
| match = re.search(json_pattern, text) | |
| if match: | |
| try: | |
| return json.loads(match.group(1).strip()) | |
| except json.JSONDecodeError: | |
| pass | |
| # Try a more aggressive pattern | |
| json_pattern = r"\{[\s\S]*\}|\[[\s\S]*\]" | |
| match = re.search(json_pattern, text) | |
| if match: | |
| try: | |
| return json.loads(match.group(0)) | |
| except json.JSONDecodeError: | |
| pass | |
| raise ValueError("Could not extract valid JSON from the response") | |
| def read_file(file_path: str) -> str: | |
| document_text = None | |
| with open(file_path, "r", encoding="utf-8") as f: | |
| document_text = f.read() | |
| return document_text | |
| async def async_read_file(file_path: str) -> str: | |
| """Asynchronously read a file's contents. | |
| Args: | |
| file_path: Path to the file to read | |
| Returns: | |
| The file's contents as a string | |
| """ | |
| async with aiofiles.open(file_path, mode="r", encoding="utf-8") as f: | |
| return await f.read() | |