Merge pull request #26 from fumito-ito/feature/function-calling

Support `Tool Use`

Author: fumito-ito

.swiftpm/configuration/Package.resolved

@@ -61,6 +61,11 @@
             value = ""
             isEnabled = "YES">
          </EnvironmentVariable>
+         <EnvironmentVariable
+            key = "IDEPreferLogStreaming"
+            value = "YES"
+            isEnabled = "YES">
+         </EnvironmentVariable>
       </EnvironmentVariables>
    </LaunchAction>
    <ProfileAction

Example.swiftpm/.swiftpm/configuration/Package.resolved

@@ -0,0 +1,20 @@
+//
+//  FunctionTools.swift
+//  Example
+//
+//  Created by 伊藤史 on 2024/07/23.
+//
+
+import Foundation
+import FunctionCalling
+
+@FunctionCalling(service: .claude)
+struct FunctionTools {
+    /// Returns the temperature at the location specified by the argument, together with the units.
+    /// - Parameter location: location to specify the temperature
+    /// - Returns: the temperature at the location
+    @CallableFunction
+    func getWeather(location: String) -> String {
+        return "32 Celsius"
+    }
+}

Example.swiftpm/.swiftpm/xcode/xcshareddata/xcschemes/Example.xcscheme

@@ -8,6 +8,7 @@
 import Foundation
 import SwiftUI
 import AnthropicSwiftSDK
+import FunctionCalling
 
 protocol MessagesSubject {
     var messages: [ChatMessage] { get }
@@ -42,7 +43,9 @@ protocol MessageSendable {
         stopSequence: [String]?,
         temperature: Double?,
         topP: Double?,
-        topK: Int?
+        topK: Int?,
+        toolContainer: ToolContainer?,
+        toolChoice: ToolChoice
     ) async throws -> MessagesResponse
 }
 
@@ -56,6 +59,8 @@ protocol MessageStreamable {
         stopSequence: [String]?,
         temperature: Double?,
         topP: Double?,
-        topK: Int?
+        topK: Int?,
+        toolContainer: ToolContainer?,
+        toolChoice: ToolChoice
     ) async throws -> AsyncThrowingStream<StreamingResponse, Error>
 }

Example.swiftpm/FunctionTools.swift

@@ -7,6 +7,7 @@
 
 import Foundation
 import AnthropicSwiftSDK
+import FunctionCalling
 
 @Observable class MockViewModel: StreamMessagesSubject, SendMessagesSubject {
     required init(messageHandler: any MessageStreamable, title: String, model: AnthropicSwiftSDK.Model) {
@@ -71,13 +72,13 @@ import AnthropicSwiftSDK
 }
 
 struct MockMessageStreamable: MessageStreamable {
-    func streamMessage(_ messages: [AnthropicSwiftSDK.Message], model: AnthropicSwiftSDK.Model, system: String?, maxTokens: Int, metaData: AnthropicSwiftSDK.MetaData?, stopSequence: [String]?, temperature: Double?, topP: Double?, topK: Int?) async throws -> AsyncThrowingStream<any AnthropicSwiftSDK.StreamingResponse, any Error> {
+    func streamMessage(_ messages: [AnthropicSwiftSDK.Message], model: AnthropicSwiftSDK.Model, system: String?, maxTokens: Int, metaData: AnthropicSwiftSDK.MetaData?, stopSequence: [String]?, temperature: Double?, topP: Double?, topK: Int?, toolContainer: ToolContainer?, toolChoice: ToolChoice) async throws -> AsyncThrowingStream<any AnthropicSwiftSDK.StreamingResponse, any Error> {
         fatalError()
     }
 }
 
 struct MockMessagesSendable: MessageSendable {
-    func createMessage(_ messages: [AnthropicSwiftSDK.Message], model: AnthropicSwiftSDK.Model, system: String?, maxTokens: Int, metaData: AnthropicSwiftSDK.MetaData?, stopSequence: [String]?, temperature: Double?, topP: Double?, topK: Int?) async throws -> AnthropicSwiftSDK.MessagesResponse {
+    func createMessage(_ messages: [AnthropicSwiftSDK.Message], model: AnthropicSwiftSDK.Model, system: String?, maxTokens: Int, metaData: AnthropicSwiftSDK.MetaData?, stopSequence: [String]?, temperature: Double?, topP: Double?, topK: Int?, toolContainer: ToolContainer?, toolChoice: ToolChoice) async throws -> AnthropicSwiftSDK.MessagesResponse {
         fatalError()
     }
 }

Example.swiftpm/Package.resolved

@@ -10,6 +10,7 @@ import AnthropicSwiftSDK
 
 @Observable class SendViewModel: SendMessagesSubject {
     private let messageHandler: MessageSendable
+    private let functionTools = FunctionTools()
     let title: String
     let model: Model
 
@@ -49,7 +50,9 @@ import AnthropicSwiftSDK
                     stopSequence: nil,
                     temperature: nil,
                     topP: nil,
-                    topK: nil
+                    topK: nil,
+                    toolContainer: functionTools,
+                    toolChoice: .auto
                 )
 
                 if case let .text(reply) = result.content.first {

Example.swiftpm/Protocol/MessagesSubject.swift

@@ -10,6 +10,7 @@ import AnthropicSwiftSDK
 
 @Observable class StreamViewModel: StreamMessagesSubject {
     private let messageHandler: MessageStreamable
+    private let functionTools = FunctionTools()
     let title: String
     let model: Model
 
@@ -40,7 +41,7 @@ import AnthropicSwiftSDK
         task = Task {
             do {
                 isLoading = true
-                let stream = try await messageHandler.streamMessage(
+                let stream = try await self.messageHandler.streamMessage(
                     [message],
                     model: model,
                     system: nil,
@@ -49,13 +50,15 @@ import AnthropicSwiftSDK
                     stopSequence: nil,
                     temperature: nil,
                     topP: nil,
-                    topK: nil
+                    topK: nil,
+                    toolContainer: functionTools,
+                    toolChoice: .auto
                 )
                 for try await chunk in stream {
                     switch chunk.type {
                     case .contentBlockDelta:
                         if let response = chunk as? StreamingContentBlockDeltaResponse {
-                            messages.append(.init(user: .assistant, text: response.delta.text))
+                            messages.append(.init(user: .assistant, text: response.delta.text ?? ""))
                         }
                     default:
                         break