forked from sanbuphy/learn-coding-agent
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathutils.ts
More file actions
40 lines (38 loc) · 1.08 KB
/
utils.ts
File metadata and controls
40 lines (38 loc) · 1.08 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import type {
AssistantMessage,
AttachmentMessage,
SystemMessage,
UserMessage,
} from 'src/types/message.js'
/**
* Tags user messages with a sourceToolUseID so they stay transient until the tool resolves.
* This prevents the "is running" message from being duplicated in the UI.
*/
export function tagMessagesWithToolUseID(
messages: (UserMessage | AttachmentMessage | SystemMessage)[],
toolUseID: string | undefined,
): (UserMessage | AttachmentMessage | SystemMessage)[] {
if (!toolUseID) {
return messages
}
return messages.map(m => {
if (m.type === 'user') {
return { ...m, sourceToolUseID: toolUseID }
}
return m
})
}
/**
* Extracts the tool use ID from a parent message for a given tool name.
*/
export function getToolUseIDFromParentMessage(
parentMessage: AssistantMessage,
toolName: string,
): string | undefined {
const toolUseBlock = parentMessage.message.content.find(
block => block.type === 'tool_use' && block.name === toolName,
)
return toolUseBlock && toolUseBlock.type === 'tool_use'
? toolUseBlock.id
: undefined
}