fix some bugs with single node layouts

Author: sawka

frontend/app/store/keymodel.ts

@@ -146,6 +146,19 @@ function switchBlockByBlockNum(index: number) {
 function switchBlockInDirection(tabId: string, direction: NavigateDirection) {
     const layoutModel = getLayoutModelForTabById(tabId);
     const inWaveAI = globalStore.get(atoms.waveAIFocusedAtom);
+    
+    if (direction === NavigateDirection.Left) {
+        const numBlocks = globalStore.get(layoutModel.numLeafs);
+        if (inWaveAI) {
+            return;
+        }
+        if (numBlocks === 1) {
+            // special case: layout model doesn't set "rect" when there's only one block
+            WaveAIModel.getInstance().focusInput();
+            return;
+        }
+    }
+    
     const navResult = layoutModel.switchNodeFocusInDirection(direction, inWaveAI);
     if (navResult.atLeft) {
         WaveAIModel.getInstance().focusInput();

frontend/app/store/tabrpcclient.ts

@@ -27,25 +27,34 @@ export class TabClient extends WshClient {
             throw new Error(`Block not found: ${blockId}`);
         }
 
-        const additionalProps = layoutModel.getNodeAdditionalProperties(node);
-        if (!additionalProps?.rect) {
-            throw new Error(`Block rect not found for: ${blockId}`);
-        }
-
         const displayContainer = layoutModel.displayContainerRef.current;
         if (!displayContainer) {
             throw new Error("Display container not found");
         }
 
         const containerRect = displayContainer.getBoundingClientRect();
-        const blockRect = additionalProps.rect;
-
-        const electronRect: Electron.Rectangle = {
-            x: Math.round(containerRect.x + blockRect.left),
-            y: Math.round(containerRect.y + blockRect.top),
-            width: Math.round(blockRect.width),
-            height: Math.round(blockRect.height),
-        };
+        const additionalProps = layoutModel.getNodeAdditionalProperties(node);
+
+        let electronRect: Electron.Rectangle;
+
+        if (!additionalProps?.rect) {
+            // Bug: rect is not set when there is only one block in the layout
+            // In this case, use the full container rect
+            electronRect = {
+                x: Math.round(containerRect.x),
+                y: Math.round(containerRect.y),
+                width: Math.round(containerRect.width),
+                height: Math.round(containerRect.height),
+            };
+        } else {
+            const blockRect = additionalProps.rect;
+            electronRect = {
+                x: Math.round(containerRect.x + blockRect.left),
+                y: Math.round(containerRect.y + blockRect.top),
+                width: Math.round(blockRect.width),
+                height: Math.round(blockRect.height),
+            };
+        }
 
         return await getApi().captureScreenshot(electronRect);
     }

pkg/aiusechat/openai/openai-backend.go

@@ -627,6 +627,9 @@ func handleOpenAIEvent(
 		}
 		return nil, nil
 
+	case "response.output_text.done":
+		return nil, nil
+
 	case "response.content_part.done":
 		var ev openaiResponseContentPartDoneEvent
 		if err := json.Unmarshal([]byte(data), &ev); err != nil {

pkg/aiusechat/openai/openai-convertmessage.go

@@ -100,16 +100,15 @@ func debugPrintReq(req *OpenAIRequest, endpoint string) {
 	for _, tool := range req.Tools {
 		toolNames = append(toolNames, tool.Name)
 	}
+	log.Printf("model %s\n", req.Model)
 	if len(toolNames) > 0 {
 		log.Printf("tools: %s\n", strings.Join(toolNames, ","))
 	}
-	
+
 	log.Printf("inputs (%d):", len(req.Input))
 	for idx, input := range req.Input {
 		debugPrintInput(idx, input)
 	}
-	
-	log.Printf("baseurl: %s\n", endpoint)
 }
 
 // buildOpenAIHTTPRequest creates a complete HTTP request for the OpenAI API