fix: support vision prompt modes

2026-08-12 19:13:04 +08:00
parent 09b07751b6
commit 94a02d7e2b
29 changed files with 676 additions and 439 deletions
@@ -670,7 +670,10 @@ class WorkflowBrain(BaseBrain):
                append_function(self._flow_tool(tool, node_id))
        append_function(self._knowledge_function(node_id))
        if stage.vision_enabled and self._require_runtime().vision_function:
-            append_function(self._require_runtime().vision_function)
+            vision_function = self._require_runtime().vision_function
+            append_function(
+                vision_function(stage) if callable(vision_function) else vision_function
+            )
        if self._engine.llm_routing_mode() == "edge_tool":
            for edge in self._engine.edge_tool_edges(node_id):
                append_function(self._edge_tool(edge, node_id))