Specify the document URL when resolving the context

Author: intitni

Core/Sources/SuggestionWidget/FeatureReducers/PromptToCodePanel.swift

@@ -139,9 +139,10 @@ public struct PromptToCodePanel {
 
                 return .run { send in
                     do {
-                        let context = await contextInputController.resolveContext(onStatusChange: {
-                            await send(.statusUpdated($0))
-                        })
+                        let context = await contextInputController.resolveContext(
+                            forDocumentURL: copiedState.promptToCodeState.source.documentURL,
+                            onStatusChange: { await send(.statusUpdated($0)) }
+                        )
                         await send(.referencesUpdated(context.references))
                         let agentFactory = context.agent ?? { SimpleModificationAgent() }
                         _ = try await withThrowingTaskGroup(of: Void.self) { group in

Tool/Sources/PromptToCodeCustomization/PromptToCodeCustomization.swift

@@ -55,7 +55,10 @@ public protocol PromptToCodeContextInputControllerDelegate {
 public protocol PromptToCodeContextInputController: Perception.Perceptible {
     var instruction: NSAttributedString { get set }
 
-    func resolveContext(onStatusChange: @escaping ([String]) async -> Void) async -> (
+    func resolveContext(
+        forDocumentURL: URL,
+        onStatusChange: @escaping ([String]) async -> Void
+    ) async -> (
         instruction: String,
         references: [ChatMessage.Reference],
         topics: [ChatMessage.Reference],
@@ -100,7 +103,10 @@ public final class DefaultPromptToCodeContextInputController: PromptToCodeContex
         instruction = mutable
     }
 
-    public func resolveContext(onStatusChange: @escaping ([String]) async -> Void) -> (
+    public func resolveContext(
+        forDocumentURL: URL,
+        onStatusChange: @escaping ([String]) async -> Void
+    ) -> (
         instruction: String,
         references: [ChatMessage.Reference],
         topics: [ChatMessage.Reference],