feat: Update Copilot model capabilities with context size and token limits, and hover details (#165)

Co-authored-by: Copilot <copilot@github.com>

Author: ethanyhou

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

@@ -16,6 +16,7 @@ public class CopilotModel {
   private String modelFamily;
   private String modelName;
   private String id;
+  private String vendor;
   private CopilotModelPolicy modelPolicy;
   private List<String> scopes;
   private boolean preview;
@@ -26,6 +27,7 @@ public class CopilotModel {
   private String degradationReason;
   private String providerName;
   private String modelPickerCategory;
+  private String modelPickerPriceCategory;
 
   /**
    * Policy for the model.
@@ -52,27 +54,65 @@ public String toString() {
     }
   }
 
+  /**
+   * Capabilities limits for the model. All components are optional ({@code null} when the server does not provide a
+   * value), mirroring the {@code number | undefined} fields in the language-server schema.
+   */
+  public record CopilotModelCapabilitiesLimits(Integer maxContextWindowTokens, Integer maxOutputTokens,
+      Integer maxInputTokens, Integer maxNonStreamingOutputTokens) {
+    @Override
+    public String toString() {
+      ToStringBuilder builder = new ToStringBuilder(this);
+      builder.append("maxContextWindowTokens", maxContextWindowTokens);
+      builder.append("maxOutputTokens", maxOutputTokens);
+      builder.append("maxInputTokens", maxInputTokens);
+      builder.append("maxNonStreamingOutputTokens", maxNonStreamingOutputTokens);
+      return builder.toString();
+    }
+  }
+
   /**
    * Capabilities for the model.
    */
-  public record CopilotModelCapabilities(CopilotModelCapabilitiesSupports supports) {
+  public record CopilotModelCapabilities(CopilotModelCapabilitiesSupports supports,
+      CopilotModelCapabilitiesLimits limits) {
     @Override
     public String toString() {
       ToStringBuilder builder = new ToStringBuilder(this);
       builder.append("supports", supports);
+      builder.append("limits", limits);
+      return builder.toString();
+    }
+  }
+
+  /**
+   * Per-token prices for the model, returned in USD.
+   */
+  public record CopilotModelBillingTokenPrices(Double cachePrice, Double inputPrice, Double outputPrice,
+      Double tokenUnit) {
+    @Override
+    public String toString() {
+      ToStringBuilder builder = new ToStringBuilder(this);
+      builder.append("cachePrice", cachePrice);
+      builder.append("inputPrice", inputPrice);
+      builder.append("outputPrice", outputPrice);
+      builder.append("tokenUnit", tokenUnit);
       return builder.toString();
     }
   }
 
   /**
    * Billing for the model.
    */
-  public record CopilotModelBilling(boolean isPremium, double multiplier) {
+  public record CopilotModelBilling(boolean isPremium, double multiplier, boolean tokenBasedBillingEnabled,
+      CopilotModelBillingTokenPrices tokenPrices) {
     @Override
     public String toString() {
       ToStringBuilder builder = new ToStringBuilder(this);
       builder.append("isPremium", isPremium);
       builder.append("multiplier", multiplier);
+      builder.append("tokenBasedBillingEnabled", tokenBasedBillingEnabled);
+      builder.append("tokenPrices", tokenPrices);
       return builder.toString();
     }
   }
@@ -101,6 +141,14 @@ public void setId(String id) {
     this.id = id;
   }
 
+  public String getVendor() {
+    return vendor;
+  }
+
+  public void setVendor(String vendor) {
+    this.vendor = vendor;
+  }
+
   public CopilotModelPolicy getModelPolicy() {
     return modelPolicy;
   }
@@ -181,6 +229,14 @@ public void setModelPickerCategory(String modelPickerCategory) {
     this.modelPickerCategory = modelPickerCategory;
   }
 
+  public String getModelPickerPriceCategory() {
+    return modelPickerPriceCategory;
+  }
+
+  public void setModelPickerPriceCategory(String modelPickerPriceCategory) {
+    this.modelPickerPriceCategory = modelPickerPriceCategory;
+  }
+
   @Override
   public boolean equals(Object obj) {
     if (this == obj) {
@@ -198,14 +254,16 @@ public boolean equals(Object obj) {
         && isChatDefault == other.isChatDefault && isChatFallback == other.isChatFallback
         && Objects.equals(modelFamily, other.modelFamily) && Objects.equals(modelName, other.modelName)
         && Objects.equals(modelPickerCategory, other.modelPickerCategory)
+        && Objects.equals(modelPickerPriceCategory, other.modelPickerPriceCategory)
         && Objects.equals(modelPolicy, other.modelPolicy) && preview == other.preview
-        && Objects.equals(providerName, other.providerName) && Objects.equals(scopes, other.scopes);
+        && Objects.equals(providerName, other.providerName) && Objects.equals(scopes, other.scopes)
+        && Objects.equals(vendor, other.vendor);
   }
 
   @Override
   public int hashCode() {
     return Objects.hash(billing, capabilities, degradationReason, id, isChatDefault, isChatFallback, modelFamily,
-        modelName, modelPickerCategory, modelPolicy, preview, providerName, scopes);
+        modelName, modelPickerCategory, modelPickerPriceCategory, modelPolicy, preview, providerName, scopes, vendor);
   }
 
   @Override
@@ -214,6 +272,7 @@ public String toString() {
     builder.append("modelFamily", modelFamily);
     builder.append("modelName", modelName);
     builder.append("id", id);
+    builder.append("vendor", vendor);
     builder.append("modelPolicy", modelPolicy);
     builder.append("scopes", scopes);
     builder.append("preview", preview);
@@ -224,6 +283,7 @@ public String toString() {
     builder.append("degradationReason", degradationReason);
     builder.append("providerName", providerName);
     builder.append("modelPickerCategory", modelPickerCategory);
+    builder.append("modelPickerPriceCategory", modelPickerPriceCategory);
     return builder.toString();
   }
 

com.microsoft.copilot.eclipse.ui.test/src/com/microsoft/copilot/eclipse/ui/utils/ModelUtilsTests.java

@@ -5,6 +5,7 @@
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNotNull;
+import static org.junit.jupiter.api.Assertions.assertNull;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
 import org.junit.jupiter.api.Test;
@@ -56,4 +57,23 @@ void testConvertByokModelToCopilotModel_withToolCallingCapability() {
     assertTrue(result.getScopes().contains(CopilotScope.CHAT_PANEL));
     assertTrue(result.getScopes().contains(CopilotScope.AGENT_PANEL));
   }
+
+  @Test
+  void testConvertByokModelToCopilotModel_preservesTokenLimits() {
+    ByokModel byokModel = new ByokModel();
+    byokModel.setModelId("gpt-4.1");
+
+    ByokModelCapabilities capabilities = new ByokModelCapabilities();
+    capabilities.setMaxInputTokens(128000);
+    capabilities.setMaxOutputTokens(16000);
+    byokModel.setModelCapabilities(capabilities);
+
+    CopilotModel result = ModelUtils.convertByokModelToCopilotModel(byokModel);
+
+    assertNotNull(result.getCapabilities());
+    assertNotNull(result.getCapabilities().limits());
+    assertNull(result.getCapabilities().limits().maxContextWindowTokens());
+    assertEquals(128000, result.getCapabilities().limits().maxInputTokens());
+    assertEquals(16000, result.getCapabilities().limits().maxOutputTokens());
+  }
 }