Missed a spot

newrelic · Sep 4, 2024 · 2f215cb · 2f215cb
1 parent 71c905e
commit 2f215cb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/Agent/IntegrationTests/IntegrationTests/LLM/BedrockTests.cs b/tests/Agent/IntegrationTests/IntegrationTests/LLM/BedrockTests.cs
@@ -112,7 +112,7 @@ public void BedrockTest()
         {
             var expectedMetrics = new List<Assertions.ExpectedMetric>
             {
-                new Assertions.ExpectedMetric { metricName = @"Custom/Llm/completion/Bedrock/InvokeModelAsync", CallCountAllHarvests = 5 },
+                new Assertions.ExpectedMetric { metricName = @"Custom/Llm/completion/Bedrock/InvokeModelAsync", CallCountAllHarvests = _bedrockModelsToTest.Count - 1 },
                 new Assertions.ExpectedMetric { metricName = @"Custom/Llm/embedding/Bedrock/InvokeModelAsync", CallCountAllHarvests = 1 },
                 new Assertions.ExpectedMetric { metricName = @"Supportability/DotNet/ML/.*", IsRegexName = true}
             };