✨Supported no_offload

cfdraw/app/endpoints/queue.py

@@ -56,7 +56,10 @@ async def run(self) -> None:
             try:
                 plugin.elapsed_times.start()
                 if await self._broadcast_working(uid):
-                    await offload(plugin(request))
+                    future = plugin(request)
+                    if not plugin.settings.no_offload:
+                        future = offload(future)
+                    await future
             except Exception as err:
                 await self._broadcast_exception(uid, get_err_msg(err))
             # cleanup

cfdraw/schema/plugins.py

@@ -244,6 +244,16 @@ class IPluginSettings(IChakra):
         None,
         description="Extra (chakra) props of the plugin button",
     )
+    # internal fields
+    no_offload: bool = Field(
+        False,
+        description=(
+            "Whether not to offload the plugin to sub-thread when it is executed, "
+            "useful when you know the plugin is fast enough.\n"
+            "> This is introduced mainly because some libraries (e.g., `matplotlib`) "
+            "need to be executed in the main thread."
+        ),
+    )
 
     def to_react(self, type: str, hash: str, identifier: str) -> Dict[str, Any]:
         d = self.dict(exclude={"pluginInfo"})