Skip to content

Commit 577a39f

Browse files
Merge pull request #114 from suhaibbinyounis/feat/106-forward-message-name
feat: forward name field from messages to Copilot API
2 parents b696c91 + b399c3e commit 577a39f

1 file changed

Lines changed: 12 additions & 12 deletions

File tree

src/CopilotApiGateway.ts

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)