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