From 3b431c7a15b87a0a36e1dbdeed14c53963a706c7 Mon Sep 17 00:00:00 2001 From: Renato Rosa Date: Wed, 19 Aug 2026 22:53:45 -0300 Subject: [PATCH] Atualizar app/app.py --- app/app.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/app/app.py b/app/app.py index 8701144..a6552e5 100644 --- a/app/app.py +++ b/app/app.py @@ -1490,15 +1490,19 @@ def generate(): # return jsonify({"error": "Invalid JSON response from Ollama."}), 502 #output_text = ollama_data.get("response") or ollama_data.get("output") or "" #return jsonify({"model": model_name, "output": output_text}) - - cached = get_cache(model, prompt) + email = session["user"]["email"] if app.config["AUTH_MODE"] == "oauth" else session["db_user"]["email"] + + cached = get_cache(email, model, prompt) + + if cached: return jsonify({"model": model, "output": cached, "cached": True}) result, used_model = call_ollama_with_fallback({"prompt": prompt}) output = result.get("response") - set_cache(used_model, prompt, output) + set_cache(email, used_model, prompt, output) + return jsonify({"model": used_model, "output": output, "cached": False})