fix(kernel): recognize streaming PUT after a closing quote

Signed-off-by: Aakash Saravanan <aakash.saravanan@databricks.com>

Author: aakash-saravanan-db

src/databricks/sql/backend/kernel/client.py

@@ -563,7 +563,7 @@ def execute_command(
                 stripped_sql = _strip_leading_sql_comments(operation)
                 staging_verb = _staging_verb(stripped_sql)
                 if re.match(
-                    r"""PUT\s*(['"])__input_stream__\1(?:\s|$)""",
+                    r"""PUT\s*(['"])__input_stream__\1""",
                     stripped_sql,
                     re.IGNORECASE,
                 ):

tests/unit/test_kernel_client.py

@@ -932,6 +932,8 @@ def test_staging_verb_matches_only_the_leading_statement(operation, expected):
     "query",
     [
         "PUT '__input_stream__' INTO '/Volumes/c/s/v/file'",
+        "PUT '__input_stream__'INTO '/Volumes/c/s/v/file'",
+        'PUT"__input_stream__"INTO "/Volumes/c/s/v/file"',
         "/* upload */ PUT '__input_stream__' INTO '/Volumes/c/s/v/file'",
         "PUT'__input_stream__' INTO '/Volumes/c/s/v/file'",
         "/* upload */ PUT\"__input_stream__\" INTO '/Volumes/c/s/v/file'",