使用Flask部署TensorFlow模型时遇到请求上下文错误

嘿，这个坑我踩过！你遇到的RuntimeError: Working outside of request context本质是在没有活跃HTTP请求上下文的情况下调用了request对象——看你的回溯信息，是在app.py第61行的if request.args:触发的，说明这行代码根本没在Flask的视图函数里执行。

为什么会出现这个错误？ #

Flask的request对象是和每一个进来的HTTP请求绑定的，只有当服务器收到请求、调用对应的视图函数时，Flask才会自动创建请求上下文。如果直接在脚本顶层、或者在没有请求触发的代码块里调用request，它根本不知道要绑定哪个请求，自然就报错了。

具体怎么解决？ #

分两种常见场景给你方案：

场景1：这段代码是用来处理请求参数的（最常见）

把所有依赖request的逻辑移到视图函数内部，比如：

# 错误写法：在视图函数外调用request
if request.args:
    input = request.args.get('input')
    # 模型预测逻辑

@app.route('/predict')
def predict():
    return jsonify({'result': 'xxx'})

改成正确的写法：

@app.route('/predict')
def predict():
    # 把request相关逻辑放到视图函数里，这里才有活跃的请求上下文
    if request.args:
        input_data = request.args.get('input')
        # 调用你的TensorFlow模型做预测
        prediction = your_model.predict(input_data)
        return jsonify({'prediction': prediction.tolist()})
    else:
        return jsonify({'error': 'No input provided'}), 400

场景2：需要在非请求场景测试request相关代码

如果是想在应用启动时调试、或者写测试用例，可以手动创建一个测试请求上下文：

from flask import request

# 假设你的app是这样初始化的
app = Flask(__name__)

# 手动模拟请求上下文
with app.test_request_context('/predict?input=test_data'):
    # 这里可以安全使用request对象
    if request.args:
        print(request.args.get('input'))  # 会输出test_data

额外提醒 #

检查一下你的代码执行顺序：是不是把if request.args:写在了if __name__ == '__main__':外面？如果是，那应用一启动就会执行这行代码，这时候还没有任何请求进来，肯定会触发错误。把这类逻辑要么移到视图函数，要么放到测试上下文里。

内容的提问来源于stack exchange，提问作者Max S-T