diff --git a/pyproject.toml b/pyproject.toml index c0a624a1..ea4e4b76 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "flit_core.buildapi" [project] name = "vanna" -version = "0.0.19" +version = "0.0.20" authors = [ { name="Zain Hoda", email="zain@vanna.ai" }, ] diff --git a/src/vanna/__init__.py b/src/vanna/__init__.py index fde17752..c4d52b99 100644 --- a/src/vanna/__init__.py +++ b/src/vanna/__init__.py @@ -1068,7 +1068,7 @@ def generate_followup_questions(question: str, df: pd.DataFrame) -> List[str]: params = [DataResult( question=question, sql=None, - table_markdown=df.head().to_markdown(), + table_markdown="", error=None, correction_attempts=0, )] @@ -1258,7 +1258,7 @@ def generate_plotly_code(question: Union[str, None], sql: Union[str, None], df: params = [DataResult( question=question, sql=sql, - table_markdown=df.head().to_markdown(), + table_markdown=str(df.dtypes), error=None, correction_attempts=0, )]