fix(vertex): add beta.messages.countTokens method

anthropics · Jan 21, 2025 · 2cbeabc · 2cbeabc
1 parent e670066
commit 2cbeabc
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/packages/vertex-sdk/src/client.ts b/packages/vertex-sdk/src/client.ts
@@ -179,23 +179,17 @@ function makeMessagesResource(client: AnthropicVertex): MessagesResource {
 }
 
 /**
- * The Vertex API does not currently support prompt caching, token counting or the Batch API.
+ * The Vertex API does not currently support the Batch API.
  */
-type BetaResource = Omit<Resources.Beta, 'promptCaching' | 'messages'> & {
-  messages: Omit<Resources.Beta['messages'], 'batches' | 'countTokens'>;
+type BetaResource = Omit<Resources.Beta, 'messages'> & {
+  messages: Omit<Resources.Beta['messages'], 'batches'>;
 };
 
 function makeBetaResource(client: AnthropicVertex): BetaResource {
   const resource = new Resources.Beta(client);
 
-  // @ts-expect-error we're deleting non-optional properties
-  delete resource.promptCaching;
-
   // @ts-expect-error we're deleting non-optional properties
   delete resource.messages.batches;
 
-  // @ts-expect-error we're deleting non-optional properties
-  delete resource.messages.countTokens;
-
   return resource;
 }