fix: handle only images and audios keys in ToolRunner artifact handli…

…ng feature
RobotecAI · Sep 30, 2024 · c092be4 · c092be4
1 parent 364ad48
commit c092be4
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/rai/rai/agents/tool_runner.py b/src/rai/rai/agents/tool_runner.py
@@ -109,7 +109,10 @@ def run_one(call: ToolCall):
                 artifact = cast(MultimodalArtifact, artifact)
                 store_artifacts(output.tool_call_id, [artifact])
 
-            if artifact is not None:  # multimodal case
+            if artifact is not None and (
+                len(artifact.get("images", [])) > 0
+                or len(artifact.get("audios", [])) > 0
+            ):  # multimodal case, we currently support images and audios artifacts
                 return ToolMultimodalMessage(
                     content=str_output(output.content),
                     name=call["name"],