(airflow) branch main updated: Set stream=False for Snowflake Cortex Agent requests. (#69731)

[henrychen]

This is an automated email from the ASF dual-hosted git repository.

henry3260 pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git


The following commit(s) were added to refs/heads/main by this push:
     new 2b48289b230 Set stream=False for Snowflake Cortex Agent requests. 
(#69731)
2b48289b230 is described below

commit 2b48289b23010fe033d4197e787df9a71b862d9b
Author: SameerMesiah97 <75502260+sameermesia...@users.noreply.github.com>
AuthorDate: Sat Jul 18 13:48:03 2026 +0100

    Set stream=False for Snowflake Cortex Agent requests. (#69731)
---
 .../src/airflow/providers/snowflake/hooks/snowflake_cortex_agent.py  | 5 ++++-
 .../tests/unit/snowflake/hooks/test_snowflake_cortex_agent.py        | 3 +++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git 
a/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_cortex_agent.py
 
b/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_cortex_agent.py
index 7b7f37a2497..1413d7fee71 100644
--- 
a/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_cortex_agent.py
+++ 
b/providers/snowflake/src/airflow/providers/snowflake/hooks/snowflake_cortex_agent.py
@@ -130,7 +130,10 @@ class SnowflakeCortexAgentHook(SnowflakeHook):
         if thread_id is not None and parent_message_id is None:
             raise ValueError("parent_message_id must be provided when 
thread_id is specified.")
 
-        payload: dict[str, Any] = {"messages": messages}
+        payload: dict[str, Any] = {
+            "messages": messages,
+            "stream": False,
+        }
 
         if thread_id is not None:
             payload["thread_id"] = thread_id
diff --git 
a/providers/snowflake/tests/unit/snowflake/hooks/test_snowflake_cortex_agent.py 
b/providers/snowflake/tests/unit/snowflake/hooks/test_snowflake_cortex_agent.py
index 8ec86d532f9..e686b698574 100644
--- 
a/providers/snowflake/tests/unit/snowflake/hooks/test_snowflake_cortex_agent.py
+++ 
b/providers/snowflake/tests/unit/snowflake/hooks/test_snowflake_cortex_agent.py
@@ -125,6 +125,7 @@ class TestSnowflakeCortexAgentHook:
                         ],
                     }
                 ],
+                "stream": False,
             },
             timeout=REQUEST_TIMEOUT,
         )
@@ -175,6 +176,7 @@ class TestSnowflakeCortexAgentHook:
 
         assert payload["thread_id"] == 123
         assert payload["parent_message_id"] == 456
+        assert payload["stream"] is False
 
     @mock.patch(f"{MODULE_PATH}.requests.request")
     @mock.patch(f"{HOOK_PATH}._get_conn_params")
@@ -215,6 +217,7 @@ class TestSnowflakeCortexAgentHook:
         assert payload["orchestration"] == {"max_tokens": 1000}
         assert payload["tools"] == [{"name": "search_tool"}]
         assert payload["tool_resources"] == {"search_tool": {"config": 
"value"}}
+        assert payload["stream"] is False
 
     @mock.patch(f"{MODULE_PATH}.requests.request")
     @mock.patch(f"{HOOK_PATH}._get_conn_params")