add cli
This commit is contained in:
@@ -0,0 +1,30 @@
|
|||||||
|
import argparse
|
||||||
|
from tools import add_to_knowledge_base, search_knowledge_base
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
parser = argparse.ArgumentParser(description="RAG Agent CLI")
|
||||||
|
subparsers = parser.add_subparsers(dest='command')
|
||||||
|
|
||||||
|
add_parser = subparsers.add_parser('add', help='Add document to knowledge base')
|
||||||
|
add_parser.add_argument('--title', required=True, help='Title of the document')
|
||||||
|
add_parser.add_argument('--content', required=True, help='Content of the document')
|
||||||
|
|
||||||
|
search_parser = subparsers.add_parser('search', help='Search knowledge base')
|
||||||
|
search_parser.add_argument('--query', required=True, help='Query string')
|
||||||
|
search_parser.add_argument('--max', type=int, default=5, help='Max results')
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
if args.command == 'add':
|
||||||
|
result = add_to_knowledge_base(args.content, args.title)
|
||||||
|
print(result)
|
||||||
|
elif args.command == 'search':
|
||||||
|
results = search_knowledge_base(args.query, args.max)
|
||||||
|
for i, r in enumerate(results, 1):
|
||||||
|
print(f"{i}. {r}")
|
||||||
|
else:
|
||||||
|
parser.print_help()
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
Reference in New Issue
Block a user