fix: cell doesn't working

TexteaInc · Feb 16, 2024 · 286d96e · 286d96e
1 parent f1b4f69
commit 286d96e
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/backend/funix/decorator/call.py b/backend/funix/decorator/call.py
@@ -277,6 +277,8 @@ def output_to_web_function(**wrapped_function_kwargs):
                                 }
                             )
                         )
+                        ws.close()
+                        return
                     else:
                         result = []
                         for temp_function_result in function(**arg):
@@ -285,16 +287,18 @@ def output_to_web_function(**wrapped_function_kwargs):
                                 result.extend(function_result)
                             else:
                                 result.append(function_result)
-                            ws.send(dumps({"result": result}))
-                            result = []
-                    ws.close()
                 else:
                     temp_result = wrapped_function(**arg)
                     if isinstance(temp_result, list):
                         result.extend(temp_result)
                     else:
                         result.append(temp_result)
-                    return [{"result": result}]
+            if need_websocket:
+                ws.send(dumps({"result": result}))
+                ws.close()
+                return
+            else:
+                return [{"result": result}]
         elif len(upload_base64_files) > 0:
             new_args = function_kwargs
             for upload_base64_file_key in upload_base64_files.keys():