add missing image

modelscope · Oct 28, 2024 · 27f035d · 27f035d
1 parent e3de1ef
commit 27f035d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modelscope_agent/tools/code_interpreter/code_interpreter.py b/modelscope_agent/tools/code_interpreter/code_interpreter.py
@@ -463,7 +463,8 @@ def _execute_code(self, kc: BlockingKernelClient, code: str,
                     text = msg['content']['data'].get('text/plain', '')
                     if 'image/png' in msg['content']['data']:
                         image_b64 = msg['content']['data']['image/png']
-                        image_url = self._serve_image(image_b64, 'png')
+                        image_url = self._serve_image(image_b64, 'png',
+                                                      is_remote)
                         image_idx += 1
                         image = '![IMAGEGEN](%s)' % (image_url)
                     elif 'text/html' in msg['content']['data']: