Merge branch 'add/skills' of https://github.com/gventuri/pandas-ai into add/skills

Author: ArslanSaleem

pandasai/config.py

@@ -143,4 +143,4 @@ def __str__(cls) -> str:
         Returns:
             str: String representation of all skills
         """
-        return "".join(str(skill) for skill in cls._skills)
+        return "\n".join(str(skill) for skill in cls._skills)

pandasai/core/code_generation/code_cleaning.py

@@ -32,7 +32,7 @@ def _check_direct_sql_func_def_exists(self, node: ast.AST) -> bool:
 
     def _check_if_skill_func_def_exists(self, node: ast.AST) -> bool:
         """
-        Check if the node defines a direct SQL execution function.
+        Check if the node defines a skill function.
         """
         for skill in self.context.skills:
             if isinstance(node, ast.FunctionDef) and node.name == skill.name:

pandasai/skills/__init__.py

@@ -27,7 +27,7 @@ def __init__(
             description: The description of the skill.
                 Defaults to the function docstring.
             name: The name of the function. Mandatory when `func` is a lambda.
-                Defaults to the functions name.
+                Defaults to the function's name.
             **kwargs: additional params
         """
 
@@ -84,7 +84,7 @@ def skill(*args: Union[str, Callable]) -> Callable:
         .. code-block:: python
 
             @skill
-            def compute_flight_prices(offers: pd.Dataframe) -> List[float]:
+            def compute_flight_prices(offers: pd.DataFrame) -> List[float]:
                 \"\"\"Computes the flight prices\"\"\"
                 return