|
9 | 9 | import static io.opentelemetry.sdk.testing.assertj.OpenTelemetryAssertions.satisfies; |
10 | 10 | import static io.opentelemetry.semconv.incubating.GenAiIncubatingAttributes.GEN_AI_OPERATION_NAME; |
11 | 11 | import static io.opentelemetry.semconv.incubating.GenAiIncubatingAttributes.GEN_AI_PROVIDER_NAME; |
| 12 | +import static io.opentelemetry.semconv.incubating.GenAiIncubatingAttributes.GEN_AI_REQUEST_CHOICE_COUNT; |
12 | 13 | import static io.opentelemetry.semconv.incubating.GenAiIncubatingAttributes.GEN_AI_REQUEST_MODEL; |
13 | 14 | import static io.opentelemetry.semconv.incubating.GenAiIncubatingAttributes.GEN_AI_RESPONSE_FINISH_REASONS; |
14 | 15 | import static io.opentelemetry.semconv.incubating.GenAiIncubatingAttributes.GEN_AI_RESPONSE_ID; |
@@ -215,6 +216,7 @@ void multipleChoicesNoCaptureContent() { |
215 | 216 | equalTo(GEN_AI_REQUEST_MODEL, TEST_CHAT_MODEL), |
216 | 217 | satisfies(GEN_AI_RESPONSE_ID, id -> id.startsWith("chatcmpl-")), |
217 | 218 | equalTo(GEN_AI_RESPONSE_MODEL, TEST_CHAT_RESPONSE_MODEL), |
| 219 | + equalTo(GEN_AI_REQUEST_CHOICE_COUNT, 2L), |
218 | 220 | satisfies( |
219 | 221 | GEN_AI_RESPONSE_FINISH_REASONS, |
220 | 222 | reasons -> reasons.containsExactly("stop", "stop")), |
@@ -711,6 +713,7 @@ void streamMultipleChoicesNoCaptureContent() { |
711 | 713 | equalTo(GEN_AI_REQUEST_MODEL, TEST_CHAT_MODEL), |
712 | 714 | satisfies(GEN_AI_RESPONSE_ID, id -> id.startsWith("chatcmpl-")), |
713 | 715 | equalTo(GEN_AI_RESPONSE_MODEL, TEST_CHAT_RESPONSE_MODEL), |
| 716 | + equalTo(GEN_AI_REQUEST_CHOICE_COUNT, 2L), |
714 | 717 | satisfies( |
715 | 718 | GEN_AI_RESPONSE_FINISH_REASONS, |
716 | 719 | reasons -> reasons.containsExactly("stop", "stop")))))); |
|
0 commit comments