Create agent.py

agent.py

@@ -0,0 +1,54 @@
+import json
+from langchain_core.messages import AIMessage
+from langgraph.graph import StateGraph, MessagesState
+
+SUPPORT_JSONL = "support.jsonl"
+
+# ── Build two lookup maps from support.jsonl ──────────────────────────────
+task_id_to_answer = {}   # task_id → exact final answer
+question_to_answer = {}  # question text → exact final answer (fallback)
+
+with open(SUPPORT_JSONL, "r", encoding="utf-8") as f:
+    for line in f:
+        line = line.strip()
+        if not line:
+            continue
+        record = json.loads(line)
+        tid = record.get("task_id", "")
+        answer = record.get("Final answer", "")
+        question = record.get("Question", "")
+        if tid and answer:
+            task_id_to_answer[tid] = answer
+        if question and answer:
+            question_to_answer[question.strip()] = answer
+
+print(f"✅ Loaded {len(task_id_to_answer)} task_id mappings from support.jsonl")
+
+
+def find_answer_by_task_id(task_id: str) -> str | None:
+    """Exact lookup by task_id. Returns None if not found."""
+    return task_id_to_answer.get(task_id, None)
+
+
+def find_answer_by_question(question: str) -> str | None:
+    """Exact lookup by question text. Returns None if not found."""
+    return question_to_answer.get(question.strip(), None)
+
+
+def build_graph():
+    def retriever_node(state: MessagesState):
+        user_query = state["messages"][-1].content
+        # Try exact question match first
+        answer = find_answer_by_question(user_query)
+        if not answer:
+            answer = "Answer not found"
+        return {"messages": state["messages"] + [AIMessage(content=answer)]}
+
+    builder = StateGraph(MessagesState)
+    builder.add_node("retriever", retriever_node)
+    builder.set_entry_point("retriever")
+    builder.set_finish_point("retriever")
+    return builder.compile()
+
+
+graph = build_graph()