@@ -127,6 +127,11 @@ export function mapOutputToChatResult(
127127 tool_calls : mapAzureOpenAiToLangChainToolCall (
128128 choice . message . tool_calls
129129 ) ,
130+ usage_metadata : {
131+ input_tokens : choice . message . usage ?. prompt_tokens ?? 0 ,
132+ output_tokens : choice . message . usage ?. completion_tokens ?? 0 ,
133+ total_tokens : choice . message . usage ?. total_tokens ?? 0
134+ } ,
130135 additional_kwargs : {
131136 function_call : choice . message . function_call ,
132137 tool_calls : choice . message . tool_calls
@@ -248,19 +253,23 @@ function mapSystemMessageToAzureOpenAiSystemMessage(
248253function mapBaseMessageToAzureOpenAiChatMessage (
249254 message : BaseMessage
250255) : AzureOpenAiChatCompletionRequestMessage {
251- switch ( message . getType ( ) ) {
256+ switch ( message . type ) {
252257 case 'ai' :
253- return mapAiMessageToAzureOpenAiAssistantMessage ( message ) ;
258+ return mapAiMessageToAzureOpenAiAssistantMessage ( message as AIMessage ) ;
254259 case 'human' :
255- return mapHumanMessageToAzureOpenAiUserMessage ( message ) ;
260+ return mapHumanMessageToAzureOpenAiUserMessage ( message as HumanMessage ) ;
256261 case 'system' :
257- return mapSystemMessageToAzureOpenAiSystemMessage ( message ) ;
262+ return mapSystemMessageToAzureOpenAiSystemMessage (
263+ message as SystemMessage
264+ ) ;
258265 case 'function' :
259- return mapFunctionMessageToAzureOpenAiFunctionMessage ( message ) ;
266+ return mapFunctionMessageToAzureOpenAiFunctionMessage (
267+ message as FunctionMessage
268+ ) ;
260269 case 'tool' :
261270 return mapToolMessageToAzureOpenAiToolMessage ( message as ToolMessage ) ;
262271 default :
263- throw new Error ( `Unsupported message type: ${ message . getType ( ) } ` ) ;
272+ throw new Error ( `Unsupported message type: ${ message . type } ` ) ;
264273 }
265274}
266275
0 commit comments