Skip to content

SnowflakeCortexAgentHook: Set stream=False for Cortex Agent requests#69731

Draft
SameerMesiah97 wants to merge 1 commit into
apache:mainfrom
SameerMesiah97:SnowflakeCortexAgentHook-Fix
Draft

SnowflakeCortexAgentHook: Set stream=False for Cortex Agent requests#69731
SameerMesiah97 wants to merge 1 commit into
apache:mainfrom
SameerMesiah97:SnowflakeCortexAgentHook-Fix

Conversation

@SameerMesiah97

Copy link
Copy Markdown
Contributor

Description

This change fixes the SnowflakeCortexAgentHook by explicitly setting stream=False when invoking the Cortex Agent API, ensuring the hook receives a JSON response.

Rationale

The Cortex Agent API returns a Server-Sent Events (SSE) stream by default. Since the hook expects a JSON response, explicitly disabling streaming restores the expected behaviour.

Tests

Updated the unit tests to verify that stream=False is included in all Cortex Agent requests.

@boring-cyborg boring-cyborg Bot added area:providers provider:snowflake Issues related to Snowflake provider labels Jul 10, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:providers provider:snowflake Issues related to Snowflake provider

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant