Update package

PreternaturalAI · Mar 22, 2024 · 43e8ba2 · 43e8ba2
1 parent 60f1c7b
commit 43e8ba2
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 5 deletions.
diff --git a/Sources/Anthropic/Intramodular/Anthropic+LLMRequestHandling.swift b/Sources/Anthropic/Intramodular/Anthropic+LLMRequestHandling.swift
@@ -54,7 +54,7 @@ extension Anthropic: LLMRequestHandling {
             \.complete,
              with: Anthropic.API.RequestBodies.Complete(
                 prompt: prompt.prefix.promptLiteral._stripToText(),
-                model: .claude_v2,
+                model: .claude_3_opus_20240229,
                 maxTokensToSample: parameters.tokenLimit.fixedValue ?? 256,
                 stopSequences: parameters.stops,
                 stream: false,

diff --git a/Sources/Anthropic/Intramodular/Anthropic.API.swift b/Sources/Anthropic/Intramodular/Anthropic.API.swift
@@ -11,7 +11,7 @@ extension Anthropic {
         public typealias API = Anthropic.API
 
         case apiKeyMissing
-        case incorrectAPIKeyProvided
+        case invalidAPIKey
         case rateLimitExceeded
         case badRequest(API.Request.Error)
         case runtime(AnyError)
@@ -89,10 +89,10 @@ extension Anthropic.API {
                     ).error {
                         print(error.message)
 
-                        if error.message.contains("You didn't provide an API key") {
+                        if error.message.contains("x-api-key header is required") {
                             throw Error.apiKeyMissing
-                        } else if error.message.contains("Incorrect API key provided") {
-                            throw Error.incorrectAPIKeyProvided
+                        } else if error.message.contains("invalid x-api-key") {
+                            throw Error.invalidAPIKey
                         }
                     }
 

diff --git a/Sources/LargeLanguageModels/Intramodular/LLMs/LLMRequestHandling.swift b/Sources/LargeLanguageModels/Intramodular/LLMs/LLMRequestHandling.swift
@@ -117,6 +117,26 @@ extension LLMRequestHandling {
         return try await complete(prompt: prompt)
     }
 
+    public func complete(
+        _ messages: [AbstractLLM.ChatMessage]
+    ) async throws -> AbstractLLM.ChatCompletion {
+        let prompt = AbstractLLM.ChatPrompt(
+            messages: messages
+        )
+
+        return try await complete(prompt: prompt)
+    }
+
+    public func completion(
+        for messages: [AbstractLLM.ChatMessage]
+    ) async throws -> AbstractLLM.ChatCompletionStream {
+        let prompt = AbstractLLM.ChatPrompt(
+            messages: messages
+        )
+
+        return try await completion(for: prompt)
+    }
+
     public func complete(
         _ message: AbstractLLM.ChatMessage,
         model: some _MLModelIdentifierConvertible