Fix input type identification

Author: 4adex

editor/src/messages/portfolio/document/node_graph/node_graph_message_handler.rs

@@ -2146,14 +2146,17 @@ impl NodeGraphMessageHandler {
 							.icon(Some("Node".to_string()))
 							.tooltip("Add an operation to the end of this layer's chain of nodes")
 							.popover_layout({
-								let input_connector = InputConnector::node(layer, 1);
-								let compatible_type =
-									if let Some(OutputConnector::Node { node_id, .. }) = context.network_interface.upstream_output_connector(&input_connector, &context.selection_network_path) {
-										let (output_type, _) = context.network_interface.output_type(&node_id, 0, &context.selection_network_path);
+								let layer_identifier = LayerNodeIdentifier::new(layer, &context.network_interface, context.selection_network_path);
+								let compatible_type = {
+									let graph_layer = graph_modification_utils::NodeGraphLayer::new(layer_identifier, &context.network_interface);
+									let node_type = graph_layer.horizontal_layer_flow().nth(1);
+									if let Some(node_id) = node_type {
+										let (output_type, _) = context.network_interface.output_type(&node_id, 0, &[]);
 										Some(format!("type:{}", output_type.nested_type()))
 									} else {
 										None
-									};
+									}
+								};
 
 								let mut node_chooser = NodeCatalog::new();
 								node_chooser.intial_search = compatible_type.unwrap_or("".to_string());