@@ -3,7 +3,7 @@ import {useTheme} from '@emotion/react';
3
3
import styled from '@emotion/styled' ;
4
4
5
5
import Placeholder from 'sentry/components/placeholder' ;
6
- import { IconCode , IconSort } from 'sentry/icons' ;
6
+ import { IconChevron , IconCode } from 'sentry/icons' ;
7
7
import { IconBot } from 'sentry/icons/iconBot' ;
8
8
import { IconSpeechBubble } from 'sentry/icons/iconSpeechBubble' ;
9
9
import { IconTool } from 'sentry/icons/iconTool' ;
@@ -14,6 +14,7 @@ import {
14
14
AI_AGENT_NAME_ATTRIBUTE ,
15
15
AI_GENERATION_DESCRIPTIONS ,
16
16
AI_GENERATION_OPS ,
17
+ AI_HANDOFF_OPS ,
17
18
AI_MODEL_ID_ATTRIBUTE ,
18
19
AI_RUN_DESCRIPTIONS ,
19
20
AI_RUN_OPS ,
@@ -253,7 +254,7 @@ function getNodeInfo(
253
254
icon : < IconCode size = "md" /> ,
254
255
title : 'Unknown' ,
255
256
subtitle : '' ,
256
- color : colors [ 0 ] ,
257
+ color : colors [ 1 ] ,
257
258
} ;
258
259
259
260
if ( isTransactionNode ( node ) ) {
@@ -288,7 +289,7 @@ function getNodeInfo(
288
289
nodeInfo . icon = < IconBot size = "md" /> ;
289
290
nodeInfo . title = op ;
290
291
nodeInfo . subtitle = `${ agentName } ${ model ? ` (${ model } )` : '' } ` ;
291
- nodeInfo . color = colors [ 1 ] ;
292
+ nodeInfo . color = colors [ 0 ] ;
292
293
} else if (
293
294
AI_GENERATION_OPS . includes ( op ) ||
294
295
AI_GENERATION_DESCRIPTIONS . includes ( node . value . description ?? '' )
@@ -305,10 +306,11 @@ function getNodeInfo(
305
306
nodeInfo . icon = < IconTool size = "md" /> ;
306
307
nodeInfo . title = op || 'gen_ai.toolCall' ;
307
308
nodeInfo . subtitle = getNodeAttribute ( AI_TOOL_NAME_ATTRIBUTE ) || '' ;
308
- nodeInfo . color = colors [ 3 ] ;
309
- } else if ( op === 'http.client' ) {
310
- nodeInfo . icon = < IconSort size = "md" /> ;
311
- nodeInfo . title = node . value . description || 'HTTP' ;
309
+ nodeInfo . color = colors [ 5 ] ;
310
+ } else if ( AI_HANDOFF_OPS . includes ( op ) ) {
311
+ nodeInfo . icon = < IconChevron size = "md" isDouble direction = "right" /> ;
312
+ nodeInfo . title = op ;
313
+ nodeInfo . subtitle = node . value . description || '' ;
312
314
nodeInfo . color = colors [ 4 ] ;
313
315
} else {
314
316
nodeInfo . title = op || 'Span' ;
0 commit comments