feat: adds new Anthropic model (claude-3-5-sonnet-20241022) (#2)

* feat: adds new Anthropic model (`claude-3-5-sonnet-20241022`) * fix test
kevinhermawan · Oct 22, 2024 · 2cd6b11 · 2cd6b11
1 parent 5d22906
commit 2cd6b11
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/Sources/AIModelRetriever/AIModelRetriever.swift b/Sources/AIModelRetriever/AIModelRetriever.swift
@@ -48,10 +48,12 @@ public extension AIModelRetriever {
     /// - Returns: An array of ``AIModel`` that represents Anthropic's available models.
     func anthropic() -> [AIModel] {
         return [
-            AIModel(id: "claude-3-5-sonnet-20240620", name: "Claude 3.5 Sonnet"),
-            AIModel(id: "claude-3-opus-20240229", name: "Claude 3 Opus"),
-            AIModel(id: "claude-3-sonnet-20240229", name: "Claude 3 Sonnet"),
-            AIModel(id: "claude-3-haiku-20240307", name: "Claude 3 Haiku")
+            AIModel(id: "claude-3-5-sonnet-latest", name: "Claude 3.5 Sonnet (Latest)"),
+            AIModel(id: "claude-3-5-sonnet-20241022", name: "Claude 3.5 Sonnet (20241022)"),
+            AIModel(id: "claude-3-opus-latest", name: "Claude 3 Opus (Latest)"),
+            AIModel(id: "claude-3-opus-20240229", name: "Claude 3 Opus (20240229)"),
+            AIModel(id: "claude-3-sonnet-20240229", name: "Claude 3 Sonnet (20240229)"),
+            AIModel(id: "claude-3-haiku-20240307", name: "Claude 3 Haiku (20240307)")
         ]
     }
 }

diff --git a/Tests/AIModelRetrieverTests/AIModelRetrieverTests.swift b/Tests/AIModelRetrieverTests/AIModelRetrieverTests.swift
@@ -32,8 +32,8 @@ final class AIModelRetrieverTests: XCTestCase {
         let models = retriever.anthropic()
 
         XCTAssertFalse(models.isEmpty)
-        XCTAssertTrue(models.contains { $0.id == "claude-3-5-sonnet-20240620" })
-        XCTAssertTrue(models.contains { $0.name == "Claude 3.5 Sonnet" })
+        XCTAssertTrue(models.contains { $0.id == "claude-3-5-sonnet-latest" })
+        XCTAssertTrue(models.contains { $0.name == "Claude 3.5 Sonnet (Latest)" })
     }
 
     func testGoogle() {