fix: validate configure probe for streaming providers (#7564)
Signed-off-by: rabi <ramishra@redhat.com>
This commit is contained in:
@@ -1,5 +1,6 @@
|
|||||||
use crate::{conversation::message::Message, model::ModelConfig, providers::create};
|
use crate::{conversation::message::Message, model::ModelConfig, providers::create};
|
||||||
use anyhow::Result;
|
use anyhow::Result;
|
||||||
|
use futures::StreamExt;
|
||||||
use rmcp::model::ToolAnnotations;
|
use rmcp::model::ToolAnnotations;
|
||||||
use rmcp::{model::Tool, object};
|
use rmcp::{model::Tool, object};
|
||||||
|
|
||||||
@@ -27,8 +28,8 @@ pub async fn test_provider_configuration(
|
|||||||
};
|
};
|
||||||
|
|
||||||
let provider_model_config = provider.get_model_config();
|
let provider_model_config = provider.get_model_config();
|
||||||
let _result = provider
|
let mut stream = provider
|
||||||
.complete(
|
.stream(
|
||||||
&provider_model_config,
|
&provider_model_config,
|
||||||
"test-session-id",
|
"test-session-id",
|
||||||
"You are an AI agent called goose. You use tools of connected extensions to solve problems.",
|
"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?;
|
.await?;
|
||||||
|
|
||||||
|
let first_chunk = stream
|
||||||
|
.next()
|
||||||
|
.await
|
||||||
|
.ok_or_else(|| anyhow::anyhow!("Provider test stream returned no events"))?;
|
||||||
|
first_chunk?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user