File tree Expand file tree Collapse file tree 1 file changed +24
-2
lines changed
Expand file tree Collapse file tree 1 file changed +24
-2
lines changed Original file line number Diff line number Diff 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 ( ) ,
You can’t perform that action at this time.
0 commit comments