chore: update attachment types to include selection support

Author: friggeri

nodejs/src/types.ts

@@ -803,13 +803,30 @@ export interface MessageOptions {
     prompt: string;
 
     /**
-     * File or directory attachments
-     */
-    attachments?: Array<{
-        type: "file" | "directory";
-        path: string;
-        displayName?: string;
-    }>;
+     * File, directory, or selection attachments
+     */
+    attachments?: Array<
+        | {
+              type: "file";
+              path: string;
+              displayName?: string;
+          }
+        | {
+              type: "directory";
+              path: string;
+              displayName?: string;
+          }
+        | {
+              type: "selection";
+              filePath: string;
+              displayName: string;
+              selection?: {
+                  start: { line: number; character: number };
+                  end: { line: number; character: number };
+              };
+              text?: string;
+          }
+    >;
 
     /**
      * Message delivery mode

python/copilot/types.py

@@ -26,13 +26,45 @@
 LogLevel = Literal["none", "error", "warning", "info", "debug", "all"]
 
 
-# Attachment type
-class Attachment(TypedDict):
-    type: Literal["file", "directory"]
+# Selection range for text attachments
+class SelectionRange(TypedDict):
+    line: int
+    character: int
+
+
+class Selection(TypedDict):
+    start: SelectionRange
+    end: SelectionRange
+
+
+# Attachment types - discriminated union based on 'type' field
+class FileAttachment(TypedDict):
+    """File attachment."""
+    type: Literal["file"]
+    path: str
+    displayName: NotRequired[str]
+
+
+class DirectoryAttachment(TypedDict):
+    """Directory attachment."""
+    type: Literal["directory"]
     path: str
     displayName: NotRequired[str]
 
 
+class SelectionAttachment(TypedDict):
+    """Selection attachment with text from a file."""
+    type: Literal["selection"]
+    filePath: str
+    displayName: str
+    selection: NotRequired[Selection]
+    text: NotRequired[str]
+
+
+# Attachment type - union of all attachment types
+Attachment = Union[FileAttachment, DirectoryAttachment, SelectionAttachment]
+
+
 # Options for creating a CopilotClient
 class CopilotClientOptions(TypedDict, total=False):
     """Options for creating a CopilotClient"""