fix timeout

Signed-off-by: ale <ale@manalejandro.com>

python/cudaq_bridge.py

@@ -539,11 +539,18 @@ def main():
             
             if command:
                 result = dispatch_command(command, **data)
-                response = {
-                    "success": True,
-                    "data": result,
-                    "requestId": request_id
-                }
+                
+                # Check if the result already has success/error structure
+                if isinstance(result, dict) and ("success" in result or "error" in result):
+                    # Result already has proper structure, just add request ID
+                    response = {**result, "requestId": request_id}
+                else:
+                    # Wrap result in standard response structure
+                    response = {
+                        "success": True,
+                        "data": result,
+                        "requestId": request_id
+                    }
                 print(f"DEBUG - Command {command} completed successfully", file=sys.stderr, flush=True)
             else:
                 response = {
@@ -552,7 +559,10 @@ def main():
                     "requestId": request_id
                 }
             
-            print(json.dumps(response), flush=True)
+            # Ensure response is sent on a single line with explicit flush
+            output = json.dumps(response)
+            print(output, flush=True)
+            sys.stdout.flush()  # Extra flush to ensure it's sent
             
         except Exception as e:
             print(f"ERROR - Exception in main loop: {str(e)}", file=sys.stderr, flush=True)
@@ -563,7 +573,9 @@ def main():
                 "traceback": traceback.format_exc(),
                 "requestId": request.get("requestId", "") if 'request' in locals() else ""
             }
-            print(json.dumps(error_response), flush=True)
+            output = json.dumps(error_response)
+            print(output, flush=True)
+            sys.stdout.flush()  # Extra flush to ensure it's sent
 
 
 if __name__ == "__main__":