Skip to content

Commit 987dbad

Browse files
committed
proper config respecting jetstream
1 parent 42e8458 commit 987dbad

File tree

1 file changed

+24
-2
lines changed

1 file changed

+24
-2
lines changed

crates/pattern_cli/src/agent_ops.rs

Lines changed: 24 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -753,7 +753,18 @@ pub async fn register_data_sources_with_target<M, E>(
753753
tracing::info!("filter: {:?}", filter);
754754
let data_sources = if let Some(target) = target {
755755
DataSourceBuilder::new()
756-
.with_bluesky_source("bluesky_jetstream".to_string(), filter, true)
756+
.with_custom_bluesky_source(
757+
"bluesky_jetstream".to_string(),
758+
config
759+
.bluesky
760+
.as_ref()
761+
.map(|b| b.jetstream_endpoint.clone())
762+
.unwrap_or(
763+
"wss://jetstream1.us-east.fire.hose.cam/subscribe".to_string(),
764+
),
765+
filter,
766+
true,
767+
)
757768
.build_with_target(
758769
agent.id(),
759770
agent.name(),
@@ -767,7 +778,18 @@ pub async fn register_data_sources_with_target<M, E>(
767778
.unwrap()
768779
} else {
769780
DataSourceBuilder::new()
770-
.with_bluesky_source("bluesky_jetstream".to_string(), filter, true)
781+
.with_custom_bluesky_source(
782+
"bluesky_jetstream".to_string(),
783+
config
784+
.bluesky
785+
.as_ref()
786+
.map(|b| b.jetstream_endpoint.clone())
787+
.unwrap_or(
788+
"wss://jetstream1.us-east.fire.hose.cam/subscribe".to_string(),
789+
),
790+
filter,
791+
true,
792+
)
771793
.build(
772794
agent.id(),
773795
agent.name(),

0 commit comments

Comments
 (0)