From 5069dc05b58efc388d599fafb75b8e88e86f0d79 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=9A=D0=B8=D1=80=D0=B8=D0=BB=D0=BB=20=D0=9A=D1=83=D1=82?=
 =?UTF-8?q?=D0=BB=D0=B0=D1=85=D0=BC=D0=B5=D1=82=D0=BE=D0=B2?=
 <kirillkutlahmetov10@gmail.com>
Date: Thu, 28 May 2026 09:15:22 +0000
Subject: [PATCH] Add cli.py

---
 cli.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 cli.py

diff --git a/cli.py b/cli.py
new file mode 100644
index 0000000..9fa2872
--- /dev/null
+++ b/cli.py
@@ -0,0 +1,43 @@
+"""
+Command‑line interface for the RAG agent.
+
+Commands:
+  add <file>   – add document from file to knowledge base
+  search <q>   – search query in knowledge base
+  quit         – exit
+"""
+
+import argparse
+import asyncio
+from pathlib import Path
+from langchain_core.messages import HumanMessage
+from agent import agent
+
+async def run_cli():
+    parser = argparse.ArgumentParser(description="RAG Agent CLI")
+    subparsers = parser.add_subparsers(dest="cmd", required=True)
+
+    add_parser = subparsers.add_parser("add", help="Add document from file")
+    add_parser.add_argument("file", type=Path, help="Path to text file")
+
+    search_parser = subparsers.add_parser("search", help="Search query in knowledge base")
+    search_parser.add_argument("query", type=str, help="Search string")
+
+    args = parser.parse_args()
+
+    if args.cmd == "add":
+        content = args.file.read_text(encoding="utf-8")
+        title = args.file.stem
+        await agent.ainvoke(
+            {"messages": [HumanMessage(content=f"Add document: {title}")]},
+            {"configurable": {"thread_id": "cli-add"}},
+        )
+    elif args.cmd == "search":
+        result = await agent.ainvoke(
+            {"messages": [HumanMessage(content=args.query)]},
+            {"configurable": {"thread_id": "cli-search"}},
+        )
+        print(result["messages"][-1].content)
+
+if __name__ == "__main__":  # pragma: no cover
+    asyncio.run(run_cli())