@@ -2818,27 +2818,27 @@ export class CopilotApiGateway implements vscode.Disposable {
28182818
28192819 switch ( msg . role ) {
28202820 case 'system' :
2821- lmMessages . push ( vscode . LanguageModelChatMessage . User ( `[System]: ${ content } ` ) ) ;
2821+ lmMessages . push ( vscode . LanguageModelChatMessage . User ( `[System]: ${ content } ` , ( msg as any ) . name ) ) ;
28222822 break ;
28232823 case 'user' :
2824- lmMessages . push ( vscode . LanguageModelChatMessage . User ( content ) ) ;
2824+ lmMessages . push ( vscode . LanguageModelChatMessage . User ( content , ( msg as any ) . name ) ) ;
28252825 break ;
28262826 case 'assistant' :
28272827 if ( msg . tool_calls && msg . tool_calls . length > 0 ) {
28282828 const toolCallInfo = msg . tool_calls . map ( ( tc : any ) =>
28292829 `[Called function: ${ tc . function ?. name || tc . name } (${ tc . function ?. arguments || JSON . stringify ( tc . arguments ) } )]`
28302830 ) . join ( '\n' ) ;
2831- lmMessages . push ( vscode . LanguageModelChatMessage . Assistant ( toolCallInfo ) ) ;
2831+ lmMessages . push ( vscode . LanguageModelChatMessage . Assistant ( toolCallInfo , ( msg as any ) . name ) ) ;
28322832 } else {
2833- lmMessages . push ( vscode . LanguageModelChatMessage . Assistant ( content ) ) ;
2833+ lmMessages . push ( vscode . LanguageModelChatMessage . Assistant ( content , ( msg as any ) . name ) ) ;
28342834 }
28352835 break ;
28362836 case 'tool' :
28372837 const toolResultContent = `[Tool result for ${ msg . tool_call_id || 'unknown' } ]: ${ content } ` ;
2838- lmMessages . push ( vscode . LanguageModelChatMessage . User ( toolResultContent ) ) ;
2838+ lmMessages . push ( vscode . LanguageModelChatMessage . User ( toolResultContent , ( msg as any ) . name ) ) ;
28392839 break ;
28402840 default :
2841- lmMessages . push ( vscode . LanguageModelChatMessage . User ( content ) ) ;
2841+ lmMessages . push ( vscode . LanguageModelChatMessage . User ( content , ( msg as any ) . name ) ) ;
28422842 }
28432843 }
28442844
@@ -3978,29 +3978,29 @@ export class CopilotApiGateway implements vscode.Disposable {
39783978
39793979 switch ( msg . role ) {
39803980 case 'system' :
3981- lmMessages . push ( vscode . LanguageModelChatMessage . User ( `[System]: ${ content } ` ) ) ;
3981+ lmMessages . push ( vscode . LanguageModelChatMessage . User ( `[System]: ${ content } ` , ( msg as any ) . name ) ) ;
39823982 break ;
39833983 case 'user' :
3984- lmMessages . push ( vscode . LanguageModelChatMessage . User ( content ) ) ;
3984+ lmMessages . push ( vscode . LanguageModelChatMessage . User ( content , ( msg as any ) . name ) ) ;
39853985 break ;
39863986 case 'assistant' :
39873987 if ( msg . tool_calls && msg . tool_calls . length > 0 ) {
39883988 // Assistant message with tool calls - include tool call info
39893989 const toolCallInfo = msg . tool_calls . map ( ( tc : any ) =>
39903990 `[Called function: ${ tc . function ?. name || tc . name } (${ tc . function ?. arguments || JSON . stringify ( tc . arguments ) } )]`
39913991 ) . join ( '\n' ) ;
3992- lmMessages . push ( vscode . LanguageModelChatMessage . Assistant ( toolCallInfo ) ) ;
3992+ lmMessages . push ( vscode . LanguageModelChatMessage . Assistant ( toolCallInfo , ( msg as any ) . name ) ) ;
39933993 } else {
3994- lmMessages . push ( vscode . LanguageModelChatMessage . Assistant ( content ) ) ;
3994+ lmMessages . push ( vscode . LanguageModelChatMessage . Assistant ( content , ( msg as any ) . name ) ) ;
39953995 }
39963996 break ;
39973997 case 'tool' :
39983998 // Tool result message
39993999 const toolResultContent = `[Tool result for ${ msg . tool_call_id || 'unknown' } ]: ${ content } ` ;
4000- lmMessages . push ( vscode . LanguageModelChatMessage . User ( toolResultContent ) ) ;
4000+ lmMessages . push ( vscode . LanguageModelChatMessage . User ( toolResultContent , ( msg as any ) . name ) ) ;
40014001 break ;
40024002 default :
4003- lmMessages . push ( vscode . LanguageModelChatMessage . User ( content ) ) ;
4003+ lmMessages . push ( vscode . LanguageModelChatMessage . User ( content , ( msg as any ) . name ) ) ;
40044004 }
40054005 }
40064006
0 commit comments