Add query_tags support to executemany method

- Added query_tags parameter to executemany() method
- Query tags are applied to all queries in the batch
- Updated example to demonstrate executemany usage with query_tags
- All tests pass (122/122 client tests)

Signed-off-by: Jiabin Hu <jiabin.hu@databricks.com>

Author: jiabin-hu

examples/query_tags_example.py

@@ -93,4 +93,24 @@
         result = cursor.fetchone()
         print(f"  Async Query Result: {result[0]}")
 
+print()
+
+# Example 4: executemany with query tags
+print("Example 4: executemany with query tags")
+with sql.connect(
+    server_hostname=os.getenv("DATABRICKS_SERVER_HOSTNAME"),
+    http_path=os.getenv("DATABRICKS_HTTP_PATH"),
+    access_token=os.getenv("DATABRICKS_TOKEN"),
+) as connection:
+
+    with connection.cursor() as cursor:
+        # Execute multiple queries with the same tags
+        cursor.executemany(
+            "SELECT ?",
+            [[5], [6], [7]],
+            query_tags={"team": "data-eng", "batch": "executemany"}
+        )
+        result = cursor.fetchone()
+        print(f"  Executemany Query Result (last): {result[0]}")
+
 print("\n=== Query Tags Example Complete ===")

src/databricks/sql/client.py

@@ -1459,7 +1459,12 @@ def get_async_execution_result(self):
                 session_id_hex=self.connection.get_session_id_hex(),
             )
 
-    def executemany(self, operation, seq_of_parameters):
+    def executemany(
+        self,
+        operation,
+        seq_of_parameters,
+        query_tags: Optional[Dict[str, Optional[str]]] = None,
+    ):
         """
         Execute the operation once for every set of passed in parameters.
 
@@ -1468,10 +1473,14 @@ def executemany(self, operation, seq_of_parameters):
 
         Only the final result set is retained.
 
+        :param query_tags: Optional dictionary of query tags to apply for all queries in this batch.
+            Tags are key-value pairs that can be used to identify and categorize queries.
+            Example: {"team": "data-eng", "application": "etl"}
+
         :returns self
         """
         for parameters in seq_of_parameters:
-            self.execute(operation, parameters)
+            self.execute(operation, parameters, query_tags=query_tags)
         return self
 
     @log_latency(StatementType.METADATA)