Testing prompt

utils/send_query.py

@@ -8,6 +8,7 @@ Usage:
     uv run utils/send_query.py "Hola, ¿cómo estás?"
     uv run utils/send_query.py --phone 5551234 "¿Qué servicios ofrecen?"
     uv run utils/send_query.py --base-url http://localhost:8080 "Hola"
+    uv run utils/send_query.py -i                   # interactive chat mode
 """
 
 from __future__ import annotations
@@ -16,33 +17,69 @@ import argparse
 
 import httpx
 from rich import print as rprint
+from rich.console import Console
+
+console = Console()
 
 
-def main() -> None:
-    parser = argparse.ArgumentParser(description="Send a query to the RAG agent")
-    parser.add_argument("text", help="Message to send")
-    parser.add_argument("--phone", default="test-user", help="Phone number / session id")
-    parser.add_argument("--base-url", default="http://localhost:8000", help="Server base URL")
-    args = parser.parse_args()
-
+def send_message(url: str, phone: str, text: str) -> dict:
     payload = {
-        "phone_number": args.phone,
-        "text": args.text,
+        "phone_number": phone,
+        "text": text,
         "type": "conversation",
         "language_code": "es",
     }
-
-    url = f"{args.base_url}/api/v1/query"
-    rprint(f"[bold]POST[/bold] {url}")
-    rprint(f"[dim]{payload}[/dim]\n")
-
     resp = httpx.post(url, json=payload, timeout=120)
     resp.raise_for_status()
-    data = resp.json()
+    return resp.json()
 
+
+def one_shot(url: str, phone: str, text: str) -> None:
+    rprint(f"[bold]POST[/bold] {url}")
+    rprint(f"[dim]{{'phone_number': {phone!r}, 'text': {text!r}}}[/dim]\n")
+    data = send_message(url, phone, text)
     rprint(f"[green bold]Response ([/green bold]{data['response_id']}[green bold]):[/green bold]")
     rprint(data["response_text"])
 
 
+def interactive(url: str, phone: str) -> None:
+    rprint(f"[bold cyan]Interactive chat[/bold cyan] → {url}  (session: {phone})")
+    rprint("[dim]Type /quit or Ctrl-C to exit[/dim]\n")
+    while True:
+        try:
+            text = console.input("[bold yellow]You>[/bold yellow] ").strip()
+        except (EOFError, KeyboardInterrupt):
+            rprint("\n[dim]Bye![/dim]")
+            break
+        if not text:
+            continue
+        if text.lower() in {"/quit", "/exit", "/q"}:
+            rprint("[dim]Bye![/dim]")
+            break
+        try:
+            data = send_message(url, phone, text)
+            rprint(f"[green bold]Agent>[/green bold] {data['response_text']}\n")
+        except httpx.HTTPStatusError as exc:
+            rprint(f"[red bold]Error {exc.response.status_code}:[/red bold] {exc.response.text}\n")
+        except httpx.ConnectError:
+            rprint("[red bold]Connection error:[/red bold] could not reach the server\n")
+
+
+def main() -> None:
+    parser = argparse.ArgumentParser(description="Send a query to the RAG agent")
+    parser.add_argument("text", nargs="?", default=None, help="Message to send (omit for interactive mode)")
+    parser.add_argument("-i", "--interactive", action="store_true", help="Start interactive chat session")
+    parser.add_argument("--phone", default="test-user", help="Phone number / session id")
+    parser.add_argument("--base-url", default="http://localhost:8000", help="Server base URL")
+    args = parser.parse_args()
+
+    url = f"{args.base_url}/api/v1/query"
+
+    if args.interactive or args.text is None:
+        interactive(url, args.phone)
+    else:
+        one_shot(url, args.phone, args.text)
+
+
 if __name__ == "__main__":
     main()