pacnpal/Roo-Code
1
0
Fork 0
mirror of https://github.com/pacnpal/Roo-Code.git synced 2025-12-20 12:21:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

Enable prompt caching for Opus

Browse Source
This commit is contained in:
Saoud Rizwan
2024-08-24 22:08:02 -04:00
parent 6739e4536a
commit 5dcf836d81
3 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/api/anthropic.ts
View File

@@ -19,6 +19,7 @@ export class AnthropicHandler implements ApiHandler {
const modelId = this.getModel().id const modelId = this.getModel().id
switch (modelId) { switch (modelId) {
case "claude-3-5-sonnet-20240620": case "claude-3-5-sonnet-20240620":
case "claude-3-opus-20240229":
case "claude-3-haiku-20240307": case "claude-3-haiku-20240307":
/* /*
The latest message will be the new user message, one before will be the assistant message from a previous request, and the user message before that will be a previously cached user message. So we need to mark the latest user message as ephemeral to cache it for the next request, and mark the second to last user message as ephemeral to let the server know the last message to retrieve from the cache for the current request.. The latest message will be the new user message, one before will be the assistant message from a previous request, and the user message before that will be a previously cached user message. So we need to mark the latest user message as ephemeral to cache it for the next request, and mark the second to last user message as ephemeral to let the server know the last message to retrieve from the cache for the current request..

1
src/api/kodu.ts
View File

@@ -41,6 +41,7 @@ export class KoduHandler implements ApiHandler {
let requestBody: Anthropic.Beta.PromptCaching.Messages.MessageCreateParamsNonStreaming let requestBody: Anthropic.Beta.PromptCaching.Messages.MessageCreateParamsNonStreaming
switch (modelId) { switch (modelId) {
case "claude-3-5-sonnet-20240620": case "claude-3-5-sonnet-20240620":
case "claude-3-opus-20240229":
case "claude-3-haiku-20240307": case "claude-3-haiku-20240307":
const userMsgIndices = messages.reduce( const userMsgIndices = messages.reduce(
(acc, msg, index) => (msg.role === "user" ? [...acc, index] : acc), (acc, msg, index) => (msg.role === "user" ? [...acc, index] : acc),

2
src/shared/api.ts
View File

@@ -46,7 +46,7 @@ export const anthropicModels = {
"claude-3-opus-20240229": { "claude-3-opus-20240229": {
maxTokens: 4096, maxTokens: 4096,
supportsImages: true, supportsImages: true,
supportsPromptCache: false, supportsPromptCache: true,
inputPrice: 15.0, inputPrice: 15.0,
outputPrice: 75.0, outputPrice: 75.0,
cacheWritesPrice: 18.75, cacheWritesPrice: 18.75,