Adjust API of ChatGPTFunction

intitni · intitni · commit 48acf0498ef8 · 2023-09-20T22:15:42.000+08:00
diff --git a/Core/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/Functions/ExpandFocusRangeFunction.swift b/Core/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/Functions/ExpandFocusRangeFunction.swift
@@ -13,13 +13,11 @@ struct ExpandFocusRangeFunction: ChatGPTFunction {
             "Editing Document Context is updated to display code at \(range)."
         }
     }
-    
+
     struct E: Error, LocalizedError {
         var errorDescription: String?
     }
 
-    var reportProgress: (String) async -> Void = { _ in }
-
     var name: String {
         "expandFocusRange"
     }
@@ -32,18 +30,21 @@ struct ExpandFocusRangeFunction: ChatGPTFunction {
         .type: "object",
         .properties: [:],
     ] }
-    
+
     weak var contextCollector: ActiveDocumentChatContextCollector?
-    
+
     init(contextCollector: ActiveDocumentChatContextCollector) {
         self.contextCollector = contextCollector
     }
 
-    func prepare() async {
+    func prepare(reportProgress: @escaping (String) async -> Void) async {
         await reportProgress("Finding the focused code..")
     }
 
-    func call(arguments: Arguments) async throws -> Result {
+    func call(
+        arguments: Arguments,
+        reportProgress: @escaping (String) async -> Void
+    ) async throws -> Result {
         await reportProgress("Finding the focused code..")
         contextCollector?.activeDocumentContext?.expandFocusedRangeToContextRange()
         guard let newContext = contextCollector?.activeDocumentContext?.focusedContext else {
@@ -56,3 +57,4 @@ struct ExpandFocusRangeFunction: ChatGPTFunction {
         return .init(range: newContext.codeRange)
     }
 }
+
diff --git a/Core/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/Functions/MoveToCodeAroundLineFunction.swift b/Core/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/Functions/MoveToCodeAroundLineFunction.swift
@@ -15,13 +15,11 @@ struct MoveToCodeAroundLineFunction: ChatGPTFunction {
             "Editing Document Context is updated to display code at \(range)."
         }
     }
-    
+
     struct E: Error, LocalizedError {
         var errorDescription: String?
     }
 
-    var reportProgress: (String) async -> Void = { _ in }
-
     var name: String {
         "getCodeAtLine"
     }
@@ -36,7 +34,7 @@ struct MoveToCodeAroundLineFunction: ChatGPTFunction {
             "line": [
                 .type: "number",
                 .description: "The line number in the file",
-            ]
+            ],
         ],
         .required: ["line"],
     ] }
@@ -47,11 +45,14 @@ struct MoveToCodeAroundLineFunction: ChatGPTFunction {
         self.contextCollector = contextCollector
     }
 
-    func prepare() async {
+    func prepare(reportProgress: @escaping (String) async -> Void) async {
         await reportProgress("Finding code around..")
     }
 
-    func call(arguments: Arguments) async throws -> Result {
+    func call(
+        arguments: Arguments,
+        reportProgress: @escaping (String) async -> Void
+    ) async throws -> Result {
         await reportProgress("Finding code around line \(arguments.line)..")
         contextCollector?.activeDocumentContext?.moveToCodeAroundLine(arguments.line)
         guard let newContext = contextCollector?.activeDocumentContext?.focusedContext else {
diff --git a/Core/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/Functions/MoveToFocusedCodeFunction.swift b/Core/Sources/ChatContextCollectors/ActiveDocumentChatContextCollector/Functions/MoveToFocusedCodeFunction.swift
@@ -4,7 +4,7 @@ import OpenAIService
 import SuggestionModel
 
 struct MoveToFocusedCodeFunction: ChatGPTFunction {
-    struct Arguments: Codable {}
+    typealias Arguments = NoArguments
 
     struct Result: ChatGPTFunctionResult {
         var range: CursorRange
@@ -13,13 +13,11 @@ struct MoveToFocusedCodeFunction: ChatGPTFunction {
             "Editing Document Context is updated to display code at \(range)."
         }
     }
-    
+
     struct E: Error, LocalizedError {
         var errorDescription: String?
     }
 
-    var reportProgress: (String) async -> Void = { _ in }
-
     var name: String {
         "moveToFocusedCode"
     }
@@ -28,22 +26,20 @@ struct MoveToFocusedCodeFunction: ChatGPTFunction {
         "Move editing document context to the selected or focused code"
     }
 
-    var argumentSchema: JSONSchemaValue { [
-        .type: "object",
-        .properties: [:],
-    ] }
-    
     weak var contextCollector: ActiveDocumentChatContextCollector?
-    
+
     init(contextCollector: ActiveDocumentChatContextCollector) {
         self.contextCollector = contextCollector
     }
 
-    func prepare() async {
+    func prepare(reportProgress: @escaping (String) async -> Void) async {
         await reportProgress("Finding the focused code..")
     }
 
-    func call(arguments: Arguments) async throws -> Result {
+    func call(
+        arguments: Arguments,
+        reportProgress: @escaping (String) async -> Void
+    ) async throws -> Result {
         await reportProgress("Finding the focused code..")
         contextCollector?.activeDocumentContext?.moveToFocusedCode()
         guard let newContext = contextCollector?.activeDocumentContext?.focusedContext else {
@@ -56,3 +52,4 @@ struct MoveToFocusedCodeFunction: ChatGPTFunction {
         return .init(range: newContext.codeRange)
     }
 }
+
diff --git a/Core/Sources/ChatContextCollectors/WebChatContextCollector/QueryWebsiteFunction.swift b/Core/Sources/ChatContextCollectors/WebChatContextCollector/QueryWebsiteFunction.swift
@@ -17,8 +17,6 @@ struct QueryWebsiteFunction: ChatGPTFunction {
         }
     }
 
-    var reportProgress: (String) async -> Void = { _ in }
-
     var name: String {
         "queryWebsite"
     }
@@ -47,11 +45,14 @@ struct QueryWebsiteFunction: ChatGPTFunction {
         ]
     }
 
-    func prepare() async {
+    func prepare(reportProgress: @escaping (String) async -> Void) async {
         await reportProgress("Reading..")
     }
 
-    func call(arguments: Arguments) async throws -> Result {
+    func call(
+        arguments: Arguments,
+        reportProgress: @escaping (String) async -> Void
+    ) async throws -> Result {
         do {
             let embedding = OpenAIEmbedding(configuration: UserPreferenceEmbeddingConfiguration())
 
@@ -61,10 +62,13 @@ struct QueryWebsiteFunction: ChatGPTFunction {
                     group.addTask {
                         // 1. grab the website content
                         await reportProgress("Loading \(url)..")
-                        
+
                         if let database = await TemporaryUSearch.view(identifier: urlString) {
                             await reportProgress("Getting relevant information..")
-                            let qa = QAInformationRetrievalChain(vectorStore: database, embedding: embedding)
+                            let qa = QAInformationRetrievalChain(
+                                vectorStore: database,
+                                embedding: embedding
+                            )
                             return try await qa.call(.init(arguments.query)).information
                         }
                         let loader = WebLoader(urls: [url])
@@ -83,7 +87,10 @@ struct QueryWebsiteFunction: ChatGPTFunction {
                         try await database.set(embeddedDocuments)
                         // 4. generate answer
                         await reportProgress("Getting relevant information..")
-                        let qa = QAInformationRetrievalChain(vectorStore: database, embedding: embedding)
+                        let qa = QAInformationRetrievalChain(
+                            vectorStore: database,
+                            embedding: embedding
+                        )
                         let result = try await qa.call(.init(arguments.query))
                         return result.information
                     }
@@ -101,7 +108,7 @@ struct QueryWebsiteFunction: ChatGPTFunction {
                         .joined(separator: "\n")
                 )
                 """)
-                                     
+
                 return all
             }
 
diff --git a/Core/Sources/ChatContextCollectors/WebChatContextCollector/SearchFunction.swift b/Core/Sources/ChatContextCollectors/WebChatContextCollector/SearchFunction.swift
@@ -28,10 +28,8 @@ struct SearchFunction: ChatGPTFunction {
             }.joined(separator: "\n")
         }
     }
-    
-    let maxTokens: Int
 
-    var reportProgress: (String) async -> Void = { _ in }
+    let maxTokens: Int
 
     var name: String {
         "searchWeb"
@@ -62,19 +60,22 @@ struct SearchFunction: ChatGPTFunction {
         ]
     }
 
-    func prepare() async {
+    func prepare(reportProgress: @escaping ReportProgress) async {
         await reportProgress("Searching..")
     }
 
-    func call(arguments: Arguments) async throws -> Result {
+    func call(
+        arguments: Arguments,
+        reportProgress: @escaping ReportProgress
+    ) async throws -> Result {
         await reportProgress("Searching \(arguments.query)")
 
         do {
             let bingSearch = BingSearchService(
                 subscriptionKey: UserDefaults.shared.value(for: \.bingSearchSubscriptionKey),
                 searchURL: UserDefaults.shared.value(for: \.bingSearchEndpoint)
             )
-            
+
             let result = try await bingSearch.search(
                 query: arguments.query,
                 numberOfResult: maxTokens > 5000 ? 5 : 3,
diff --git a/Pro b/Pro
@@ -1 +1 @@
-Subproject commit f17934c1b63005f187a1fda43de9001bdfcfcddb
+Subproject commit d02bbce00428d326bb91bddd946240714781f838
diff --git a/Tool/Sources/LangChain/Agent.swift b/Tool/Sources/LangChain/Agent.swift
@@ -44,6 +44,19 @@ public extension CallbackEvents {
     var agentActionDidEnd: AgentActionDidEnd.Type {
         AgentActionDidEnd.self
     }
+    
+    struct AgentFunctionCallingToolReportProgress: CallbackEvent {
+        public struct Info {
+            let functionName: String
+            let progress: String
+        }
+        
+        public let info: Info
+    }
+    
+    var agentFunctionCallingToolReportProgress: AgentFunctionCallingToolReportProgress.Type {
+        AgentFunctionCallingToolReportProgress.self
+    }
 }
 
 public struct AgentFinish<Output: AgentOutputParsable> {
@@ -68,7 +81,7 @@ public enum AgentNextStep<Output: AgentOutputParsable> {
 
 public struct AgentScratchPad<Content: Equatable>: Equatable {
     public var content: Content
-    
+
     public init(content: Content) {
         self.content = content
     }
@@ -99,7 +112,8 @@ public protocol Agent {
 
     func validateTools(tools: [AgentTool]) throws
     func constructScratchpad(intermediateSteps: [AgentAction]) -> AgentScratchPad<ScratchPadContent>
-    func constructFinalScratchpad(intermediateSteps: [AgentAction]) -> AgentScratchPad<ScratchPadContent>
+    func constructFinalScratchpad(intermediateSteps: [AgentAction])
+        -> AgentScratchPad<ScratchPadContent>
     func extraPlan(input: AgentInput<Input, ScratchPadContent>)
     func parseOutput(_ output: ChatModelChain<AgentInput<Input, ScratchPadContent>>.Output) async
         -> AgentNextStep<Output>
@@ -146,8 +160,12 @@ public extension Agent {
             case let .finish(finish):
                 return finish
             case .actions:
-                return .init(returnValue: .unstructured(output.content ?? ""), log: output.content ?? "")
+                return .init(
+                    returnValue: .unstructured(output.content ?? ""),
+                    log: output.content ?? ""
+                )
             }
         }
     }
 }
+
diff --git a/Tool/Sources/LangChain/AgentTool.swift b/Tool/Sources/LangChain/AgentTool.swift
@@ -31,18 +31,17 @@ public struct SimpleAgentTool: AgentTool {
     }
 }
 
-public struct FunctionCallingAgentTool<F: ChatGPTFunction>: AgentTool, ChatGPTFunction {
+public class FunctionCallingAgentTool<F: ChatGPTFunction>: AgentTool {
     public func call(arguments: F.Arguments) async throws -> F.Result {
-        try await function.call(arguments: arguments)
+        try await function.call(arguments: arguments, reportProgress: reportProgress)
     }
 
     public var argumentSchema: OpenAIService.JSONSchemaValue { function.argumentSchema }
 
-    public func prepare() async { await function.prepare() }
-
-    public var reportProgress: (String) async -> Void {
-        get { function.reportProgress }
-        set { function.reportProgress = newValue }
+    public func prepare() async {
+        await function.prepare(reportProgress: { [weak self] p in
+            self?.reportProgress(p)
+        })
     }
 
     public typealias Arguments = F.Arguments
@@ -53,15 +52,37 @@ public struct FunctionCallingAgentTool<F: ChatGPTFunction>: AgentTool, ChatGPTFu
     public var description: String
     public var returnDirectly: Bool
 
-    public init(function: F, returnDirectly: Bool = false) {
+    let callbackManagers: [CallbackManager]
+
+    public init(
+        function: F,
+        returnDirectly: Bool = false,
+        callbackManagers: [CallbackManager] = []
+    ) {
         self.function = function
+        self.callbackManagers = callbackManagers
         name = function.name
-        description = "Run an action: \(function.description)"
+        description = function.description
         self.returnDirectly = returnDirectly
     }
 
+    func reportProgress(_ progress: String) {
+        callbackManagers.send(
+            CallbackEvents.AgentFunctionCallingToolReportProgress(info: .init(
+                functionName: name,
+                progress: progress
+            ))
+        )
+    }
+
     public func run(input: String) async throws -> String {
-        try await function.call(argumentsJsonString: input).botReadableContent
+        try await function.call(
+            argumentsJsonString: input,
+            reportProgress: { [weak self] p in
+                self?.reportProgress(p)
+            }
+        )
+        .botReadableContent
     }
 }
 
diff --git a/Tool/Sources/LangChain/Chains/RefineDocumentChain.swift b/Tool/Sources/LangChain/Chains/RefineDocumentChain.swift
@@ -46,15 +46,8 @@ public final class RefineDocumentChain: Chain {
         var functions: [any ChatGPTFunction] = [RespondFunction()]
     }
 
-    struct RespondFunction: ChatGPTFunction {
+    struct RespondFunction: ChatGPTArgumentsCollectingFunction {
         typealias Arguments = IntermediateAnswer
-
-        struct Result: ChatGPTFunctionResult {
-            var botReadableContent: String { "" }
-        }
-
-        var reportProgress: (String) async -> Void = { _ in }
-
         var name: String = "respond"
         var description: String = "Respond with the refined answer"
         var argumentSchema: JSONSchemaValue {
@@ -77,12 +70,6 @@ public final class RefineDocumentChain: Chain {
                 .required: ["answer", "more", "usefulness"],
             ]
         }
-
-        func prepare() async {}
-
-        func call(arguments: Arguments) async throws -> Result {
-            return Result()
-        }
     }
 
     func buildChatModel() -> ChatModelChain<RefinementInput> {
diff --git a/Tool/Sources/LangChain/Chains/RelevantInformationExtractionChain.swift b/Tool/Sources/LangChain/Chains/RelevantInformationExtractionChain.swift
diff --git a/Tool/Sources/LangChain/Chains/StructuredOutputChatModelChain.swift b/Tool/Sources/LangChain/Chains/StructuredOutputChatModelChain.swift
diff --git a/Tool/Sources/OpenAIService/ChatGPTService.swift b/Tool/Sources/OpenAIService/ChatGPTService.swift
diff --git a/Tool/Sources/OpenAIService/FucntionCall/ChatGPTFunction.swift b/Tool/Sources/OpenAIService/FucntionCall/ChatGPTFunction.swift

Original file line number	Diff line number	Diff line change
`@@ -13,13 +13,11 @@ struct ExpandFocusRangeFunction: ChatGPTFunction {`
`13`	`13`	`"Editing Document Context is updated to display code at \(range)."`
`14`	`14`	`}`
`15`	`15`	`}`
`16`		`-`
	`16`	`+`
`17`	`17`	`struct E: Error, LocalizedError {`
`18`	`18`	`var errorDescription: String?`
`19`	`19`	`}`
`20`	`20`
`21`		`- var reportProgress: (String) async -> Void = { _ in }`
`22`		`-`
`23`	`21`	`var name: String {`
`24`	`22`	`"expandFocusRange"`
`25`	`23`	`}`
`@@ -32,18 +30,21 @@ struct ExpandFocusRangeFunction: ChatGPTFunction {`
`32`	`30`	`.type: "object",`
`33`	`31`	`.properties: [:],`
`34`	`32`	`] }`
`35`		`-`
	`33`	`+`
`36`	`34`	`weak var contextCollector: ActiveDocumentChatContextCollector?`
`37`		`-`
	`35`	`+`
`38`	`36`	`init(contextCollector: ActiveDocumentChatContextCollector) {`
`39`	`37`	`self.contextCollector = contextCollector`
`40`	`38`	`}`
`41`	`39`
`42`		`- func prepare() async {`
	`40`	`+ func prepare(reportProgress: @escaping (String) async -> Void) async {`
`43`	`41`	`await reportProgress("Finding the focused code..")`
`44`	`42`	`}`
`45`	`43`
`46`		`- func call(arguments: Arguments) async throws -> Result {`
	`44`	`+ func call(`
	`45`	`+ arguments: Arguments,`
	`46`	`+ reportProgress: @escaping (String) async -> Void`
	`47`	`+ ) async throws -> Result {`
`47`	`48`	`await reportProgress("Finding the focused code..")`
`48`	`49`	`contextCollector?.activeDocumentContext?.expandFocusedRangeToContextRange()`
`49`	`50`	`guard let newContext = contextCollector?.activeDocumentContext?.focusedContext else {`
`@@ -56,3 +57,4 @@ struct ExpandFocusRangeFunction: ChatGPTFunction {`
`56`	`57`	`return .init(range: newContext.codeRange)`
`57`	`58`	`}`
`58`	`59`	`}`
	`60`	`+`
Original file line number	Diff line number	Diff line change
`@@ -4,7 +4,7 @@ import OpenAIService`
`4`	`4`	`import SuggestionModel`
`5`	`5`
`6`	`6`	`struct MoveToFocusedCodeFunction: ChatGPTFunction {`
`7`		`- struct Arguments: Codable {}`
	`7`	`+ typealias Arguments = NoArguments`
`8`	`8`
`9`	`9`	`struct Result: ChatGPTFunctionResult {`
`10`	`10`	`var range: CursorRange`
`@@ -13,13 +13,11 @@ struct MoveToFocusedCodeFunction: ChatGPTFunction {`
`13`	`13`	`"Editing Document Context is updated to display code at \(range)."`
`14`	`14`	`}`
`15`	`15`	`}`
`16`		`-`
	`16`	`+`
`17`	`17`	`struct E: Error, LocalizedError {`
`18`	`18`	`var errorDescription: String?`
`19`	`19`	`}`
`20`	`20`
`21`		`- var reportProgress: (String) async -> Void = { _ in }`
`22`		`-`
`23`	`21`	`var name: String {`
`24`	`22`	`"moveToFocusedCode"`
`25`	`23`	`}`
`@@ -28,22 +26,20 @@ struct MoveToFocusedCodeFunction: ChatGPTFunction {`
`28`	`26`	`"Move editing document context to the selected or focused code"`
`29`	`27`	`}`
`30`	`28`
`31`		`- var argumentSchema: JSONSchemaValue { [`
`32`		`- .type: "object",`
`33`		`- .properties: [:],`
`34`		`- ] }`
`35`		`-`
`36`	`29`	`weak var contextCollector: ActiveDocumentChatContextCollector?`
`37`		`-`
	`30`	`+`
`38`	`31`	`init(contextCollector: ActiveDocumentChatContextCollector) {`
`39`	`32`	`self.contextCollector = contextCollector`
`40`	`33`	`}`
`41`	`34`
`42`		`- func prepare() async {`
	`35`	`+ func prepare(reportProgress: @escaping (String) async -> Void) async {`
`43`	`36`	`await reportProgress("Finding the focused code..")`
`44`	`37`	`}`
`45`	`38`
`46`		`- func call(arguments: Arguments) async throws -> Result {`
	`39`	`+ func call(`
	`40`	`+ arguments: Arguments,`
	`41`	`+ reportProgress: @escaping (String) async -> Void`
	`42`	`+ ) async throws -> Result {`
`47`	`43`	`await reportProgress("Finding the focused code..")`
`48`	`44`	`contextCollector?.activeDocumentContext?.moveToFocusedCode()`
`49`	`45`	`guard let newContext = contextCollector?.activeDocumentContext?.focusedContext else {`
`@@ -56,3 +52,4 @@ struct MoveToFocusedCodeFunction: ChatGPTFunction {`
`56`	`52`	`return .init(range: newContext.codeRange)`
`57`	`53`	`}`
`58`	`54`	`}`
	`55`	`+`
Original file line number	Diff line number	Diff line change
`@@ -17,8 +17,6 @@ struct QueryWebsiteFunction: ChatGPTFunction {`
`17`	`17`	`}`
`18`	`18`	`}`
`19`	`19`
`20`		`- var reportProgress: (String) async -> Void = { _ in }`
`21`		`-`
`22`	`20`	`var name: String {`
`23`	`21`	`"queryWebsite"`
`24`	`22`	`}`
`@@ -47,11 +45,14 @@ struct QueryWebsiteFunction: ChatGPTFunction {`
`47`	`45`	`]`
`48`	`46`	`}`
`49`	`47`
`50`		`- func prepare() async {`
	`48`	`+ func prepare(reportProgress: @escaping (String) async -> Void) async {`
`51`	`49`	`await reportProgress("Reading..")`
`52`	`50`	`}`
`53`	`51`
`54`		`- func call(arguments: Arguments) async throws -> Result {`
	`52`	`+ func call(`
	`53`	`+ arguments: Arguments,`
	`54`	`+ reportProgress: @escaping (String) async -> Void`
	`55`	`+ ) async throws -> Result {`
`55`	`56`	`do {`
`56`	`57`	`let embedding = OpenAIEmbedding(configuration: UserPreferenceEmbeddingConfiguration())`
`57`	`58`
`@@ -61,10 +62,13 @@ struct QueryWebsiteFunction: ChatGPTFunction {`
`61`	`62`	`group.addTask {`
`62`	`63`	`// 1. grab the website content`
`63`	`64`	`await reportProgress("Loading \(url)..")`
`64`		`-`
	`65`	`+`
`65`	`66`	`if let database = await TemporaryUSearch.view(identifier: urlString) {`
`66`	`67`	`await reportProgress("Getting relevant information..")`
`67`		`- let qa = QAInformationRetrievalChain(vectorStore: database, embedding: embedding)`
	`68`	`+ let qa = QAInformationRetrievalChain(`
	`69`	`+ vectorStore: database,`
	`70`	`+ embedding: embedding`
	`71`	`+ )`
`68`	`72`	`return try await qa.call(.init(arguments.query)).information`
`69`	`73`	`}`
`70`	`74`	`let loader = WebLoader(urls: [url])`
`@@ -83,7 +87,10 @@ struct QueryWebsiteFunction: ChatGPTFunction {`
`83`	`87`	`try await database.set(embeddedDocuments)`
`84`	`88`	`// 4. generate answer`
`85`	`89`	`await reportProgress("Getting relevant information..")`
`86`		`- let qa = QAInformationRetrievalChain(vectorStore: database, embedding: embedding)`
	`90`	`+ let qa = QAInformationRetrievalChain(`
	`91`	`+ vectorStore: database,`
	`92`	`+ embedding: embedding`
	`93`	`+ )`
`87`	`94`	`let result = try await qa.call(.init(arguments.query))`
`88`	`95`	`return result.information`
`89`	`96`	`}`
`@@ -101,7 +108,7 @@ struct QueryWebsiteFunction: ChatGPTFunction {`
`101`	`108`	`.joined(separator: "\n")`
`102`	`109`	`)`
`103`	`110`	`""")`
`104`		`-`
	`111`	`+`
`105`	`112`	`return all`
`106`	`113`	`}`
`107`	`114`
Original file line number	Diff line number	Diff line change
`@@ -28,10 +28,8 @@ struct SearchFunction: ChatGPTFunction {`
`28`	`28`	`}.joined(separator: "\n")`
`29`	`29`	`}`
`30`	`30`	`}`
`31`		`-`
`32`		`- let maxTokens: Int`
`33`	`31`
`34`		`- var reportProgress: (String) async -> Void = { _ in }`
	`32`	`+ let maxTokens: Int`
`35`	`33`
`36`	`34`	`var name: String {`
`37`	`35`	`"searchWeb"`
`@@ -62,19 +60,22 @@ struct SearchFunction: ChatGPTFunction {`
`62`	`60`	`]`
`63`	`61`	`}`
`64`	`62`
`65`		`- func prepare() async {`
	`63`	`+ func prepare(reportProgress: @escaping ReportProgress) async {`
`66`	`64`	`await reportProgress("Searching..")`
`67`	`65`	`}`
`68`	`66`
`69`		`- func call(arguments: Arguments) async throws -> Result {`
	`67`	`+ func call(`
	`68`	`+ arguments: Arguments,`
	`69`	`+ reportProgress: @escaping ReportProgress`
	`70`	`+ ) async throws -> Result {`
`70`	`71`	`await reportProgress("Searching \(arguments.query)")`
`71`	`72`
`72`	`73`	`do {`
`73`	`74`	`let bingSearch = BingSearchService(`
`74`	`75`	`subscriptionKey: UserDefaults.shared.value(for: \.bingSearchSubscriptionKey),`
`75`	`76`	`searchURL: UserDefaults.shared.value(for: \.bingSearchEndpoint)`
`76`	`77`	`)`
`77`		`-`
	`78`	`+`
`78`	`79`	`let result = try await bingSearch.search(`
`79`	`80`	`query: arguments.query,`
`80`	`81`	`numberOfResult: maxTokens > 5000 ? 5 : 3,`