Delete playground_chat.py

playground_chat.py

@@ -1,79 +0,0 @@
-"""Playground script for testing the full chat pipeline (Orchestrator → Retriever → Chatbot).
-
-Usage:
-    uv run python playground_chat.py
-
-Tweak the variables in the CONFIG section below and re-run.
-"""
-
-import asyncio
-import json
-from langchain_core.messages import HumanMessage
-
-from src.agents.orchestration import orchestrator
-from src.agents.chatbot import chatbot
-from src.rag.retriever import retriever
-from src.api.v1.chat import _format_context, _extract_sources
-
-# ──────────────────────────────────────────────
-# CONFIG — change these freely
-# ──────────────────────────────────────────────
-MESSAGE = "Berapa digital rate card untuk Net Message?"
-USER_ID = "8b6c18fd-8971-46e5-b106-35b7afb412e0"
-TOP_K   = 5   # number of chunks to retrieve
-# ──────────────────────────────────────────────
-
-
-async def main():
-    print(f"\n{'='*60}")
-    print(f"Message : {MESSAGE}")
-    print(f"User ID : {USER_ID}")
-    print(f"Top-K   : {TOP_K}")
-    print(f"{'='*60}\n")
-
-    # Step 1: Orchestrator — intent analysis
-    print("── Step 1: Orchestrator ──────────────────────────────────")
-    intent_result = await orchestrator.analyze_message(MESSAGE)
-    print(json.dumps(intent_result, indent=2, ensure_ascii=False))
-
-    context = ""
-    sources = []
-
-    # Step 2: Retriever (only if orchestrator says so)
-    if intent_result.get("needs_search"):
-        search_query = intent_result.get("search_query", MESSAGE)
-        print(f"\n── Step 2: Retriever (query: {search_query!r}) ──────────────")
-        raw_results = await retriever.retrieve(
-            query=search_query,
-            user_id=USER_ID,
-            db=None,
-            k=TOP_K,
-        )
-        context = _format_context(raw_results)
-        sources = _extract_sources(raw_results)
-
-        print(f"Retrieved {len(raw_results)} chunk(s). Sources:")
-        for s in sources:
-            print(f"  - {s['filename']} p.{s['page_label']}")
-        print(f"\nContext preview:\n{context[:500]}{'...' if len(context) > 500 else ''}")
-    else:
-        print("\n── Step 2: Retriever — skipped (no search needed)")
-
-    # Step 3: Direct response (greetings, etc.)
-    if intent_result.get("direct_response"):
-        print("\n── Step 3: Direct response (no LLM call) ────────────────")
-        print(intent_result["direct_response"])
-        return
-
-    # Step 4: Chatbot — generate answer
-    print("\n── Step 4: Chatbot response ──────────────────────────────")
-    messages = [HumanMessage(content=MESSAGE)]
-    response = await chatbot.generate_response(messages, context)
-    print(response)
-
-    print(f"\n── Sources ───────────────────────────────────────────────")
-    print(json.dumps(sources, indent=2, ensure_ascii=False))
-
-
-if __name__ == "__main__":
-    asyncio.run(main())