File tree 1 file changed +7
-7
lines changed
libs/langchain-google-genai/src
1 file changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -905,21 +905,21 @@ export class ChatGoogleGenerativeAI
905
905
options . streamUsage !== false
906
906
) {
907
907
const genAIUsageMetadata = response . usageMetadata as {
908
- promptTokenCount : number ;
909
- candidatesTokenCount : number ;
910
- totalTokenCount : number ;
908
+ promptTokenCount : number | undefined ;
909
+ candidatesTokenCount : number | undefined ;
910
+ totalTokenCount : number | undefined ;
911
911
} ;
912
912
if ( ! usageMetadata ) {
913
913
usageMetadata = {
914
- input_tokens : genAIUsageMetadata . promptTokenCount ,
915
- output_tokens : genAIUsageMetadata . candidatesTokenCount ,
916
- total_tokens : genAIUsageMetadata . totalTokenCount ,
914
+ input_tokens : genAIUsageMetadata . promptTokenCount ?? 0 ,
915
+ output_tokens : genAIUsageMetadata . candidatesTokenCount ?? 0 ,
916
+ total_tokens : genAIUsageMetadata . totalTokenCount ?? 0 ,
917
917
} ;
918
918
} else {
919
919
// Under the hood, LangChain combines the prompt tokens. Google returns the updated
920
920
// total each time, so we need to find the difference between the tokens.
921
921
const outputTokenDiff =
922
- genAIUsageMetadata . candidatesTokenCount -
922
+ ( genAIUsageMetadata . candidatesTokenCount ?? 0 ) -
923
923
usageMetadata . output_tokens ;
924
924
usageMetadata = {
925
925
input_tokens : 0 ,
You can’t perform that action at this time.
0 commit comments