Merge pull request #893 from paschal533/keyerror-fix

seetadev · web-flow · commit a01811a435a6 · 2025-09-22T01:57:06.000+05:30
fix: GossipSub peer propagation to include FloodSub peers
diff --git a/libp2p/pubsub/gossipsub.py b/libp2p/pubsub/gossipsub.py
@@ -308,7 +308,8 @@ def _get_peers_to_send(
             floodsub_peers: set[ID] = {
                 peer_id
                 for peer_id in self.pubsub.peer_topics[topic]
-                if self.peer_protocol[peer_id] == floodsub.PROTOCOL_ID
+                if peer_id in self.peer_protocol
+                and self.peer_protocol[peer_id] == floodsub.PROTOCOL_ID
             }
             send_to.update(floodsub_peers)
 

Original file line number	Diff line number	Diff line change
`@@ -308,7 +308,8 @@ def _get_peers_to_send(`
`308`	`308`	`floodsub_peers: set[ID] = {`
`309`	`309`	`peer_id`
`310`	`310`	`for peer_id in self.pubsub.peer_topics[topic]`
`311`		`- if self.peer_protocol[peer_id] == floodsub.PROTOCOL_ID`
	`311`	`+ if peer_id in self.peer_protocol`
	`312`	`+ and self.peer_protocol[peer_id] == floodsub.PROTOCOL_ID`
`312`	`313`	`}`
`313`	`314`	`send_to.update(floodsub_peers)`
`314`	`315`