test-utils/org.jetbrains.ai.tracy.test.utils/BaseAITracingTest

BaseAITracingTest

@TestInstance(value = TestInstance.Lifecycle.PER_CLASS)

abstract class BaseAITracingTest : BaseOpenTelemetryTracingTest(source)

Constructors

BaseAITracingTest

constructor()

Types

protected object Companion

Functions

protected fun analyzeSpans(): List<SpanData>

assertTracesCount

protected fun assertTracesCount(expectedCount: Int, traces: List<SpanData>)

assumeTracesCount

protected fun assumeTracesCount(assumedCount: Int, traces: List<SpanData>)

enableTracingBeforeTest

@BeforeEach

fun enableTracingBeforeTest()

flushTracesAndAssumeToolCalled

protected fun <T> flushTracesAndAssumeToolCalled(response: T, toolName: String, containsToolCall: (T, String) -> Boolean)

protected fun SpanData.getAttribute(spanAttributeKey: TracySpanAttributes): String?

provideContentCapturePolicies

protected fun provideContentCapturePolicies(): Stream<Arguments>

provideFilesForUpload

protected fun provideFilesForUpload(): Stream<Arguments>

Expects sample.pdf file under resources directory of the test module.

provideImagesForUpload

protected fun provideImagesForUpload(): Stream<Arguments>

Expects image.jpg image under resources directory of the test module.

protected fun readResource(filepath: String): InputStream

@AfterEach

fun resetExporter()

@BeforeAll

fun setupTelemetry()

shutdownTelemetry

@AfterAll

fun shutdownTelemetry()

unquoteAndUnescapeNewlines

protected fun String.unquoteAndUnescapeNewlines(): String

Removes enclosing double quotes from the string if present and replaces escaped newline characters ("\n") with actual newlines.

validateBasicTracing

protected fun validateBasicTracing(url: String, model: String)

verifyMediaContentUploadAttributes

protected fun verifyMediaContentUploadAttributes(span: SpanData, expected: List<MediaContentAttributeValues>)