|
|
# utils/json_utils.py
|
|
|
from config import *
|
|
|
from agent.llm_client import create_chat_completion
|
|
|
import re
|
|
|
import json
|
|
|
|
|
|
def find_json(text):
|
|
|
"""
|
|
|
Attempt to extract and parse a JSON object from the provided text.
|
|
|
The function tries up to three attempts using two patterns:
|
|
|
1. A Markdown code block with ```json ... ```
|
|
|
2. A more general JSON-like pattern using { and }
|
|
|
|
|
|
If successful, returns the parsed JSON data.
|
|
|
If parsing fails after all attempts, returns the original text.
|
|
|
|
|
|
Parameters:
|
|
|
text (str): The input text from which to extract JSON.
|
|
|
|
|
|
Returns:
|
|
|
dict or str: Parsed JSON dictionary if successful, else the original text.
|
|
|
"""
|
|
|
max_attempts = 3
|
|
|
for attempt in range(1, max_attempts + 1):
|
|
|
json_pattern = r"```json\n(.*?)\n```"
|
|
|
match = re.search(json_pattern, text, re.DOTALL)
|
|
|
if not match:
|
|
|
json_pattern2 = r"({.*?})"
|
|
|
match = re.search(json_pattern2, text, re.DOTALL)
|
|
|
|
|
|
if match:
|
|
|
json_string = match.group(1) if match.lastindex == 1 else match.group(0)
|
|
|
# Remove Markdown formatting if present
|
|
|
json_string = json_string.replace("```json\n", "").replace("\n```", "")
|
|
|
try:
|
|
|
data = json.loads(json_string)
|
|
|
return data
|
|
|
except json.JSONDecodeError as e:
|
|
|
if DEBUG_VER == 3:
|
|
|
print(f"--------------------find_json if match error text:{text}")
|
|
|
if attempt < max_attempts:
|
|
|
text = fix_json(json_string, e, model=MODEL)
|
|
|
if DEBUG_VER == 3:
|
|
|
print(f"Attempt {attempt}: Failed to parse JSON, reason: {e}. Retrying...")
|
|
|
else:
|
|
|
if DEBUG_VER == 3:
|
|
|
print(f"All {max_attempts} attempts to parse JSON failed. Returning original text.")
|
|
|
pass
|
|
|
else:
|
|
|
if DEBUG_VER == 3:
|
|
|
print(f"--------------------find_json else text:{text}")
|
|
|
if attempt < max_attempts:
|
|
|
if DEBUG_VER == 3:
|
|
|
print(f"Attempt {attempt}: No JSON string found in the text. Retrying...")
|
|
|
pass
|
|
|
else:
|
|
|
if DEBUG_VER == 3:
|
|
|
print("No matching JSON string found. Returning original text.")
|
|
|
pass
|
|
|
|
|
|
# If no match or no success in this attempt, return the original text
|
|
|
return text
|
|
|
|
|
|
|
|
|
def fix_json(text, json_error, model):
|
|
|
"""
|
|
|
修复JSON字符串，使其成为有效的JSON。
|
|
|
"""
|
|
|
NAIVE_FIX = f"""Instructions:
|
|
|
--------------
|
|
|
请修复JSON字符串，使其成为有效的JSON。
|
|
|
--------------
|
|
|
|
|
|
下面是原始的JSON字符串：
|
|
|
--------------
|
|
|
{text}
|
|
|
--------------
|
|
|
下面是的错误信息：
|
|
|
--------------
|
|
|
{json_error}
|
|
|
--------------
|
|
|
|
|
|
请仅回复json，用```json ... ```包裹json字符串："""
|
|
|
|
|
|
messages = [{"role": "user", "content": NAIVE_FIX}]
|
|
|
response = create_chat_completion(messages, model)
|
|
|
answer = response.choices[0].message.content
|
|
|
return answer
|
|
|
|
|
|
def dict_to_sentence(data):
|
|
|
"""
|
|
|
Convert a dictionary into a descriptive sentence by enumerating key-value pairs.
|
|
|
For example: {"name": "John", "age": 30} -> "name 是 John, age 是 30"
|
|
|
|
|
|
Parameters:
|
|
|
data (dict): The dictionary to convert.
|
|
|
|
|
|
Returns:
|
|
|
str: A sentence describing the dictionary keys and values.
|
|
|
"""
|
|
|
try:
|
|
|
if not isinstance(data, dict):
|
|
|
raise ValueError("Input is not a dictionary")
|
|
|
|
|
|
return ", ".join(f"{key} 是 {value}" for key, value in data.items())
|
|
|
except Exception as e:
|
|
|
if DEBUG_VER == 3:
|
|
|
print(f"Error in dict_to_sentence: {e}")
|
|
|
return str(data) |
