add ALL of the interfaces

Author: cwaldren-ld

pkgs/sdk/server-ai/src/Interfaces/ILdAiClient.cs

@@ -11,7 +11,7 @@ public interface ILdAiClient
 
     /// <summary>
     /// Retrieves a LaunchDarkly AI config identified by the given key. The return value
-    /// is an <see cref="LdAiConfigTracker"/>, which makes the configuration available and
+    /// is an <see cref="ILdAiConfigTracker"/>, which makes the configuration available and
     /// provides convenience methods for generating events related to model usage.
     ///
     /// Any variables provided will be interpolated into the prompt's messages.
@@ -24,6 +24,6 @@ public interface ILdAiClient
     /// <param name="defaultValue">the default config, if unable to retrieve from LaunchDarkly</param>
     /// <param name="variables">the list of variables used when interpolating the prompt</param>
     /// <returns>an AI config tracker</returns>
-    public LdAiConfigTracker ModelConfig(string key, Context context, LdAiConfig defaultValue,
+    public ILdAiConfigTracker ModelConfig(string key, Context context, LdAiConfig defaultValue,
         IReadOnlyDictionary<string, object> variables = null);
 }

pkgs/sdk/server-ai/src/Interfaces/ILdAiConfigTracker.cs

@@ -0,0 +1,57 @@
+using System;
+using System.Threading.Tasks;
+using LaunchDarkly.Sdk.Server.Ai.Config;
+using LaunchDarkly.Sdk.Server.Ai.Metrics;
+
+namespace LaunchDarkly.Sdk.Server.Ai.Interfaces;
+
+/// <summary>
+/// Represents the interface of the AI Config Tracker, useful for mocking.
+/// </summary>
+public interface ILdAiConfigTracker
+{
+    /// <summary>
+    /// The retrieved AI model configuration.
+    /// </summary>
+    public LdAiConfig Config { get; }
+
+    /// <summary>
+    /// Tracks a duration metric related to this config.
+    /// </summary>
+    /// <param name="durationMs">the duration in milliseconds</param>
+    public void TrackDuration(float durationMs);
+
+    /// <summary>
+    /// Tracks the duration of a task, and returns the result of the task.
+    /// </summary>
+    /// <param name="task">the task</param>
+    /// <typeparam name="T">type of the task's result</typeparam>
+    /// <returns>the task</returns>
+    public Task<T> TrackDurationOfTask<T>(Task<T> task);
+
+    /// <summary>
+    /// Tracks feedback (positive or negative) related to the output of the model.
+    /// </summary>
+    /// <param name="feedback">the feedback</param>
+    /// <exception cref="ArgumentOutOfRangeException">thrown if the feedback value is not <see cref="Feedback.Positive"/> or <see cref="Feedback.Negative"/></exception>
+    public void TrackFeedback(Feedback feedback);
+
+    /// <summary>
+    /// Tracks a generation event related to this config.
+    /// </summary>
+    public void TrackSuccess();
+
+    /// <summary>
+    /// Tracks a request to a provider. The request is a task that returns a <see cref="ProviderResponse"/>, which
+    /// contains information about the request such as token usage and metrics.
+    /// </summary>
+    /// <param name="request">a task representing the request</param>
+    /// <returns>the task</returns>
+    public Task<ProviderResponse> TrackRequest(Task<ProviderResponse> request);
+
+    /// <summary>
+    /// Tracks token usage related to this config.
+    /// </summary>
+    /// <param name="usage">the usage</param>
+    public void TrackTokens(Usage usage);
+}

pkgs/sdk/server-ai/src/LdAiClient.cs

@@ -43,7 +43,7 @@ public LdAiClient(ILaunchDarklyClient client)
     private const string LdContextVariable = "ldctx";
 
     /// <inheritdoc/>
-    public LdAiConfigTracker ModelConfig(string key, Context context, LdAiConfig defaultValue,
+    public ILdAiConfigTracker ModelConfig(string key, Context context, LdAiConfig defaultValue,
         IReadOnlyDictionary<string, object> variables = null)
     {
 

pkgs/sdk/server-ai/src/LdAiConfigTracker.cs

@@ -11,14 +11,8 @@ namespace LaunchDarkly.Sdk.Server.Ai;
 /// <summary>
 /// A tracker capable of reporting events related to a particular AI configuration.
 /// </summary>
-public class LdAiConfigTracker
+public class LdAiConfigTracker : ILdAiConfigTracker
 {
-    /// <summary>
-    /// The retrieved AI model configuration.
-    /// </summary>
-    public readonly LdAiConfig Config;
-
-
     private readonly ILaunchDarklyClient _client;
     private readonly Context _context;
     private readonly LdValue _trackData;
@@ -52,20 +46,16 @@ public LdAiConfigTracker(ILaunchDarklyClient client, string configKey, LdAiConfi
         });
     }
 
-    /// <summary>
-    /// Tracks a duration metric related to this config.
-    /// </summary>
-    /// <param name="durationMs">the duration in milliseconds</param>
+
+    /// <inheritdoc/>
+    public LdAiConfig Config { get; }
+
+    /// <inheritdoc/>
     public void TrackDuration(float durationMs) =>
         _client.Track(Duration, _context, _trackData, durationMs);
 
 
-    /// <summary>
-    /// Tracks the duration of a task, and returns the result of the task.
-    /// </summary>
-    /// <param name="task">the task</param>
-    /// <typeparam name="T">type of the task's result</typeparam>
-    /// <returns>the task</returns>
+    /// <inheritdoc/>
     public async Task<T> TrackDurationOfTask<T>(Task<T> task)
     {
         var result = await MeasureDurationOfTaskMs(task);
@@ -81,11 +71,7 @@ private static async Task<Tuple<T, long>> MeasureDurationOfTaskMs<T>(Task<T> tas
         return Tuple.Create(result, sw.ElapsedMilliseconds);
     }
 
-    /// <summary>
-    /// Tracks feedback (positive or negative) related to the output of the model.
-    /// </summary>
-    /// <param name="feedback">the feedback</param>
-    /// <exception cref="ArgumentOutOfRangeException">thrown if the feedback value is not <see cref="Feedback.Positive"/> or <see cref="Feedback.Negative"/></exception>
+    /// <inheritdoc/>
     public void TrackFeedback(Feedback feedback)
     {
         switch (feedback)
@@ -101,21 +87,13 @@ public void TrackFeedback(Feedback feedback)
         }
     }
 
-    /// <summary>
-    /// Tracks a generation event related to this config.
-    /// </summary>
+    /// <inheritdoc/>
     public void TrackSuccess()
     {
         _client.Track(Generation, _context, _trackData, 1);
     }
 
-
-    /// <summary>
-    /// Tracks a request to a provider. The request is a task that returns a <see cref="ProviderResponse"/>, which
-    /// contains information about the request such as token usage and metrics.
-    /// </summary>
-    /// <param name="request">a task representing the request</param>
-    /// <returns>the task</returns>
+    /// <inheritdoc/>
     public async Task<ProviderResponse> TrackRequest(Task<ProviderResponse> request)
     {
         var (result, durationMs) = await MeasureDurationOfTaskMs(request);
@@ -131,10 +109,7 @@ public async Task<ProviderResponse> TrackRequest(Task<ProviderResponse> request)
         return result;
     }
 
-    /// <summary>
-    /// Tracks token usage related to this config.
-    /// </summary>
-    /// <param name="usage">the usage</param>
+    /// <inheritdoc/>
     public void TrackTokens(Usage usage)
     {
         if (usage.Total is > 0)