fixup! minor text selection mapping improvement

Signed-off-by: Petr Kadlec <petr@puradesign.cz>

Author: kapetr

apps/agentstack-sdk-py/examples/canvas_ui_code_agent.py

@@ -29,6 +29,7 @@
     "Word Counter",
 ]
 
+
 def generate_code_response(code_title: str, description: str, closing_message: str) -> str:
     """Generate a code response with the given title, description, and closing message."""
     return f"""\
@@ -45,7 +46,7 @@ def fibonacci(n):
         return [0]
     elif n == 2:
         return [0, 1]
-    
+
     sequence = [0, 1]
     for i in range(2, n):
         sequence.append(sequence[i-1] + sequence[i-2])
@@ -78,20 +79,20 @@ async def artifacts_agent(
 
     canvas_edit_request = await canvas.parse_canvas_edit_request(message=input)
 
-    if canvas_edit_request:
-        print("Canvas Edit Request:")
-        print(f"Start Index: {canvas_edit_request.start_index}")
-        print(f"End Index: {canvas_edit_request.end_index}")
-        print(f"Artifact ID: {canvas_edit_request.artifact.artifact_id}")
-        print(f"Selected Text: {canvas_edit_request.content}")
-    
     if canvas_edit_request:
         original_code = (
             canvas_edit_request.artifact.parts[0].root.text
             if isinstance(canvas_edit_request.artifact.parts[0].root, TextPart)
             else ""
         )
         edited_part = original_code[canvas_edit_request.start_index : canvas_edit_request.end_index]
+
+        print("Canvas Edit Request:")
+        print(f"Start Index: {canvas_edit_request.start_index}")
+        print(f"End Index: {canvas_edit_request.end_index}")
+        print(f"Artifact ID: {canvas_edit_request.artifact.artifact_id}")
+        print(f"Edited part: {edited_part}")
+
         description = f"You requested to edit this part:\n\n{edited_part}\n\n"
         code_title = "Edited Code"
         closing_message = "Your code has been updated!"
@@ -119,13 +120,10 @@ async def artifacts_agent(
 
     # Keep the full match including the code block formatting
     code_content = match.group(0).strip()
-    
+
     # Extract artifact name from the comment line if present
     first_line = match.group(1).strip().split("\n", 1)[0]
-    if first_line.startswith("#"):
-        artifact_name = first_line.lstrip("# ").strip()
-    else:
-        artifact_name = "Python Script"
+    artifact_name = first_line.lstrip("# ").strip() if first_line.startswith("#") else "Python Script"
 
     # Split code content into x chunks for streaming
     num_chunks = 8

apps/agentstack-sdk-py/examples/canvas_ui_test_agent.py

@@ -51,6 +51,7 @@ def generate_recipe_response(recipe_title: str, description: str, closing_messag
 {closing_message}
 """
 
+
 @server.agent(
     name="Canvas example agent",
 )
@@ -68,20 +69,20 @@ async def artifacts_agent(
 
     canvas_edit_request = await canvas.parse_canvas_edit_request(message=input)
 
-    if canvas_edit_request:
-        print("Canvas Edit Request:")
-        print(f"Start Index: {canvas_edit_request.start_index}")
-        print(f"End Index: {canvas_edit_request.end_index}")
-        print(f"Artifact ID: {canvas_edit_request.artifact.artifact_id}")
-        print(f"Selected Text: {canvas_edit_request.content}")
-
     if canvas_edit_request:
         original_recipe = (
             canvas_edit_request.artifact.parts[0].root.text
             if isinstance(canvas_edit_request.artifact.parts[0].root, TextPart)
             else ""
         )
         edited_part = original_recipe[canvas_edit_request.start_index : canvas_edit_request.end_index]
+
+        print("Canvas Edit Request:")
+        print(f"Start Index: {canvas_edit_request.start_index}")
+        print(f"End Index: {canvas_edit_request.end_index}")
+        print(f"Artifact ID: {canvas_edit_request.artifact.artifact_id}")
+        print(f"Edited part: {edited_part}")
+
         description = f"You requested to edit this part:\n\n{edited_part}\n\n"
         recipe_title = "Canvas Recipe EDITED"
         closing_message = "Enjoy your edited meal!"

apps/agentstack-ui/src/modules/canvas/markdown/utils/mapDOMSelectionToMarkdown.ts

@@ -60,18 +60,21 @@ export function mapDOMSelectionToMarkdown(range: Range, markdownSource: string):
   }
 
   const { startContainer, endContainer, startOffset, endOffset } = range;
+
   const isSingleNodeSelection = startContainer === endContainer;
 
   // Find positioned ancestors
   const startInfo = findNearestElementWithPosition(startContainer);
   const endInfo = findNearestElementWithPosition(endContainer);
 
+  const isSinglePositionElementSelection = startInfo?.element === endInfo?.element;
+
   if (!startInfo || !endInfo) {
     throw new Error('Could not find source position attributes for selection');
   }
 
-  const nodeStartOffset = getNodeOffsetWithin(startInfo.element, startContainer, range.startOffset);
-  const nodeEndOffset = getNodeOffsetWithin(endInfo.element, endContainer, range.endOffset);
+  const nodeStartOffset = getNodeOffsetWithin(startInfo.element, startContainer, startOffset);
+  const nodeEndOffset = getNodeOffsetWithin(endInfo.element, endContainer, endOffset);
 
   // Map to markdown positions
   let startIndex = startInfo.start + nodeStartOffset;
@@ -85,7 +88,7 @@ export function mapDOMSelectionToMarkdown(range: Range, markdownSource: string):
     startIndex += startIndexInRegion;
   }
 
-  let endIndex = endInfo.start + nodeEndOffset + (isSingleNodeSelection ? startIndexInRegion : 0);
+  let endIndex = endInfo.start + nodeEndOffset + (isSinglePositionElementSelection ? startIndexInRegion : 0);
   const endSearchContent = endContainer.textContent?.slice(isSingleNodeSelection ? startOffset : 0, endOffset);
 
   // Search for the end content in the markdown to adjust for syntax shifts