Skip to content

Commit 08a8928

Browse files
RobertCraigiestainless-app[bot]
authored andcommitted
feat(vertex): add beta.messages.create()
1 parent b9fa45a commit 08a8928

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

packages/vertex-sdk/src/client.ts

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,7 @@ export class AnthropicVertex extends Core.APIClient {
8383
}
8484

8585
messages: Resources.Messages = new Resources.Messages(this);
86+
beta: BetaResource = makeBetaResource(this);
8687

8788
protected override defaultQuery(): Core.DefaultQuery | undefined {
8889
return this._options.defaultQuery;
@@ -142,3 +143,22 @@ export class AnthropicVertex extends Core.APIClient {
142143
return super.buildRequest(options);
143144
}
144145
}
146+
147+
/**
148+
* The Vertex API does not currently support prompt caching or the Batch API.
149+
*/
150+
type BetaResource = Omit<Resources.Beta, 'promptCaching' | 'messages'> & {
151+
messages: Omit<Resources.Beta['messages'], 'batches'>;
152+
};
153+
154+
function makeBetaResource(client: AnthropicVertex): BetaResource {
155+
const resource = new Resources.Beta(client);
156+
157+
// @ts-expect-error we're deleting non-optional properties
158+
delete resource.promptCaching;
159+
160+
// @ts-expect-error we're deleting non-optional properties
161+
delete resource.messages.batches;
162+
163+
return resource;
164+
}

0 commit comments

Comments
 (0)