Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 11:42:29 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge 643fb5c4b1 into d05fee1182

Browse Source
This commit is contained in:
Benjamin 2024-11-15 11:51:22 +00:00 committed by GitHub
commit 1fde2c7d0a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 17 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
api/core/tools/provider/builtin/json_process/tools/parse.py
View File

@ -28,13 +28,17 @@ class JSONParseTool(BuiltinTool):
ensure_ascii = tool_parameters.get("ensure_ascii", True) ensure_ascii = tool_parameters.get("ensure_ascii", True)
try: try:
result = self._extract(content, json_filter, ensure_ascii) json_string, json_objs = self._extract(content, json_filter, ensure_ascii)
return self.create_text_message(str(result)) json_objs_dict = {str(index): item for index, item in enumerate(json_objs)}
except Exception: return [
return self.create_text_message("Failed to extract JSON content") self.create_text_message(str(json_string)),
self.create_json_message(json_objs_dict),
]
except Exception as e:
return self.create_text_message(f"Failed to extract JSON content: {str(e)}")
# Extract data from JSON content # Extract data from JSON content
def _extract(self, content: str, json_filter: str, ensure_ascii: bool) -> str: def _extract(self, content: str, json_filter: str, ensure_ascii: bool) -> tuple[str, list]:
try: try:
input_data = json.loads(content) input_data = json.loads(content)
expr = parse(json_filter) expr = parse(json_filter)
@ -44,10 +48,14 @@ class JSONParseTool(BuiltinTool):
result = result[0] result = result[0]
if isinstance(result, dict | list): if isinstance(result, dict | list):
return json.dumps(result, ensure_ascii=ensure_ascii) json_string = json.dumps(result, ensure_ascii=ensure_ascii)
elif isinstance(result, str | int | float | bool) or result is None: elif isinstance(result, str | int | float | bool) or result is None:
return str(result) json_string = str(result)
else: else:
return repr(result) json_string = repr(result)
return json_string, result
except json.JSONDecodeError as e:
raise ValueError(f"Invalid JSON content: {str(e)}")
except Exception as e: except Exception as e:
return str(e) raise ValueError(f"Error processing JSON content: {str(e)}")