From fa1cad3faa9833c695a927fc562fcac6ca6c6383 Mon Sep 17 00:00:00 2001 From: Rabi Mishra Date: Sat, 28 Feb 2026 06:22:20 +0530 Subject: [PATCH] fix: validate configure probe for streaming providers (#7564) Signed-off-by: rabi --- crates/goose/src/providers/provider_test.rs | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/crates/goose/src/providers/provider_test.rs b/crates/goose/src/providers/provider_test.rs index 33450bd978..d13c8b6b88 100644 --- a/crates/goose/src/providers/provider_test.rs +++ b/crates/goose/src/providers/provider_test.rs @@ -1,5 +1,6 @@ use crate::{conversation::message::Message, model::ModelConfig, providers::create}; use anyhow::Result; +use futures::StreamExt; use rmcp::model::ToolAnnotations; use rmcp::{model::Tool, object}; @@ -27,8 +28,8 @@ pub async fn test_provider_configuration( }; let provider_model_config = provider.get_model_config(); - let _result = provider - .complete( + let mut stream = provider + .stream( &provider_model_config, "test-session-id", "You are an AI agent called goose. You use tools of connected extensions to solve problems.", @@ -37,6 +38,12 @@ pub async fn test_provider_configuration( ) .await?; + let first_chunk = stream + .next() + .await + .ok_or_else(|| anyhow::anyhow!("Provider test stream returned no events"))?; + first_chunk?; + Ok(()) }