WIP

Author: intitni

Core/Sources/ChatTab/ChatTab.swift

@@ -62,12 +62,18 @@ open class BaseChatTab: Equatable {
 
     @ViewBuilder
     public var body: some View {
+        let id = "BaseChatTab\(info.id)"
         if let tab = self as? ChatTabType {
-            ContentView(info: info, buildView: tab.buildView).id(info.id)
+            ContentView(info: info, buildView: tab.buildView).id(id)
         } else {
-            EmptyView().id(info.id)
+            EmptyView().id(id)
         }
     }
+    
+    @ViewBuilder
+    public var menu: some View {
+        EmptyView()
+    }
 
     public static func == (lhs: BaseChatTab, rhs: BaseChatTab) -> Bool {
         lhs.id == rhs.id

Tool/Sources/LangChain/Chain.swift

@@ -13,7 +13,7 @@ public extension Chain {
         return parseOutput(output)
     }
 
-    func call(_ input: Input, callbackManagers: [ChainCallbackManager]) async throws -> Output {
+    func call(_ input: Input, callbackManagers: [ChainCallbackManager] = []) async throws -> Output {
         for callbackManager in callbackManagers {
             callbackManager.onChainStart(type: Self.self, input: input)
         }

Tool/Sources/LangChain/Chains/RetrievalQA.swift

@@ -0,0 +1,30 @@
+import Foundation
+
+final class RetrievalQA: Chain {
+    let vectorStore: VectorStore
+    let embedding: Embeddings
+
+    struct Output {
+        var answer: String
+        var sourceDocuments: [Document]
+    }
+
+    init(vectorStore: VectorStore, embedding: Embeddings) {
+        self.vectorStore = vectorStore
+        self.embedding = embedding
+    }
+
+    func callLogic(
+        _ input: String,
+        callbackManagers: [ChainCallbackManager]
+    ) async throws -> Output {
+        let embeddedQuestion = try awa
+        
+        return .init(answer: "", sourceDocuments: [])
+    }
+
+    func parseOutput(_ output: Output) -> String {
+        return output.answer
+    }
+}
+