fixes

Author: jamesrochabrun

PolyAIExample/PolyAIExample.xcodeproj/project.xcworkspace/xcshareddata/swiftpm/Package.resolved

@@ -67,7 +67,7 @@ struct MessageDemoView: View {
                switch selectedSegment {
                case .openAI:
                   parameters = .openAI(
-                     model: .gpt35Turbo, 
+                     model: .o1Preview,
                      messages: [
                         .init(role: .user, content: prompt)
                      ])

PolyAIExample/PolyAIExample/MessageDemoView.swift

@@ -28,7 +28,7 @@ extension MessageResponse: LLMMessageResponse {
    }
 
    public var usageMetrics: UsageMetrics {
-      ChatUsageMetrics(inputTokens: usage.inputTokens, outputTokens: usage.outputTokens, totalTokens: nil)
+      ChatUsageMetrics(inputTokens: usage.inputTokens ?? 0, outputTokens: usage.outputTokens, totalTokens: nil)
    }
 
    public var tools: [ToolUsage] {

Sources/PolyAI/Interfaces/Response/Message/LLMMessageResponse+Anthropic.swift

@@ -47,8 +47,8 @@ struct DefaultPolyAIService: PolyAIService {
                openAIService = OpenAIServiceFactory.service(aiproxyPartialKey: aiproxyPartialKey, aiproxyClientID: aiproxyClientID)
             }
 
-         case .anthropic(let apiKey, let configuration):
-            anthropicService = AnthropicServiceFactory.service(apiKey: apiKey, configuration: configuration)
+         case .anthropic(let apiKey, let configuration, let betaHeaders):
+            anthropicService = AnthropicServiceFactory.service(apiKey: apiKey, betaHeaders: betaHeaders, configuration: configuration)
 
          case .gemini(let apiKey):
             gemini = .init(apiKey: apiKey)
@@ -89,7 +89,7 @@ struct DefaultPolyAIService: PolyAIService {
             )
          }
          let systemMessage  = messages.first { $0.role == "system" }
-         let messageParameter = MessageParameter(model: model, messages: messageParams, maxTokens: maxTokens, system: systemMessage?.content, stream: false)
+         let messageParameter = MessageParameter(model: model, messages: messageParams, maxTokens: maxTokens, system: .text(systemMessage?.content ?? ""), stream: false)
          return try await anthropicService.createMessage(messageParameter)
 
       case .gemini(let model, let messages, let maxTokens):
@@ -150,7 +150,7 @@ struct DefaultPolyAIService: PolyAIService {
              )
          }
          let systemMessage  = messages.first { $0.role == "system" }
-         let messageParameter = MessageParameter(model: model, messages: messageParams, maxTokens: maxTokens, system: systemMessage?.content)
+         let messageParameter = MessageParameter(model: model, messages: messageParams, maxTokens: maxTokens, system: .text(systemMessage?.content ?? ""))
          let stream = try await anthropicService.streamMessage(messageParameter)
          return try mapToLLMMessageStreamResponse(stream: stream)
 

Sources/PolyAI/Service/DefaultPolyAIService.swift

@@ -41,7 +41,8 @@ public enum LLMConfiguration {
    /// - Parameters:
    ///   - apiKey: The API key for authenticating requests to Anthropic.
    ///   - configuration: The URLSession configuration to use for network requests. Defaults to `.default`.
-   case anthropic(apiKey: String, configuration: URLSessionConfiguration = .default)
+   ///   - betaHeaders: An array of headers for Anthropic's beta features.
+   case anthropic(apiKey: String, configuration: URLSessionConfiguration = .default, betaHeaders: [String]? = nil)
 
    /// Configuration for accessing Gemini's API.
    /// - Parameters: