ReActFunctionCallingChatAgent

Author: intitni

Pro

@@ -25,7 +25,7 @@ public extension CallbackEvents {
         public let info: AgentFinish<Output>
     }
 
-    func agentDidFinish<Output: AgentOutputParsable>() -> AgentDidFinish<Output>.Type {
+    static func agentDidFinish<Output: AgentOutputParsable>() -> AgentDidFinish<Output>.Type {
         AgentDidFinish<Output>.self
     }
 

Tool/Sources/LangChain/Agent.swift

@@ -25,7 +25,7 @@ public actor AgentExecutor<InnerAgent: Agent>: Chain
         tools: [AgentTool],
         maxIteration: Int? = 10,
         maxExecutionTime: Double? = nil,
-        earlyStopHandleType: AgentEarlyStopHandleType = .force,
+        earlyStopHandleType: AgentEarlyStopHandleType = .generate,
         initialSteps: [AgentAction] = []
     ) {
         self.agent = agent
@@ -151,8 +151,7 @@ extension AgentExecutor {
             let completedActions = try await withThrowingTaskGroup(of: AgentAction.self) {
                 taskGroup in
                 for action in actions {
-                    callbackManagers
-                        .forEach { $0.send(CallbackEvents.AgentActionDidStart(info: action)) }
+                    callbackManagers.send(CallbackEvents.AgentActionDidStart(info: action))
                     if action.observation != nil {
                         taskGroup.addTask { action }
                         continue
@@ -167,8 +166,7 @@ extension AgentExecutor {
                 for try await action in taskGroup {
                     try Task.checkCancellation()
                     completedActions.append(action)
-                    callbackManagers
-                        .forEach { $0.send(CallbackEvents.AgentActionDidEnd(info: action)) }
+                    callbackManagers.send(CallbackEvents.AgentActionDidEnd(info: action))
                 }
                 return completedActions
             }

Tool/Sources/LangChain/AgentExecutor.swift

@@ -10,6 +10,10 @@ public struct CallbackEvents {
 }
 
 public struct CallbackManager {
+    struct Observer<Event: CallbackEvent> {
+        let handler: (Event.Info) -> Void
+    }
+    
     fileprivate var observers = [Any]()
 
     public init() {}
@@ -24,28 +28,28 @@ public struct CallbackManager {
         _: Event.Type = Event.self,
         _ handler: @escaping (Event.Info) -> Void
     ) {
-        observers.append(handler)
+        observers.append(Observer<Event>(handler: handler))
     }
 
     public mutating func on<Event: CallbackEvent>(
         _: KeyPath<CallbackEvents, Event.Type>,
         _ handler: @escaping (Event.Info) -> Void
     ) {
-        observers.append(handler)
+        observers.append(Observer<Event>(handler: handler))
     }
 
     public func send<Event: CallbackEvent>(_ event: Event) {
-        for case let observer as ((Event.Info) -> Void) in observers {
-            observer(event.info)
+        for case let observer as Observer<Event> in observers {
+            observer.handler(event.info)
         }
     }
 
     func send<Event: CallbackEvent>(
         _: KeyPath<CallbackEvents, Event.Type>,
         _ info: Event.Info
     ) {
-        for case let observer as ((Event.Info) -> Void) in observers {
-            observer(info)
+        for case let observer as Observer<Event> in observers {
+            observer.handler(info)
         }
     }
 }