update beta description

fumito-ito · Sep 6, 2024 · db3cd17 · db3cd17
1 parent 2032bda
commit db3cd17
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/Sources/AnthropicSwiftSDK/Network/HeaderProvider/AnthropicHeaderProvider.swift b/Sources/AnthropicSwiftSDK/Network/HeaderProvider/AnthropicHeaderProvider.swift
@@ -20,7 +20,7 @@ struct DefaultAnthropicHeaderProvider: AnthropicHeaderProvider {
     /// content type of response, now only support JSON
     let contentType = "application/json"
 
-    private let betaDescription = "messages-2023-12-15"
+    private let betaDescription = "prompt-caching-2024-07-31"
 
     func getAnthropicAPIHeaders() -> [String: String] {
         var headers: [String: String] = [

diff --git a/Tests/AnthropicSwiftSDKTests/Network/AnthropicAPIClientTests.swift b/Tests/AnthropicSwiftSDKTests/Network/AnthropicAPIClientTests.swift
@@ -69,7 +69,7 @@ final class AnthropicAPIClientTests: XCTestCase {
             XCTAssertEqual(headers!["x-api-key"], "test-api-key")
             XCTAssertEqual(headers!["anthropic-version"], "2023-06-01")
             XCTAssertEqual(headers!["Content-Type"], "application/json")
-            XCTAssertEqual(headers!["anthropic-beta"], "messages-2023-12-15")
+            XCTAssertEqual(headers!["anthropic-beta"], "prompt-caching-2024-07-31")
 
             expectation.fulfill()
         })
@@ -90,7 +90,7 @@ final class AnthropicAPIClientTests: XCTestCase {
             XCTAssertEqual(headers!["x-api-key"], "test-api-key")
             XCTAssertEqual(headers!["anthropic-version"], "2023-06-01")
             XCTAssertEqual(headers!["Content-Type"], "application/json")
-            XCTAssertEqual(headers!["anthropic-beta"], "messages-2023-12-15")
+            XCTAssertEqual(headers!["anthropic-beta"], "prompt-caching-2024-07-31")
 
             expectation.fulfill()
         })

diff --git a/...s/AnthropicSwiftSDKTests/Network/HeaderProvider/DefaultAnthropicHeaderProviderTests.swift b/...s/AnthropicSwiftSDKTests/Network/HeaderProvider/DefaultAnthropicHeaderProviderTests.swift
@@ -29,7 +29,7 @@ final class DefaultAnthropicHeaderProviderTests: XCTestCase {
         let provider = DefaultAnthropicHeaderProvider(useBeta: true)
         let headers = provider.getAnthropicAPIHeaders()
 
-        XCTAssertEqual(headers["anthropic-beta"], "messages-2023-12-15")
+        XCTAssertEqual(headers["anthropic-beta"], "prompt-caching-2024-07-31")
     }
 
     func testBetaHeaderShouldNotBeProvidedIfUseBeta() {