simplify interactive query path in main() for all examples

examples/amazon_s3_embedding/main.py

@@ -57,19 +57,16 @@ def _main():
     with cocoindex.FlowLiveUpdater(amazon_s3_text_embedding_flow):
         # Run queries in a loop to demonstrate the query capabilities.
         while True:
-            try:
-                query = input("Enter search query (or Enter to quit): ")
-                if query == '':
-                    break
-                results, _ = query_handler.search(query, 10)
-                print("\nSearch results:")
-                for result in results:
-                    print(f"[{result.score:.3f}] {result.data['filename']}")
-                    print(f"    {result.data['text']}")
-                    print("---")
-                print()
-            except KeyboardInterrupt:
+            query = input("Enter search query (or Enter to quit): ")
+            if query == '':
                 break
+            results, _ = query_handler.search(query, 10)
+            print("\nSearch results:")
+            for result in results:
+                print(f"[{result.score:.3f}] {result.data['filename']}")
+                print(f"    {result.data['text']}")
+                print("---")
+            print()
 
 if __name__ == "__main__":
     load_dotenv()

examples/code_embedding/main.py

@@ -74,20 +74,17 @@ def _main():
     pool = ConnectionPool(os.getenv("COCOINDEX_DATABASE_URL"))
     # Run queries in a loop to demonstrate the query capabilities.
     while True:
-        try:
-            query = input("Enter search query (or Enter to quit): ")
-            if query == '':
-                break
-            # Run the query function with the database connection pool and the query.
-            results = search(pool, query)
-            print("\nSearch results:")
-            for result in results:
-                print(f"[{result['score']:.3f}] {result['filename']}")
-                print(f"    {result['code']}")
-                print("---")
-            print()
-        except KeyboardInterrupt:
+        query = input("Enter search query (or Enter to quit): ")
+        if query == '':
             break
+        # Run the query function with the database connection pool and the query.
+        results = search(pool, query)
+        print("\nSearch results:")
+        for result in results:
+            print(f"[{result['score']:.3f}] {result['filename']}")
+            print(f"    {result['code']}")
+            print("---")
+        print()
 
 if __name__ == "__main__":
     load_dotenv()

examples/gdrive_text_embedding/main.py

@@ -72,20 +72,17 @@ def _main():
     pool = ConnectionPool(os.getenv("COCOINDEX_DATABASE_URL"))
     # Run queries in a loop to demonstrate the query capabilities.
     while True:
-        try:
-            query = input("Enter search query (or Enter to quit): ")
-            if query == '':
-                break
-            # Run the query function with the database connection pool and the query.
-            results = search(pool, query)
-            print("\nSearch results:")
-            for result in results:
-                print(f"[{result['score']:.3f}] {result['filename']}")
-                print(f"    {result['text']}")
-                print("---")
-            print()
-        except KeyboardInterrupt:
+        query = input("Enter search query (or Enter to quit): ")
+        if query == '':
             break
+        # Run the query function with the database connection pool and the query.
+        results = search(pool, query)
+        print("\nSearch results:")
+        for result in results:
+            print(f"[{result['score']:.3f}] {result['filename']}")
+            print(f"    {result['text']}")
+            print("---")
+        print()
 
 if __name__ == "__main__":
     load_dotenv()

examples/pdf_embedding/main.py

@@ -111,15 +111,12 @@ def _main():
     pool = ConnectionPool(os.getenv("COCOINDEX_DATABASE_URL"))
     # Run queries in a loop to demonstrate the query capabilities.
     while True:
-        try:
-            query = input("Enter search query (or Enter to quit): ")
-            if query == '':
-                break
-            # Run the query function with the database connection pool and the query.
-            results = search(pool, query)
-            print(SEARCH_RESULTS_TEMPLATE.render(results=results))
-        except KeyboardInterrupt:
+        query = input("Enter search query (or Enter to quit): ")
+        if query == '':
             break
+        # Run the query function with the database connection pool and the query.
+        results = search(pool, query)
+        print(SEARCH_RESULTS_TEMPLATE.render(results=results))
 
 
 if __name__ == "__main__":

examples/text_embedding/main.py

@@ -65,20 +65,17 @@ def _main():
     pool = ConnectionPool(os.getenv("COCOINDEX_DATABASE_URL"))
     # Run queries in a loop to demonstrate the query capabilities.
     while True:
-        try:
-            query = input("Enter search query (or Enter to quit): ")
-            if query == '':
-                break
-            # Run the query function with the database connection pool and the query.
-            results = search(pool, query)
-            print("\nSearch results:")
-            for result in results:
-                print(f"[{result['score']:.3f}] {result['filename']}")
-                print(f"    {result['text']}")
-                print("---")
-            print()
-        except KeyboardInterrupt:
+        query = input("Enter search query (or Enter to quit): ")
+        if query == '':
             break
+        # Run the query function with the database connection pool and the query.
+        results = search(pool, query)
+        print("\nSearch results:")
+        for result in results:
+            print(f"[{result['score']:.3f}] {result['filename']}")
+            print(f"    {result['text']}")
+            print("---")
+        print()
 
 if __name__ == "__main__":
     load_dotenv()

examples/text_embedding_qdrant/main.py

@@ -66,29 +66,26 @@ def _main():
 
     # Run queries in a loop to demonstrate the query capabilities.
     while True:
-        try:
-            query = input("Enter search query (or Enter to quit): ")
-            if query == "":
-                break
-
-            # Get the embedding for the query
-            query_embedding = text_to_embedding.eval(query)
-
-            search_results = client.search(
-                collection_name=QDRANT_COLLECTION,
-                query_vector=("text_embedding", query_embedding),
-                limit=10
-            )
-            print("\nSearch results:")
-            for result in search_results:
-                score = result.score
-                payload = result.payload
-                print(f"[{score:.3f}] {payload['filename']}")
-                print(f"    {payload['text']}")
-                print("---")
-            print()
-        except KeyboardInterrupt:
+        query = input("Enter search query (or Enter to quit): ")
+        if query == "":
             break
+
+        # Get the embedding for the query
+        query_embedding = text_to_embedding.eval(query)
+
+        search_results = client.search(
+            collection_name=QDRANT_COLLECTION,
+            query_vector=("text_embedding", query_embedding),
+            limit=10
+        )
+        print("\nSearch results:")
+        for result in search_results:
+            score = result.score
+            payload = result.payload
+            print(f"[{score:.3f}] {payload['filename']}")
+            print(f"    {payload['text']}")
+            print("---")
+        print()
 
 
 if __name__ == "__main__":