code combiner fix v1

agents.py

@@ -426,8 +426,11 @@ class auto_analyst(dspy.Module):
 
         # Execute code combiner after all agents complete
         code_list = [result['code'] for _, result in completed_results if 'code' in result]
-        combiner_result = self.code_combiner_agent(agent_code_list=str(code_list), dataset=dict_['dataset'])
-        yield 'code_combiner_agent', dict(combiner_result)
+        try:
+            combiner_result = self.code_combiner_agent(agent_code_list=str(code_list), dataset=dict_['dataset'])
+            yield 'code_combiner_agent', dict(combiner_result)
+        except Exception as e:
+            yield 'code_combiner_agent', {'error': str(e)}
 
 # Agent to make a Chat history name from a query
 class chat_history_name_agent(dspy.Signature):

format_response.py

@@ -149,8 +149,14 @@ def format_response_to_markdown(api_response, agent_name = None, dataframe=None)
                 markdown.append(f"### Commentary\n{content['commentary']}\n")
 
             if 'refined_complete_code' in content:
-                clean_code = format_code_block(content['refined_complete_code']) 
-                output, json_outputs = execute_code_from_markdown(clean_code, dataframe)
+                print("content['refined_complete_code']: ", content['refined_complete_code'])
+                try:
+                    clean_code = format_code_block(content['refined_complete_code']) 
+                    output, json_outputs = execute_code_from_markdown(clean_code, dataframe)
+                except Exception as e:
+                    logger.error(f"Error in execute_code_from_markdown: {str(e)}")
+                    markdown.append(f"**Error**: {str(e)}")
+                    continue
                 
                 markdown.append(f"### Refined Complete Code\n{format_code_backticked_block(content['refined_complete_code'])}\n")
 
@@ -162,8 +168,6 @@ def format_response_to_markdown(api_response, agent_name = None, dataframe=None)
                 if json_outputs:
                     markdown.append("### Plotly JSON Outputs\n")
                     for idx, json_output in enumerate(json_outputs):
-                        if len(json_output) > 1000000:  # If JSON is larger than 1MB
-                            logger.warning(f"Large JSON output detected: {len(json_output)} bytes")
                         markdown.append(f"```plotly\n{json_output}\n```\n")
                 # print("Length of json_outputs: ", len(json_outputs))
 
@@ -174,13 +178,9 @@ def format_response_to_markdown(api_response, agent_name = None, dataframe=None)
     except Exception as e:
         logger.error(f"Error in format_response_to_markdown: {str(e)}")
         return f"**Error**: {str(e)}"
-    finally:
-        duration = time.time() - start_time
-        if duration > 1.0:
-            logger.warning(f"Slow response formatting: {duration:.2f}s")
             
     if not markdown:
-        return "No content generated"
+        return "Please provide a valid query..."
         
     return '\n'.join(markdown)