Atualizar app/app.py

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})