Fix image file name not in AI issue (#282)

* fix for image file name not in output

* fix for server port

* fix for events and image file name issue

Author: vizsatiz

flo_ai/flo_ai/arium/arium.py

@@ -647,8 +647,11 @@ def _serialize_node_output(self, result: Any) -> Optional[str]:
         if isinstance(result, str):
             return result
         if isinstance(result, list):
-            parts = [self._serialize_node_output(item) for item in result]
-            return '\n'.join(p for p in parts if p) or None
+            # agent.run() returns conversation_history (all messages); take only
+            # the last item, which is the agent's own reply for this node.
+            if not result:
+                return None
+            return self._serialize_node_output(result[-1])
         if hasattr(result, 'content'):
             return self._serialize_node_output(result.content)
         if hasattr(result, 'text'):

flo_ai/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flo_ai"
-version = "1.1.5"
+version = "1.1.6"
 description = "A easy way to create structured AI agents"
 authors = [{ name = "rootflo", email = "engineering.tools@rootflo.ai" }]
 requires-python = ">=3.10,<4.0"

flo_ai/setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name='flo-ai',
-    version='1.1.3',
+    version='1.1.6',
     author='Rootflo',
     description='Create composable AI agents',
     long_description=long_description,

wavefront/client/src/components/InferencePopup.tsx

@@ -243,6 +243,7 @@ const InferencePopup: React.FC<InferencePopupProps> = ({ onClose, renderModal =
         const imageMessage = {
           image_base64: imageBase64Content,
           mime_type: uploadedImage.mimeType,
+          file_name: uploadedImage.file.name,
         };
         messageInputs.push(imageMessage);
       }

wavefront/client/src/pages/apps/[appId]/agents/[id].tsx

@@ -416,6 +416,7 @@ const AgentDetail: React.FC = () => {
           const imageMessage = {
             image_base64: image.base64Content,
             mime_type: image.mimeType,
+            file_name: image.file.name,
           };
           setChatHistory((prev) => [...prev, { role: 'user', content: imageMessage }]);
           conversationInputs.push({

wavefront/client/src/pages/apps/[appId]/workflows/[id].tsx

@@ -369,6 +369,7 @@ const WorkflowDetail: React.FC = () => {
           const imageMessage = {
             image_base64: image.base64Content,
             mime_type: image.mimeType,
+            file_name: image.file.name,
           };
           messageInputs.push({ role: 'user', content: imageMessage });
           setChatHistory((prev) => [...prev, { role: 'user', content: imageMessage }]);
@@ -379,6 +380,7 @@ const WorkflowDetail: React.FC = () => {
         const imageMessage = {
           image_base64: imageBase64Content,
           mime_type: uploadedImage.mimeType,
+          file_name: uploadedImage.file?.name,
         };
         messageInputs.push({ role: 'user', content: imageMessage });
         setChatHistory((prev) => [...prev, { role: 'user', content: imageMessage }]);

wavefront/client/src/types/chat-message.ts

@@ -2,6 +2,7 @@
 export interface ImageContent {
   image_base64: string;
   mime_type?: string;
+  file_name?: string;
 }
 
 export interface DocumentContent {

wavefront/server/modules/agents_module/agents_module/utils/input_processing_utils.py

@@ -43,6 +43,17 @@ def process_inference_inputs(
             elif input_item.get('role') == 'user':
                 input_content = input_item.get('content', {})
                 if is_image_message(input_content):
+                    # Inject filename as a text message before the image so
+                    # agents can reference the original file name in their output.
+                    file_name = input_content.get('file_name')
+                    if file_name:
+                        resolved_inputs.append(
+                            UserMessage(
+                                content=TextMessageContent(
+                                    text=f'The original filename of this image is: {file_name}'
+                                )
+                            )
+                        )
                     # Extract image_bytes and mime_type from image_base64
                     try:
                         data_url_pattern = r'^data:(image/[a-zA-Z0-9.+-]+);base64,(.+)$'

wavefront/server/modules/agents_module/pyproject.toml

@@ -13,7 +13,7 @@ dependencies = [
     "flo-utils",
     "tools-module",
     "api-services-module",
-    "flo-ai==1.1.5",
+    "flo-ai==1.1.6",
 ]
 
 [tool.uv.sources]

wavefront/server/modules/knowledge_base_module/pyproject.toml

@@ -17,7 +17,7 @@ dependencies = [
     "pandas~=2.2.3",
     "ollama~=0.4.8",
     "textract~=1.6.5",
-    "flo-ai==1.1.5",
+    "flo-ai==1.1.6",
     "google-cloud-pubsub~=2.30.0",
     "boto3<=1.38.40",
     "pyyaml>=6.0.3,<7",