chore: make it simple

examples/llamaindex-tidb-vector/README.md

@@ -45,6 +45,17 @@ export TIDB_PASSWORD="****"
 
 ### Run this example
 
-```bash
-python app.py
+```text
+$ python talk_to_url.py --help
+Usage: talk_to_url.py [OPTIONS]
+
+Options:
+  --url TEXT  URL you want to talk to,
+              default=https://docs.pingcap.com/tidb/stable/overview
+  --help      Show this message and exit.
+$
+$ python talk_to_url.py
+Enter your question: : tidb vs mysql
+TiDB is an open-source distributed SQL database that supports Hybrid Transactional and Analytical Processing (HTAP) workloads. It is MySQL compatible and features horizontal scalability, strong consistency, and high availability. TiDB is designed to provide users with a one-stop database solution that covers OLTP, OLAP, and HTAP services. It offers easy horizontal scaling, financial-grade high availability, real-time HTAP capabilities, cloud-native features, and compatibility with the MySQL protocol and ecosystem.
+Enter your question: :
 ```

examples/llamaindex-tidb-vector/app.py → examples/llamaindex-tidb-vector/talk_to_url.py

@@ -29,16 +29,22 @@
 query_engine = tidb_vec_index.as_query_engine(streaming=True)
 
 
-def do_prepare_data():
-    documents = SimpleWebPageReader(html_to_text=True).load_data(
-        ["http://paulgraham.com/worked.html"]
-    )
+def do_prepare_data(url):
+    documents = SimpleWebPageReader(html_to_text=True).load_data([url,])
     tidb_vec_index.from_documents(documents, storage_context=storage_context, show_progress=True)
 
 
-if __name__ == '__main__':
-    do_prepare_data()
+_default_url = 'https://docs.pingcap.com/tidb/stable/overview'
+
+@click.command()
+@click.option('--url',default=_default_url,
+              help=f'URL you want to talk to, default={_default_url}')
+def talk_to(url):
+    do_prepare_data(url)
     while True:
         question = click.prompt("Enter your question: ")
         response = query_engine.query(question)
         click.echo(response)
+
+if __name__ == '__main__':
+    talk_to()