output for result_var name is not needed

cerebras_csv_agent.py

@@ -38,10 +38,12 @@ class CsvChatResult(BaseModel):
     casual_response: str
     
     # Data analysis components
-    analysis_operations: List[AnalysisOperation]
+    # analysis_operations: List[AnalysisOperation]
+    analysis_operations: AnalysisOperation
     
     # Visualization components
-    charts: Optional[List[ChartSpecification]] = None
+    # charts: Optional[List[ChartSpecification]] = None
+    charts: Optional[ChartSpecification] = None
     
     
 def get_csv_info(df: pd.DataFrame) -> dict:

python_code_executor_service.py

@@ -43,8 +43,10 @@ class AnalysisOperation(BaseModel):
 class CsvChatResult(BaseModel):
     """Structured response for CSV-related AI interactions"""
     casual_response: str
-    analysis_operations: List[AnalysisOperation]
-    charts: Optional[List[ChartSpecification]] = None
+    # analysis_operations: List[AnalysisOperation]
+    # charts: Optional[List[ChartSpecification]] = None
+    analysis_operations: AnalysisOperation
+    charts: ChartSpecification = None
 
 
 class PythonExecutor:
@@ -194,53 +196,105 @@ class PythonExecutor:
         return str(result)
     
     async def process_response(self, response: CsvChatResult, chat_id: str) -> str:
-     """Process the response with proper variable handling"""
-     output_parts = [response.casual_response]
+        """Process the response with proper variable handling"""
+        output_parts = [response.casual_response]
     
-     # Process analysis operations first
-     for operation in response.analysis_operations:
-         execution_result = self.execute_code(operation.code.code)
+        # Process single analysis operation
+        if response.analysis_operations:
+            operation = response.analysis_operations
+            execution_result = self.execute_code(operation.code.code)
         
-         # Get the result from locals
-         result = self.exec_locals.get(operation.result_var)
+        # Get the result from locals
+        result = self.exec_locals.get(operation.result_var)
         
-         if execution_result['error']:
-             output_parts.append(f"\n❌ Error in operation '{operation.result_var}':")
-             output_parts.append("```python\n" + execution_result['error']['message'] + "\n```")
-         elif result is not None:
-             # Handle empty/None results
-             if result is None or (hasattr(result, '__len__') and len(result) == 0):
-                 output_parts.append(f"\n⚠️ Values are missing - Operation '{operation.result_var}' returned no data")
-             else:
-                 output_parts.append(f"\n🔹 Result for '{operation.result_var}':")
-                 output_parts.append("```python\n" + self._format_result(result) + "\n```")
-         else:
-             output_str = execution_result['output'].strip()
-             if output_str:
-                 output_parts.append("```\n" + output_str + "\n```")
+        if execution_result['error']:
+            output_parts.append(f"\n❌ Error in operation '{operation.result_var}':")
+            output_parts.append("```python\n" + execution_result['error']['message'] + "\n```")
+        elif result is not None:
+            # Handle empty/None results
+            if result is None or (hasattr(result, '__len__') and len(result) == 0):
+                output_parts.append(f"\n⚠️ Values are missing - Operation '{operation.result_var}' returned no data")
+            else:
+                output_parts.append(f"\n🔹 Result for '{operation.result_var}':")
+                output_parts.append("```python\n" + self._format_result(result) + "\n```")
+        else:
+            output_str = execution_result['output'].strip()
+            if output_str:
+                output_parts.append("```\n" + output_str + "\n```")
                 
+        # Process single chart
+        if response.charts:
+            output_parts.append("\n📊 Visualizations:")
+        chart = response.charts
+        if chart.code:
+            chart_result = self.execute_code(chart.code)
+            if chart_result['plots']:
+                for plot_data in chart_result['plots']:
+                    try:
+                        public_url = await self.save_plot_to_supabase(
+                            plot_data=plot_data,
+                            description=chart.image_description,
+                            chat_id=chat_id
+                        )
+                        output_parts.append(f"\n🖼️ {chart.image_description}")
+                        output_parts.append(f"![{chart.image_description}]({public_url})")
+                    except Exception as e:
+                        output_parts.append(f"\n⚠️ Values are missing - Error uploading chart: {str(e)}")
+            elif chart_result['error']:
+                output_parts.append("```python\n" + f"Error generating {chart.image_description}: {chart_result['error']['message']}" + "\n```")
+            else:
+                output_parts.append(f"\n⚠️ Values are missing - No chart generated for '{chart.image_description}'")
     
-     # Process charts after all operations
-     if response.charts:
-         output_parts.append("\n📊 Visualizations:")
-         for chart in response.charts:
-             if chart.code:
-                 chart_result = self.execute_code(chart.code)
-                 if chart_result['plots']:
-                     for plot_data in chart_result['plots']:
-                         try:
-                             public_url = await self.save_plot_to_supabase(
-                                 plot_data=plot_data,
-                                 description=chart.image_description,
-                                 chat_id=chat_id
-                             )
-                             output_parts.append(f"\n🖼️ {chart.image_description}")
-                             output_parts.append(f"![{chart.image_description}]({public_url})")
-                         except Exception as e:
-                            output_parts.append(f"\n⚠️ Values are missing - Error uploading chart: {str(e)}")
-                 elif chart_result['error']:
-                     output_parts.append("```python\n" + f"Error generating {chart.image_description}: {chart_result['error']['message']}" + "\n```")
-                 else:
-                     output_parts.append(f"\n⚠️ Values are missing - No chart generated for '{chart.image_description}'")
+        return "\n".join(output_parts)
     
-     return "\n".join(output_parts)
+    # async def process_response(self, response: CsvChatResult, chat_id: str) -> str:
+    #  """Process the response with proper variable handling"""
+    #  output_parts = [response.casual_response]
+    
+    #  # Process analysis operations first
+    #  for operation in response.analysis_operations:
+    #      execution_result = self.execute_code(operation.code.code)
+        
+    #      # Get the result from locals
+    #      result = self.exec_locals.get(operation.result_var)
+        
+    #      if execution_result['error']:
+    #          output_parts.append(f"\n❌ Error in operation '{operation.result_var}':")
+    #          output_parts.append("```python\n" + execution_result['error']['message'] + "\n```")
+    #      elif result is not None:
+    #          # Handle empty/None results
+    #          if result is None or (hasattr(result, '__len__') and len(result) == 0):
+    #              output_parts.append(f"\n⚠️ Values are missing - Operation '{operation.result_var}' returned no data")
+    #          else:
+    #              output_parts.append(f"\n🔹 Result for '{operation.result_var}':")
+    #              output_parts.append("```python\n" + self._format_result(result) + "\n```")
+    #      else:
+    #          output_str = execution_result['output'].strip()
+    #          if output_str:
+    #              output_parts.append("```\n" + output_str + "\n```")
+                
+    
+    #  # Process charts after all operations
+    #  if response.charts:
+    #      output_parts.append("\n📊 Visualizations:")
+    #      for chart in response.charts:
+    #          if chart.code:
+    #              chart_result = self.execute_code(chart.code)
+    #              if chart_result['plots']:
+    #                  for plot_data in chart_result['plots']:
+    #                      try:
+    #                          public_url = await self.save_plot_to_supabase(
+    #                              plot_data=plot_data,
+    #                              description=chart.image_description,
+    #                              chat_id=chat_id
+    #                          )
+    #                          output_parts.append(f"\n🖼️ {chart.image_description}")
+    #                          output_parts.append(f"![{chart.image_description}]({public_url})")
+    #                      except Exception as e:
+    #                         output_parts.append(f"\n⚠️ Values are missing - Error uploading chart: {str(e)}")
+    #              elif chart_result['error']:
+    #                  output_parts.append("```python\n" + f"Error generating {chart.image_description}: {chart_result['error']['message']}" + "\n```")
+    #              else:
+    #                  output_parts.append(f"\n⚠️ Values are missing - No chart generated for '{chart.image_description}'")
+    
+    #  return "\n".join(output_parts)