first commit

files/modules/chat/routes.py

@@ -0,0 +1,82 @@
+import json
+from flask import Blueprint, request, jsonify
+from modules.core.security import requires_app_auth
+from oci_genai_llm_graphrag_rerank_rfp import answer_question, search_active_chunks
+from modules.core.audit import audit_log
+from .service import start_chat_job
+from .store import CHAT_JOBS, CHAT_LOCK
+
+chat_bp = Blueprint("chat", __name__)
+
+def parse_llm_json(raw: str) -> dict:
+    try:
+        if not isinstance(raw, str):
+            return {"answer": "ERROR", "justification": "LLM output is not a string", "raw_output": str(raw)}
+        raw = raw.replace("```json", "").replace("```", "").strip()
+        return json.loads(raw)
+    except Exception:
+        return {"answer": "ERROR", "justification": "LLM returned invalid JSON", "raw_output": raw}
+
+@chat_bp.route("/chat", methods=["POST"])
+@requires_app_auth
+def chat():
+    data = request.get_json(force=True) or {}
+    question = (data.get("question") or "").strip()
+
+    if not question:
+        return jsonify({"error": "Empty question"}), 400
+
+    raw_answer = answer_question(question)
+    parsed_answer = parse_llm_json(raw_answer)
+    audit_log("RFP_QUESTION", f"question={question}")
+
+    # (opcional) manter comportamento antigo de evidence/full_text se você quiser
+    # docs = search_active_chunks(question)
+
+    return jsonify({
+        "question": question,
+        "result": parsed_answer
+    })
+
+@chat_bp.post("/chat/start")
+def start():
+
+    question = request.json["question"]
+
+    job_id = start_chat_job(question)
+
+    return jsonify({"job_id": job_id})
+
+@chat_bp.get("/chat/<job_id>/status")
+def status(job_id):
+
+    with CHAT_LOCK:
+        job = CHAT_JOBS.get(job_id)
+
+    if not job:
+        return jsonify({"error": "not found"}), 404
+
+    return jsonify({"status": job["status"]})
+
+@chat_bp.get("/chat/<job_id>/result")
+def result(job_id):
+
+    with CHAT_LOCK:
+        job = CHAT_JOBS.get(job_id)
+
+    if not job:
+        return jsonify({"error": "not found"}), 404
+
+    return jsonify({
+        "result": parse_llm_json(job["result"]),
+        "error": job["error"]
+    })
+
+@chat_bp.get("/chat/<job_id>/logs")
+def logs(job_id):
+
+    with CHAT_LOCK:
+        job = CHAT_JOBS.get(job_id)
+
+    return jsonify({"logs": job["logs"]})
+