File size: 684 Bytes
a50063c 718aa48 a50063c 718aa48 a50063c 718aa48 a50063c 718aa48 a50063c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from flask import Blueprint, request, jsonify
from llm.agents import query_all_llms_sync
from llm.aggregator import aggregate_responses
chat_bp = Blueprint("chat", __name__)
@chat_bp.route("/chat", methods=["POST"])
def chat():
data = request.get_json()
user_input = data.get("prompt", "")
settings = data.get("settings", {})
if not user_input:
return jsonify({"error": "Empty prompt."}), 400
try:
agent_outputs = query_all_llms_sync(user_input, settings)
final_response = aggregate_responses(agent_outputs)
return jsonify({"response": final_response})
except Exception as e:
return jsonify({"error": str(e)}), 500
|