Adjust web search function

Author: intitni

Core/Sources/ChatContextCollectors/WebChatContextCollector/WebChatContextCollector.swift

@@ -28,9 +28,14 @@ public final class WebChatContextCollector: ChatContextCollector {
 
 extension WebChatContextCollector {
     static func detectLinks(from messages: [ChatMessage]) -> [String] {
-        return messages.lazy.compactMap(\.content).map(detectLinks(from:)).flatMap { $0 }
+        return messages.lazy
+            .compactMap {
+                $0.content ?? $0.functionCall?.arguments
+            }
+            .map(detectLinks(from:))
+            .flatMap { $0 }
     }
-    
+
     static func detectLinks(from content: String) -> [String] {
         var links = [String]()
         let detector = try? NSDataDetector(types: NSTextCheckingResult.CheckingType.link.rawValue)

Tool/Sources/LangChain/Chains/RetrievalQA.swift

@@ -27,7 +27,7 @@ public final class RetrievalQAChain: Chain {
         let embeddedQuestion = try await embedding.embed(query: input)
         let documents = try await vectorStore.searchWithDistance(
             embeddings: embeddedQuestion,
-            count: 10
+            count: 5
         )
         let refinementChain = RefineDocumentChain(chatModelFactory: chatModelFactory)
         let answer = try await refinementChain.run(

Tool/Sources/OpenAIService/Models.swift

@@ -16,8 +16,8 @@ public struct ChatMessage: Equatable, Codable {
     }
 
     public struct FunctionCall: Codable, Equatable {
-        var name: String
-        var arguments: String
+        public var name: String
+        public var arguments: String
     }
 
     /// The role of a message.