Skip to content

Commit 9a508d7

Browse files
committed
Update prompt caching model check
1 parent 7c7a311 commit 9a508d7

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

Tool/Sources/OpenAIService/APIs/ClaudeChatCompletionsService.swift

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -385,11 +385,11 @@ extension ClaudeChatCompletionsService.StreamDataChunk {
385385
extension ClaudeChatCompletionsService.RequestBody {
386386
init(_ body: ChatCompletionsRequestBody) {
387387
model = body.model
388-
let supportsPromptCache = if model.hasPrefix("claude-3-5-sonnet")
389-
|| model.hasPrefix("claude-3-5-haiku")
390-
|| model.hasPrefix("claude-3-opus")
391-
|| model.hasPrefix("claude-3-haiku")
392-
{
388+
let prefixChecks = [
389+
"claude-3-5-sonnet", "claude-3-5-haiku", "claude-3-opus", "claude-3-haiku",
390+
"claude-3.5-sonnet", "claude-3.5-haiku",
391+
]
392+
let supportsPromptCache = if prefixChecks.contains(where: model.hasPrefix) {
393393
true
394394
} else {
395395
false

0 commit comments

Comments
 (0)