Fix issue loading query after memory warning

duckduckgo · Dec 19, 2024 · 9c519a5 · 9c519a5
1 parent 1472616
commit 9c519a5
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/LocalPackages/AIChat/Sources/AIChat/Public API/AIChatViewController.swift b/LocalPackages/AIChat/Sources/AIChat/Public API/AIChatViewController.swift
@@ -87,8 +87,12 @@ extension AIChatViewController {
 // MARK: - Public functions
 extension AIChatViewController {
     public func loadQuery(_ query: URLQueryItem) {
-        webViewController?.loadQuery(query)
-    }
+         // Ensure the webViewController is added before loading the query
+         if webViewController == nil {
+             addWebViewController()
+         }
+         webViewController?.loadQuery(query)
+     }
 }
 
 // MARK: - Views Setup