44 lines
1.4 KiB
Python
44 lines
1.4 KiB
Python
"""
|
||
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())
|