Add additional factory methods for ChatMessageAnnotation and token usage details

Co-authored-by: KrzysztofCwalina <[email protected]>

Author: Copilot

src/Custom/ChatModelFactory.cs

@@ -209,4 +209,45 @@ public static StreamingChatToolCallUpdate StreamingChatToolCallUpdate(
             functionName: functionName,
             functionArgumentsUpdate: functionArgumentsUpdate);
     }
+
+    /// <summary> Initializes a new instance of <see cref="OpenAI.Chat.ChatMessageAnnotation"/>. </summary>
+    /// <returns> A new <see cref="OpenAI.Chat.ChatMessageAnnotation"/> instance for mocking. </returns>
+    public static ChatMessageAnnotation ChatMessageAnnotation(
+        int startIndex = default,
+        int endIndex = default,
+        Uri webResourceUri = default,
+        string webResourceTitle = default)
+    {
+        return OpenAI.Chat.OpenAIChatModelFactory.ChatMessageAnnotation(
+            startIndex: startIndex,
+            endIndex: endIndex,
+            webResourceUri: webResourceUri,
+            webResourceTitle: webResourceTitle);
+    }
+
+    /// <summary> Initializes a new instance of <see cref="OpenAI.Chat.ChatInputTokenUsageDetails"/>. </summary>
+    /// <returns> A new <see cref="OpenAI.Chat.ChatInputTokenUsageDetails"/> instance for mocking. </returns>
+    public static ChatInputTokenUsageDetails ChatInputTokenUsageDetails(
+        int audioTokenCount = default, 
+        int cachedTokenCount = default)
+    {
+        return OpenAI.Chat.OpenAIChatModelFactory.ChatInputTokenUsageDetails(
+            audioTokenCount: audioTokenCount,
+            cachedTokenCount: cachedTokenCount);
+    }
+
+    /// <summary> Initializes a new instance of <see cref="OpenAI.Chat.ChatOutputTokenUsageDetails"/>. </summary>
+    /// <returns> A new <see cref="OpenAI.Chat.ChatOutputTokenUsageDetails"/> instance for mocking. </returns>
+    public static ChatOutputTokenUsageDetails ChatOutputTokenUsageDetails(
+        int reasoningTokenCount = default, 
+        int audioTokenCount = default, 
+        int acceptedPredictionTokenCount = default, 
+        int rejectedPredictionTokenCount = default)
+    {
+        return OpenAI.Chat.OpenAIChatModelFactory.ChatOutputTokenUsageDetails(
+            reasoningTokenCount: reasoningTokenCount,
+            audioTokenCount: audioTokenCount,
+            acceptedPredictionTokenCount: acceptedPredictionTokenCount,
+            rejectedPredictionTokenCount: rejectedPredictionTokenCount);
+    }
 }

tests/Chat/ChatModelFactoryTests.cs

@@ -292,4 +292,58 @@ public void StreamingChatToolCallUpdateWorks()
         Assert.That(update.FunctionName, Is.EqualTo(functionName));
         Assert.That(update.FunctionArgumentsUpdate, Is.EqualTo(functionArgumentsUpdate));
     }
+
+    [Test]
+    public void ChatMessageAnnotationWorks()
+    {
+        int startIndex = 5;
+        int endIndex = 10;
+        Uri webResourceUri = new Uri("https://example.com");
+        string webResourceTitle = "Example Resource";
+
+        ChatMessageAnnotation annotation = ChatModelFactory.ChatMessageAnnotation(
+            startIndex: startIndex,
+            endIndex: endIndex,
+            webResourceUri: webResourceUri,
+            webResourceTitle: webResourceTitle);
+
+        Assert.That(annotation.StartIndex, Is.EqualTo(startIndex));
+        Assert.That(annotation.EndIndex, Is.EqualTo(endIndex));
+        Assert.That(annotation.WebResourceUri, Is.EqualTo(webResourceUri));
+        Assert.That(annotation.WebResourceTitle, Is.EqualTo(webResourceTitle));
+    }
+
+    [Test]
+    public void ChatInputTokenUsageDetailsWorks()
+    {
+        int audioTokenCount = 25;
+        int cachedTokenCount = 75;
+
+        ChatInputTokenUsageDetails details = ChatModelFactory.ChatInputTokenUsageDetails(
+            audioTokenCount: audioTokenCount,
+            cachedTokenCount: cachedTokenCount);
+
+        Assert.That(details.AudioTokenCount, Is.EqualTo(audioTokenCount));
+        Assert.That(details.CachedTokenCount, Is.EqualTo(cachedTokenCount));
+    }
+
+    [Test]
+    public void ChatOutputTokenUsageDetailsWorks()
+    {
+        int reasoningTokenCount = 30;
+        int audioTokenCount = 15;
+        int acceptedPredictionTokenCount = 5;
+        int rejectedPredictionTokenCount = 2;
+
+        ChatOutputTokenUsageDetails details = ChatModelFactory.ChatOutputTokenUsageDetails(
+            reasoningTokenCount: reasoningTokenCount,
+            audioTokenCount: audioTokenCount,
+            acceptedPredictionTokenCount: acceptedPredictionTokenCount,
+            rejectedPredictionTokenCount: rejectedPredictionTokenCount);
+
+        Assert.That(details.ReasoningTokenCount, Is.EqualTo(reasoningTokenCount));
+        Assert.That(details.AudioTokenCount, Is.EqualTo(audioTokenCount));
+        Assert.That(details.AcceptedPredictionTokenCount, Is.EqualTo(acceptedPredictionTokenCount));
+        Assert.That(details.RejectedPredictionTokenCount, Is.EqualTo(rejectedPredictionTokenCount));
+    }
 }