feat: [TBB] Implement quota warning handling and UI updates for TBB

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

@@ -10,12 +10,14 @@
 import static org.mockito.Mockito.verify;
 import static org.mockito.Mockito.when;
 
+import java.lang.reflect.Field;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.ExecutionException;
 
 import org.eclipse.core.resources.IFile;
+import org.eclipse.e4.core.services.events.IEventBroker;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
@@ -28,10 +30,12 @@
 import com.microsoft.copilot.eclipse.core.FeatureFlags;
 import com.microsoft.copilot.eclipse.core.chat.service.IChatServiceManager;
 import com.microsoft.copilot.eclipse.core.chat.service.IReferencedFileService;
+import com.microsoft.copilot.eclipse.core.events.CopilotEventConstants;
 import com.microsoft.copilot.eclipse.core.lsp.protocol.ConversationCapabilities;
 import com.microsoft.copilot.eclipse.core.lsp.protocol.ConversationContextParams;
 import com.microsoft.copilot.eclipse.core.lsp.protocol.CurrentEditorContext;
 import com.microsoft.copilot.eclipse.core.lsp.protocol.DidChangeFeatureFlagsParams;
+import com.microsoft.copilot.eclipse.core.lsp.protocol.quota.QuotaWarningNotification;
 import com.microsoft.copilot.eclipse.core.utils.FileUtils;
 
 @ExtendWith(MockitoExtension.class)
@@ -48,6 +52,9 @@ class CopilotLanguageClientTests {
   @Mock
   private IReferencedFileService fileService;
 
+  @Mock
+  private IEventBroker eventBroker;
+
   @BeforeEach
   void setUp() {
     client = new CopilotLanguageClient();
@@ -131,4 +138,25 @@ void testOnDidChangeFeatureFlagsWithEmptyFeatureFlags() {
       verify(mockFeatureFlags).setByokEnabled(true);
     }
   }
+
+  @Test
+  void testOnQuotaWarning_PostsNotificationToEventBroker() {
+    QuotaWarningNotification notification = new QuotaWarningNotification(
+        "Copilot Quota Usage Alert", "Approaching quota", "warning", null, null);
+    setEventBroker(eventBroker);
+
+    client.onQuotaWarning(notification);
+
+    verify(eventBroker).post(CopilotEventConstants.TOPIC_QUOTA_WARNING, notification);
+  }
+
+  private void setEventBroker(IEventBroker broker) {
+    try {
+      Field field = CopilotLanguageClient.class.getDeclaredField("eventBroker");
+      field.setAccessible(true);
+      field.set(client, broker);
+    } catch (ReflectiveOperationException e) {
+      throw new AssertionError("Failed to inject event broker", e);
+    }
+  }
 }

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/events/CopilotEventConstants.java

@@ -30,6 +30,11 @@ public class CopilotEventConstants {
    */
   private static final String TOPIC_MCP = TOPIC_BASE + "MCP/";
 
+  /**
+   * Topic for quota events.
+   */
+  private static final String TOPIC_QUOTA = TOPIC_BASE + "QUOTA/";
+
   /**
    * Topic for Next Edit Suggestion (NES) events.
    */
@@ -161,6 +166,11 @@ public class CopilotEventConstants {
    */
   public static final String TOPIC_RATE_LIMIT_WARNING = TOPIC_CHAT + "RATE_LIMIT_WARNING";
 
+  /**
+   * Event when a quota warning notification is received from the language server.
+   */
+  public static final String TOPIC_QUOTA_WARNING = TOPIC_QUOTA + "WARNING";
+
   /**
    * Event when custom prompts, skills, agents, or instructions change on the language server. Clients should re-fetch
    * conversation templates on receipt.

com.microsoft.copilot.eclipse.core/src/com/microsoft/copilot/eclipse/core/lsp/CopilotLanguageClient.java

@@ -64,6 +64,7 @@
 import com.microsoft.copilot.eclipse.core.lsp.protocol.codingagent.CodingAgentMessageRequestParams;
 import com.microsoft.copilot.eclipse.core.lsp.protocol.codingagent.CodingAgentMessageResult;
 import com.microsoft.copilot.eclipse.core.lsp.protocol.policy.DidChangePolicyParams;
+import com.microsoft.copilot.eclipse.core.lsp.protocol.quota.QuotaWarningNotification;
 import com.microsoft.copilot.eclipse.core.utils.FileUtils;
 import com.microsoft.copilot.eclipse.core.utils.PlatformUtils;
 
@@ -353,6 +354,17 @@ public CompletableFuture<CodingAgentMessageResult> onCodingAgentMessage(CodingAg
     return CompletableFuture.completedFuture(result);
   }
 
+  /**
+   * Notify when a quota warning is received from the language server.
+   */
+  @JsonNotification("copilot/quotaWarning")
+  public void onQuotaWarning(QuotaWarningNotification notification) {
+    CopilotCore.LOGGER.info("Quota warning received: " + notification);
+    if (eventBroker != null) {
+      eventBroker.post(CopilotEventConstants.TOPIC_QUOTA_WARNING, notification);
+    }
+  }
+
   /**
    * Reads the contents and stats of a file given its URI.
    */

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/IntervalQuota.java

@@ -0,0 +1,17 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+
+package com.microsoft.copilot.eclipse.core.lsp.protocol.quota;
+
+/**
+ * Interval-based quota information, used for immediateUsageInterval and extendedUsageInterval.
+ */
+public record IntervalQuota(
+    double percentRemaining,
+    boolean unlimited,
+    boolean overagePermitted,
+    Integer entitlement,
+    Integer quotaRemaining,
+    String timeStamp,
+    String resetAt) {
+}

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

@@ -0,0 +1,20 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+
+package com.microsoft.copilot.eclipse.core.lsp.protocol.quota;
+
+import com.google.gson.annotations.SerializedName;
+
+/**
+ * Parameters for the {@code copilot/quotaChange} notification, sent by the language server
+ * whenever the user's quota usage changes.
+ *
+ * @param chat current chat quota snapshot, when available
+ * @param completions current completions quota snapshot, when available
+ * @param premiumInteractions current premium interactions quota snapshot, when available
+ * @param copilotPlan the user's Copilot plan (e.g. free, individual, individual_pro, individual_max,
+ *     business, enterprise)
+ */
+public record QuotaChangeParams(QuotaSnapshotParams chat, QuotaSnapshotParams completions,
+    @SerializedName("premium_interactions") QuotaSnapshotParams premiumInteractions, String copilotPlan) {
+}

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

@@ -0,0 +1,48 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+
+package com.microsoft.copilot.eclipse.core.lsp.protocol.quota;
+
+import com.google.gson.annotations.SerializedName;
+
+/**
+ * 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 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/lsp/protocol/quota/QuotaWarningParams.java

@@ -0,0 +1,23 @@
+// Copyright (c) Microsoft Corporation.
+// Licensed under the MIT license.
+
+package com.microsoft.copilot.eclipse.core.lsp.protocol.quota;
+
+import com.google.gson.annotations.SerializedName;
+
+/**
+ * Parameters for the {@code copilot/quotaWarning} notification, sent by the language server when
+ * the user crosses a quota usage threshold or starts consuming overages.
+ *
+ * @param title warning title (e.g. "Copilot Quota Usage Alert")
+ * @param message human-readable warning message
+ * @param severity severity level, either {@code "warning"} or {@code "info"}
+ * @param chat current chat quota snapshot, when available
+ * @param completions current completions quota snapshot, when available
+ * @param premiumInteractions current premium interactions quota snapshot, when available
+ * @param copilotPlan the user's Copilot plan
+ */
+public record QuotaWarningParams(String title, String message, String severity, QuotaSnapshotParams chat,
+    QuotaSnapshotParams completions,
+    @SerializedName("premium_interactions") QuotaSnapshotParams premiumInteractions, String copilotPlan) {
+}

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

@@ -15,4 +15,4 @@ void testMessagesInitialization() {
     assertNotNull(Messages.menu_signToGitHub);
     assertNotNull(Messages.menu_signOutOfGitHub);
   }
-}
+}