Updated to the correct models from the documentation

Author: joshgrenon

README.md

@@ -46,13 +46,13 @@ do {
 
 The framework supports various Perplexity AI models through the `PerplexityModel` enum:
 
-- `.sonar`
-- `.sonarMedium`
-- `.sonarLarge`
-- `.codellama34b`
-- `.llama2_70b`
-- `.mistral7b`
-- `.mixtral8x7b`
+- `.sonarSmallOnline`: "llama-3.1-sonar-small-128k-online"
+- `.sonarLargeOnline`: "llama-3.1-sonar-large-128k-online"
+- `.sonarHugeOnline`: "llama-3.1-sonar-huge-128k-online"
+- `.sonarSmallChat`: "llama-3.1-sonar-small-128k-chat"
+- `.sonarLargeChat`: "llama-3.1-sonar-large-128k-chat"
+- `.llama8bInstruct`: "llama-3.1-8b-instruct"
+- `.llama70bInstruct`: "llama-3.1-70b-instruct"
 
 ## Error Handling
 

Sources/PerplexityApiSwift/PerplexityApiSwift.swift

@@ -9,7 +9,7 @@ public class PerplexityApiSwift {
         self.bearerToken = token
     }
 
-    public func chatCompletion(messages: [Message], model: PerplexityModel = .sonarLarge) async throws -> PerplexityResponse {
+    public func chatCompletion(messages: [Message], model: PerplexityModel = .sonarLargeOnline) async throws -> PerplexityResponse {
         guard let bearerToken = bearerToken else {
             throw PerplexityError.tokenNotSet
         }

Sources/PerplexityApiSwift/PerplexityModels.swift

@@ -1,17 +1,18 @@
 import Foundation
 
 public enum PerplexityModel: String {
-    case sonar = "sonar-small-online"
-    case sonarMedium = "sonar-medium-online"
-    case sonarLarge = "sonar-large-online"
-    case codellama34b = "codellama-34b-instruct"
-    case llama2_70b = "llama-2-70b-chat"
-    case mistral7b = "mistral-7b-instruct"
-    case mixtral8x7b = "mixtral-8x7b-instruct"
+    case sonarSmallOnline = "llama-3.1-sonar-small-128k-online"
+    case sonarLargeOnline = "llama-3.1-sonar-large-128k-online"
+    case sonarHugeOnline = "llama-3.1-sonar-huge-128k-online"
+    case sonarSmallChat = "llama-3.1-sonar-small-128k-chat"
+    case sonarLargeChat = "llama-3.1-sonar-large-128k-chat"
+    case llama8bInstruct = "llama-3.1-8b-instruct"
+    case llama70bInstruct = "llama-3.1-70b-instruct"
 }
 
+// We can keep this enum if it's still useful for your application
 public enum PerplexityModelSize: String {
     case small = "small"
-    case medium = "medium"
     case large = "large"
+    case huge = "huge"
 }