使用Python写一个适用于Dify和FastGPT的JsonPath插件
编写适用于 Dify 和 FastGPT 的 JsonPath 插件
在本文中，我将分享如何编写一个适用于 AI应用平台的 JsonPath 插件，该插件能够处理 JSON 数据的路径查询、正则表达式提取，以及 JavaScript 沙盒执行功能。这个插件的主要目的是让用户能够通过简单的 HTTP 请求，对 JSON 数据进行高效、灵活的查询和处理。接下来，我们将详细介绍如何实现这个插件的功能。
项目结构与依赖
首先，我们需要创建一个 Flask 应用程序来承载这个插件。项目的主要依赖项包括 Flask、jsonpath_ng、re、json、js2py 和 threading。这些库分别用于处理 HTTP 请求、JSONPath 查询、正则表达式匹配、JavaScript 代码执行和处理并发任务。
from flask import Flask, request, jsonify
from jsonpath_ng import parse
import re
import json
import js2py
import threading
import logging
为了便于调试和跟踪，我们配置了日志记录功能：
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
JSONPath 查询
首先，我们实现了一个 JSONPath 查询功能，通过 POST 请求获取用户提交的 JSONPath 表达式和 JSON 数据，并返回匹配的结果。
@app.route('/jsonpath', methods=['POST'])
def get_value_by_jsonpath():json_path = request.json.get('json_path')json_string = request.json.get('json_string')data = json.loads(json_string)jsonpath_expr = parse(json_path)matches = [match.value for match in jsonpath_expr.find(data)]if '*' in json_path or ('[' in json_path and ':' in json_path):return jsonify({"data": matches}) # 预期返回列表else:if len(matches) == 0:return jsonify({"data": None})else:return jsonify({"data": matches[0]}) if matches else None # 预期返回单个元素
正则表达式提取
接下来，我们实现了一个正则表达式提取功能，用户可以通过提交正则表达式和目标字符串来提取匹配的内容。
@app.route('/regex_extract', methods=['POST'])
def regex_extract():regex = request.json.get('regex')string = request.json.get('string')if not regex or not string:return jsonify({"error": "regex或string参数不正确"}), 400result = extract_target(regex, string)logging.info('regex: %s, string: %s, result: %s', regex, string, result)return jsonify({"data": result, "match": True if result else False})
JavaScript 沙盒执行
最后，我们实现了一个 JavaScript 沙盒执行功能，允许用户提交 JavaScript 代码，并在一个安全的环境中执行该代码。我们使用 js2py 库来执行 JavaScript，并设置了超时机制以防止脚本长时间运行。
@app.route('/js_sandbox', methods=['POST'])
def js_sandbox():code = request.json.get('code')if not code:return jsonify({"error": "code参数不正确"}), 400code = fix_string_conflicts(code)timer = threading.Timer(3.0, timeout)timer.start()result = Nonetry:result = js2py.eval_js(code)print(result)except Exception as e:logging.error("Error while executing JavaScript: %s", e)finally:timer.cancel()logging.info('code: %s, result: %s', code, result)return jsonify({"data": result})
总结
通过上述步骤，我们实现了一个简单但功能强大的 JsonPath 插件，适用于 Dify 和 FastGPT。它可以处理 JSONPath 查询、正则表达式提取以及 JavaScript 沙盒执行等任务。希望这篇文章能帮助你更好地理解如何构建类似的插件，并将其应用到你的项目中。如果你有任何问题或建议，欢迎在评论区留言讨论。