def add_df()

in databao/core/agent.py [0:0]

• 6 lines of code
• 3 McCabe index (conditional complexity)

    def add_df(self, df: DataFrame, *, name: str | None = None, context: str | Path | None = None) -> None:
        """Register a DataFrame in this agent and in the agent's DuckDB.

        Args:
            df: DataFrame to expose to executors/executors/SQL.
            name: Optional name; defaults to df1/df2/...
            context: Optional text or path to a file describing this dataset for the LLM.
        """
        df_name = name or f"df{len(self.__sources.dfs) + 1}"

        context_text = self._parse_context_arg(context) or ""

        source = DFDataSource(name=df_name, context=context_text, df=df)
        self.__sources.dfs[df_name] = source

        self.executor.register_df(source)