promptfoo/examples/google-live/counter_api.py

import flask

app = flask.Flask(__name__)

counter = 3


@app.route("/add_one", methods=["POST"])
def add_one():
    global counter
    counter += 1
    return flask.jsonify({})


@app.route("/get_count", methods=["GET"])
def get_count():
    return flask.jsonify({"counter": counter})


# If you want to assert on the final state the app must include this method
@app.route("/get_state", methods=["GET"])
def get_state():
    return flask.jsonify({"counter": counter})


# If you want the app to be cleaned up it must include this method
@app.route("/shutdown", methods=["POST"])
def shutdown():
    func = flask.request.environ.get("werkzeug.server.shutdown")
    if func is None:
        return flask.jsonify({"message": "Error: not running with the Werkzeug Server"})
    func()
    return flask.jsonify({"message": "Server shutting down..."})


if __name__ == "__main__":
    app.run(port=8765)