feat: Implement quota warning handling and UI updates for TBB

Author: ethanyhou

com.microsoft.copilot.eclipse.core.test/src/com/microsoft/copilot/eclipse/core/lsp/CopilotLanguageClientTests.java

@@ -141,7 +141,8 @@ void testOnDidChangeFeatureFlagsWithEmptyFeatureFlags() {
 
   @Test
   void testOnQuotaWarning_PostsNotificationToEventBroker() {
-    QuotaWarningNotification notification = new QuotaWarningNotification("Approaching quota", 90.0);
+    QuotaWarningNotification notification = new QuotaWarningNotification(
+        "Copilot Quota Usage Alert", "Approaching quota", "warning", null, null);
     setEventBroker(eventBroker);
 
     client.onQuotaWarning(notification);

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/lsp/protocol/ChatCreateResult.java

@@ -6,15 +6,12 @@
 import java.util.Objects;
 
 import org.apache.commons.lang3.builder.ToStringBuilder;
-import org.eclipse.lsp4j.jsonrpc.validation.NonNull;
 
 /**
  * Result of a chat creation.
  */
 public class ChatCreateResult {
-  @NonNull
   private String conversationId;
-  @NonNull
   private String turnId;
   private String agentSlug;
   private String modelName;

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/lsp/protocol/ChatProgressValue.java

@@ -101,6 +101,15 @@ public String getErrorReason() {
     return error != null ? error.getReason() : null;
   }
 
+  /**
+   * Returns the BYOK model-provider name from the error payload, when present.
+   *
+   * @return the name of the BYOK model provider that produced the error, or {@code null} for built-in models.
+   */
+  public String getErrorModelProviderName() {
+    return error != null ? error.getModelProviderName() : null;
+  }
+
   public List<AgentRound> getAgentRounds() {
     return editAgentRounds;
   }

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/lsp/protocol/ChatTurnResult.java

@@ -6,15 +6,12 @@
 import java.util.Objects;
 
 import org.apache.commons.lang3.builder.ToStringBuilder;
-import org.eclipse.lsp4j.jsonrpc.validation.NonNull;
 
 /**
  * Result of a chat turn.
  */
 public class ChatTurnResult {
-  @NonNull
   private String conversationId;
-  @NonNull
   private String turnId;
   private String agentSlug;
   private String modelName;

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/lsp/protocol/ConversationError.java

@@ -19,6 +19,7 @@ public class ConversationError {
   private String reason;
   private boolean responseIsIncomplete;
   private boolean responseIsFiltered;
+  private String modelProviderName;
 
   public void setMessage(String message) {
     this.message = message;
@@ -40,6 +41,10 @@ public void setResponseIsFiltered(boolean responseIsFiltered) {
     this.responseIsFiltered = responseIsFiltered;
   }
 
+  public void setModelProviderName(String modelProviderName) {
+    this.modelProviderName = modelProviderName;
+  }
+
   public String getMessage() {
     return message;
   }
@@ -60,9 +65,17 @@ public boolean getResponseIsFiltered() {
     return responseIsFiltered;
   }
 
+  /**
+   * The name of the model provider that produced the error, when the failing request was routed to a custom
+   * Bring-Your-Own-Key (BYOK) model. {@code null} or blank for built-in Copilot models.
+   */
+  public String getModelProviderName() {
+    return modelProviderName;
+  }
+
   @Override
   public int hashCode() {
-    return Objects.hash(message, code, reason, responseIsIncomplete, responseIsFiltered);
+    return Objects.hash(message, code, reason, responseIsIncomplete, responseIsFiltered, modelProviderName);
   }
 
   @Override
@@ -75,7 +88,8 @@ public boolean equals(Object o) {
     }
     ConversationError that = (ConversationError) o;
     return Objects.equals(message, that.message) && code == that.code && Objects.equals(reason, that.reason)
-        && responseIsIncomplete == that.responseIsIncomplete && responseIsFiltered == that.responseIsFiltered;
+        && responseIsIncomplete == that.responseIsIncomplete && responseIsFiltered == that.responseIsFiltered
+        && Objects.equals(modelProviderName, that.modelProviderName);
   }
 
   @Override
@@ -86,6 +100,7 @@ public String toString() {
     builder.append("reason", reason);
     builder.append("responseIsIncomplete", responseIsIncomplete);
     builder.append("responseIsFiltered", responseIsFiltered);
+    builder.append("modelProviderName", modelProviderName);
     return builder.toString();
   }
 }

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/lsp/protocol/quota/QuotaWarningNotification.java

@@ -3,9 +3,46 @@
 
 package com.microsoft.copilot.eclipse.core.lsp.protocol.quota;
 
+import com.google.gson.annotations.SerializedName;
+
 /**
- * Parameters for the "copilot/quotaWarning" notification. Sent by the language server when the user's AI quota exceeds
- * the warning threshold.
+ * Parameters for the {@code copilot/quotaWarning} notification. Sent by the language server when the user's AI quota
+ * exceeds a warning threshold.
+ *
+ * @param title the popup title supplied by the language server
+ * @param message the popup body message
+ * @param severity the language-server severity hint (e.g. {@code "info"} or {@code "warning"}); used by the client to
+ *     decide which icon to render on the banner. May be {@code null}.
+ * @param copilotPlan the user's Copilot plan
+ * @param premiumInteractions the premium-interactions snapshot for the warning, or {@code null} when the language
+ *     server does not include it
  */
-public record QuotaWarningNotification(String message, double percentUsed) {
+public record QuotaWarningNotification(
+    String title,
+    String message,
+    String severity,
+    CopilotPlan copilotPlan,
+    @SerializedName("premium_interactions") PremiumInteractions premiumInteractions) {
+
+  /**
+   * Premium-interactions snapshot embedded in a {@link QuotaWarningNotification}. The shape is dictated by the
+   * language server and differs from {@link Quota}.
+   *
+   * @param quota total monthly premium-interactions allowance
+   * @param used premium interactions consumed so far this period
+   * @param percentRemaining percentage of the allowance remaining
+   * @param overageUsed additional paid interactions consumed beyond the allowance
+   * @param overageEnabled whether the user has enabled paid overage
+   * @param resetDate ISO-8601 instant when the monthly allowance resets, or {@code null}
+   * @param unlimited whether this quota has no monthly limit
+   */
+  public record PremiumInteractions(
+      double quota,
+      double used,
+      double percentRemaining,
+      double overageUsed,
+      boolean overageEnabled,
+      String resetDate,
+      boolean unlimited) {
+  }
 }

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/persistence/ConversationDataFactory.java

@@ -179,6 +179,7 @@ private void applyConversationError(ReplyData reply, ConversationError error) {
     ErrorData errorData = new ErrorData();
     errorData.setMessage(error.getMessage());
     errorData.setCode(error.getCode());
+    errorData.setModelProviderName(error.getModelProviderName());
     ErrorMessageData em = new ErrorMessageData();
     em.setError(errorData);
     reply.getErrorMessages().add(em);

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/persistence/CopilotTurnData.java

@@ -440,6 +440,7 @@ public String toString() {
   public static class ErrorData {
     private String message;
     private int code;
+    private String modelProviderName;
     private Map<String, Object> data;
 
     public String getMessage() {
@@ -458,6 +459,18 @@ public void setCode(int code) {
       this.code = code;
     }
 
+    /**
+     * The BYOK model provider responsible for the error, or {@code null} when the failing model was a
+     * built-in Copilot model.
+     */
+    public String getModelProviderName() {
+      return modelProviderName;
+    }
+
+    public void setModelProviderName(String modelProviderName) {
+      this.modelProviderName = modelProviderName;
+    }
+
     public Map<String, Object> getData() {
       return data;
     }
@@ -468,7 +481,7 @@ public void setData(Map<String, Object> data) {
 
     @Override
     public int hashCode() {
-      return Objects.hash(code, data, message);
+      return Objects.hash(code, data, message, modelProviderName);
     }
 
     @Override
@@ -483,14 +496,16 @@ public boolean equals(Object obj) {
         return false;
       }
       ErrorData other = (ErrorData) obj;
-      return code == other.code && Objects.equals(data, other.data) && Objects.equals(message, other.message);
+      return code == other.code && Objects.equals(data, other.data) && Objects.equals(message, other.message)
+          && Objects.equals(modelProviderName, other.modelProviderName);
     }
 
     @Override
     public String toString() {
       ToStringBuilder builder = new ToStringBuilder(this);
       builder.append("message", message);
       builder.append("code", code);
+      builder.append("modelProviderName", modelProviderName);
       builder.append("data", data);
       return builder.toString();
     }

com.microsoft.copilot.eclipse.ui.test/src/com/microsoft/copilot/eclipse/ui/i18n/MessagesTest.java

@@ -14,8 +14,5 @@ void testMessagesInitialization() {
     // Ensure that the static fields are initialized
     assertNotNull(Messages.menu_signToGitHub);
     assertNotNull(Messages.menu_signOutOfGitHub);
-    assertNotNull(Messages.quotaWarning_title);
-    assertNotNull(Messages.quotaWarning_closeButton);
-    assertNotNull(Messages.quotaWarning_increaseBudgetButton);
   }
-}
+}