Update

Author: intitni

Core/Package.swift

@@ -182,6 +182,7 @@ let package = Package(
 
                 // plugins
                 "MathChatPlugin",
+                "SearchChatPlugin",
 
                 .product(name: "Preferences", package: "Tool"),
             ]
@@ -319,6 +320,17 @@ let package = Package(
             ],
             path: "Sources/ChatPlugins/MathChatPlugin"
         ),
+        
+            .target(
+                name: "SearchChatPlugin",
+                dependencies: [
+                    "ChatPlugin",
+                    "OpenAIService",
+                    .product(name: "LangChain", package: "Tool"),
+                    .product(name: "PythonKit", package: "PythonKit"),
+                ],
+                path: "Sources/ChatPlugins/SearchChatPlugin"
+            ),
     ]
 )
 

Core/Sources/ChatPlugin/SearchChatPlugin.swift

@@ -0,0 +1,99 @@
+import Foundation
+import LangChain
+import PythonHelper
+import PythonKit
+
+func search(_ query: String) async throws -> String {
+    #if DEBUG
+    let verbose = true
+    #else
+    let verbose = false
+    #endif
+
+    let task = Task {
+        try runPython {
+            let llm = try LangChainChatModel.DynamicChatOpenAI(temperature: 0)
+            let utilities = try Python.attemptImportOnPythonThread("langchain.utilities")
+            let BingSearchAPIWrapper = utilities.BingSearchAPIWrapper
+            let agents = try Python.attemptImportOnPythonThread("langchain.agents")
+            let Tool = agents.Tool
+            let initializeAgent = agents.initialize_agent
+            let AgentType = agents.AgentType
+
+            let bingSearch = BingSearchAPIWrapper(
+                bing_subscription_key: "f1eaef707e9443ddb08df2cfb6ac1eb5",
+                bing_search_url: "https://api.bing.microsoft.com/v7.0/search/",
+                k: 5
+            )
+
+            var links = [String]()
+
+            let getSearchResult = PythonInstanceMethod { arguments -> String in
+                guard let query = arguments.first else { return "Empty" }
+                let results = bingSearch.results(query, 5)
+                let resultString = results.enumerated().map { "[\($0)]:###\($1["snippet"])###" }
+                    .joined(separator: "\n")
+                links = results.map {
+                    let url = String($0["link"]) ?? "N/A"
+                    let title = String($0["title"]) ?? "Unknown Title"
+                    return "[\(title)](\(url))"
+                }
+                return resultString
+            }
+            
+            let ff = PythonClass("FF", members: [
+                "run": PythonInstanceMethod { arguments -> String in
+                    guard let query = arguments.first else { return "Empty" }
+                    let results = bingSearch.results(query, 5)
+                    let resultString = results.enumerated().map { "[\($0)]:###\($1["snippet"])###" }
+                        .joined(separator: "\n")
+                    links = results.map {
+                        let url = String($0["link"]) ?? "N/A"
+                        let title = String($0["title"]) ?? "Unknown Title"
+                        return "[\(title)](\(url))"
+                    }
+                    return resultString
+                }
+            ])
+            
+            let fi = ff.pythonObject()
+            
+            print(fi.run)
+            print(getSearchResult.pythonObject)
+
+            let tools = [
+                Tool(
+                    name: "Search",
+                    func: fi.run,
+                    description: "useful for when you need to answer questions about current events. You should ask targeted questions"
+                ),
+            ]
+
+            let chain = initializeAgent(
+                tools, llm,
+                agent: AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION,
+                verbose: verbose,
+                max_iterations: 1,
+                early_stopping_method: "generate",
+                agent_kwargs: ["system_message_prefix": "Respond to the human as helpfully and accurately as possible. Wrap any code block in thought in <code></code>. Format final answer to be more readable, in a ordered list if possible. You have access to the following tools:"]
+            )
+
+            let trimmedQuery = query.trimmingCharacters(in: [" ", "\n"])
+            do {
+                let result = try chain.run.throwing.dynamicallyCall(withArguments: trimmedQuery)
+                return (String(result) ?? "", links)
+            } catch {
+                return (error.localizedDescription, links)
+            }
+        }
+    }
+
+    let (answer, links) = try await task.value
+
+    return """
+    \(answer)
+    ------
+    \(links.map { "- \($0)" }.joined(separator: "\n"))
+    """
+}
+

Core/Sources/ChatPlugins/SearchChatPlugin/SearchAgent.swift

@@ -0,0 +1,62 @@
+import ChatPlugin
+import Environment
+import Foundation
+import OpenAIService
+
+public actor SearchChatPlugin: ChatPlugin {
+    public static var command: String { "search" }
+    public nonisolated var name: String { "Search" }
+
+    let chatGPTService: any ChatGPTServiceType
+    var isCancelled = false
+    weak var delegate: ChatPluginDelegate?
+
+    public init(inside chatGPTService: any ChatGPTServiceType, delegate: ChatPluginDelegate) {
+        self.chatGPTService = chatGPTService
+        self.delegate = delegate
+    }
+
+    public func send(content: String, originalMessage: String) async {
+        delegate?.pluginDidStart(self)
+        delegate?.pluginDidStartResponding(self)
+
+        let id = "\(Self.command)-\(UUID().uuidString)"
+        var reply = ChatMessage(id: id, role: .assistant, content: "Calculating...")
+
+        await chatGPTService.mutateHistory { history in
+            history.append(.init(role: .user, content: originalMessage, summary: content))
+            history.append(reply)
+        }
+
+        do {
+            let result = try await search(content)
+            await chatGPTService.mutateHistory { history in
+                if history.last?.id == id {
+                    history.removeLast()
+                }
+                reply.content = result
+                history.append(reply)
+            }
+        } catch {
+            await chatGPTService.mutateHistory { history in
+                if history.last?.id == id {
+                    history.removeLast()
+                }
+                reply.content = error.localizedDescription
+                history.append(reply)
+            }
+        }
+
+        delegate?.pluginDidEndResponding(self)
+        delegate?.pluginDidEnd(self)
+    }
+
+    public func cancel() async {
+        isCancelled = true
+    }
+
+    public func stopResponding() async {
+        isCancelled = true
+    }
+}
+

Core/Sources/ChatPlugins/SearchChatPlugin/SearchChatPlugin.swift

@@ -1,8 +1,10 @@
 import ChatPlugin
 import MathChatPlugin
+import SearchChatPlugin
 
 let allPlugins: [ChatPlugin.Type] = [
     TerminalChatPlugin.self,
     AITerminalChatPlugin.self,
     MathChatPlugin.self,
+    SearchChatPlugin.self,
 ]

Core/Sources/ChatService/AllPlugins.swift

@@ -5,12 +5,11 @@ var gilStateEnsure: (() -> Any)!
 var gilStateRelease: ((Any) -> Void)!
 func gilStateGuard<T>(_ closure: @escaping () throws -> T) throws -> T {
     let state = gilStateEnsure()
+    defer { gilStateRelease(state) }
     do {
         let result = try closure()
-        gilStateRelease(state)
         return result
     } catch {
-        gilStateRelease(state)
         throw error
     }
 }
@@ -30,6 +29,7 @@ public func initializePython<GilState, ThreadState>(
     guard !isPythonInitialized else { return }
     setenv("PYTHONHOME", stdLibPath, 1)
     setenv("PYTHONPATH", "\(stdLibPath):\(libDynloadPath):\(sitePackagePath)", 1)
+    setenv("PYTHONIOENCODING", "utf-8", 1)
     isPythonInitialized = true
     // Initialize python
     Py_Initialize()