Viz agent num type

[codecov[bot]]

Codecov Report

Attention: Patch coverage is 74.87521% with 151 lines in your changes are missing coverage. Please review.

Project coverage is 78.08%. Comparing base (2502d04) to head (7564404).

Files	Patch %	Lines
pandasai/ee/helpers/query_builder.py	63.63%	72 Missing :warning:
.../visualization_agent/prompts/generate_df_schema.py	26.47%	25 Missing :warning:
...ts/visualization_agent/prompts/viz_agent_prompt.py	31.81%	15 Missing :warning:
pandasai/ee/agents/visualization_agent/__init__.py	85.96%	8 Missing :warning:
pandasai/prompts/base.py	0.00%	7 Missing :warning:
pandasai/agent/base.py	75.00%	5 Missing :warning:
pandasai/connectors/pandas.py	73.68%	5 Missing :warning:
...ualization_agent/pipeline/viz_prompt_generation.py	64.28%	5 Missing :warning:
...nts/visualization_agent/pipeline/code_generator.py	96.73%	3 Missing :warning:
...r_correction_pipeline/error_correction_pipeline.py	90.00%	2 Missing :warning:
... and 4 more

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #1172 +/- ## ========================================== - Coverage 78.56% 78.08% -0.49% ========================================== Files 117 128 +11 Lines 4624 5197 +573 ========================================== + Hits 3633 4058 +425 - Misses 991 1139 +148 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.

[ArslanSaleem]

Covered in the main PR