fix: mistral ai model conditional

yiyuan-he · yiyuan-he · commit ced18cbb216a · 2024-11-08T14:04:31.000-08:00
diff --git a/aws-distro-opentelemetry-node-autoinstrumentation/src/patches/aws/services/bedrock.ts b/aws-distro-opentelemetry-node-autoinstrumentation/src/patches/aws/services/bedrock.ts
@@ -386,7 +386,7 @@ export class BedrockRuntimeServiceExtension implements ServiceExtension {
             responseBody.choices[0].finish_reason,
           ]);
         }
-      } else if (currentModelId.includes('mistral.mistral')) {
+      } else if (currentModelId.includes('mistral')) {
         if (responseBody.outputs?.[0]?.text !== undefined) {
           span.setAttribute(
             AwsSpanProcessingUtil.GEN_AI_USAGE_OUTPUT_TOKENS,

Original file line number	Diff line number	Diff line change
`@@ -386,7 +386,7 @@ export class BedrockRuntimeServiceExtension implements ServiceExtension {`
`386`	`386`	`responseBody.choices[0].finish_reason,`
`387`	`387`	`]);`
`388`	`388`	`}`
`389`		`- } else if (currentModelId.includes('mistral.mistral')) {`
	`389`	`+ } else if (currentModelId.includes('mistral')) {`
`390`	`390`	`if (responseBody.outputs?.[0]?.text !== undefined) {`
`391`	`391`	`span.setAttribute(`
`392`	`392`	`AwsSpanProcessingUtil.GEN_AI_USAGE_OUTPUT_TOKENS,`