| | | 1 | | using OpenAI.Chat; |
| | | 2 | | |
| | | 3 | | namespace TestUtilities; |
| | | 4 | | |
| | | 5 | | /// <summary> |
| | | 6 | | /// Helper methods for creating OpenAI test objects |
| | | 7 | | /// </summary> |
| | | 8 | | public static class OpenAITestHelpers |
| | | 9 | | { |
| | | 10 | | /// <summary> |
| | | 11 | | /// Creates a ChatTokenUsage instance for testing purposes |
| | | 12 | | /// </summary> |
| | | 13 | | /// <param name="inputTokens">The number of input tokens</param> |
| | | 14 | | /// <param name="outputTokens">The number of output tokens</param> |
| | | 15 | | /// <param name="cachedInputTokens">The number of cached input tokens (optional)</param> |
| | | 16 | | /// <param name="outputReasoningTokens">The number of output reasoning tokens (optional)</param> |
| | | 17 | | /// <returns>A ChatTokenUsage instance with the specified token counts</returns> |
| | | 18 | | public static ChatTokenUsage CreateChatTokenUsage( |
| | | 19 | | int inputTokens, |
| | | 20 | | int outputTokens, |
| | | 21 | | int cachedInputTokens = 0, |
| | | 22 | | int outputReasoningTokens = 0) |
| | | 23 | | { |
| | 1 | 24 | | ChatInputTokenUsageDetails? inputDetails = cachedInputTokens > 0 |
| | 1 | 25 | | ? OpenAIChatModelFactory.ChatInputTokenUsageDetails(cachedTokenCount: cachedInputTokens) |
| | 1 | 26 | | : null; |
| | | 27 | | |
| | 1 | 28 | | ChatOutputTokenUsageDetails? outputDetails = outputReasoningTokens > 0 |
| | 1 | 29 | | ? OpenAIChatModelFactory.ChatOutputTokenUsageDetails(reasoningTokenCount: outputReasoningTokens) |
| | 1 | 30 | | : null; |
| | | 31 | | |
| | 1 | 32 | | return OpenAIChatModelFactory.ChatTokenUsage( |
| | 1 | 33 | | inputTokenCount: inputTokens, |
| | 1 | 34 | | outputTokenCount: outputTokens, |
| | 1 | 35 | | inputTokenDetails: inputDetails, |
| | 1 | 36 | | outputTokenDetails: outputDetails); |
| | | 37 | | } |
| | | 38 | | } |