ai-data-science-team 0.0.0.9012__py3-none-any.whl → 0.0.0.9014__py3-none-any.whl

ai_data_science_team/multiagents/sql_data_analyst.py

@@ -1,12 +1,14 @@
 
 from langchain_core.messages import BaseMessage
-from langgraph.types import Checkpointer
+
+from langchain.prompts import PromptTemplate
+from langchain_core.output_parsers import JsonOutputParser
 
 from langgraph.graph import START, END, StateGraph
 from langgraph.graph.state import CompiledStateGraph
-from langgraph.types import Command
+from langgraph.types import Checkpointer
 
-from typing import TypedDict, Annotated, Sequence, Literal
+from typing import TypedDict, Annotated, Sequence
 import operator
 
 from typing_extensions import TypedDict
@@ -20,6 +22,7 @@ from ai_data_science_team.agents import SQLDatabaseAgent, DataVisualizationAgent
 from ai_data_science_team.utils.plotly import plotly_from_dict
 from ai_data_science_team.utils.regex import remove_consecutive_duplicates, get_generic_summary
 
+AGENT_NAME = "sql_data_analyst"
 
 class SQLDataAnalyst(BaseAgent):
     """
@@ -33,6 +36,8 @@ class SQLDataAnalyst(BaseAgent):
         The SQL Database Agent.
     data_visualization_agent: DataVisualizationAgent
         The Data Visualization Agent.
+    checkpointer: Checkpointer (optional)
+        The checkpointer to save the state of the multi-agent system.
         
     Methods:
     --------
@@ -326,17 +331,17 @@ def make_sql_data_analyst(
     """
     Creates a multi-agent system that takes in a SQL query and returns a plot or table.
     
-    - Agent 1: SQL Database Agent made with `make_sql_database_agent()`
-    - Agent 2: Data Visualization Agent made with `make_data_visualization_agent()`
+    - Agent 1: SQL Database Agent made with `SQLDatabaseAgent()`
+    - Agent 2: Data Visualization Agent made with `DataVisualizationAgent()`
     
     Parameters:
     ----------
     model: 
         The language model to be used for the agents.
     sql_database_agent: CompiledStateGraph
-        The SQL Database Agent made with `make_sql_database_agent()`.
+        The SQL Database Agent made with `SQLDatabaseAgent()`.
     data_visualization_agent: CompiledStateGraph
-        The Data Visualization Agent made with `make_data_visualization_agent()`.
+        The Data Visualization Agent made with `DataVisualizationAgent()`.
     checkpointer: Checkpointer (optional)
         The checkpointer to save the state of the multi-agent system.
         Default: None
@@ -348,10 +353,39 @@ def make_sql_data_analyst(
     """
     
     llm = model
+    
+    
+    routing_preprocessor_prompt = PromptTemplate(
+        template="""
+        You are an expert in routing decisions for a SQL Database Agent, a Charting Visualization Agent, and a Pandas Table Agent. Your job is to:
+        
+        1. Determine what the correct format for a Users Question should be for use with a SQL Database Agent based on the incoming user question. Anything related to database and data manipulation should be passed along.
+        2. Determine whether or not a chart should be generated or a table should be returned based on the users question.
+        3. If a chart is requested, determine the correct format of a Users Question should be used with a Data Visualization Agent. Anything related to plotting and visualization should be passed along.
+        
+        Use the following criteria on how to route the the initial user question:
+        
+        From the incoming user question, remove any details about the format of the final response as either a Chart or Table and return only the important part of the incoming user question that is relevant for the SQL generator agent. This will be the 'user_instructions_sql_database'. If 'None' is found, return the original user question.
+        
+        Next, determine if the user would like a data visualization ('chart') or a 'table' returned with the results of the Data Wrangling Agent. If unknown, not specified or 'None' is found, then select 'table'.  
+        
+        If a 'chart' is requested, return the 'user_instructions_data_visualization'. If 'None' is found, return None.
+        
+        Return JSON with 'user_instructions_sql_database', 'user_instructions_data_visualization' and 'routing_preprocessor_decision'.
+        
+        INITIAL_USER_QUESTION: {user_instructions}
+        """,
+        input_variables=["user_instructions"]
+    )
+
+    routing_preprocessor = routing_preprocessor_prompt | llm | JsonOutputParser()
 
     class PrimaryState(TypedDict):
         messages: Annotated[Sequence[BaseMessage], operator.add]    
         user_instructions: str
+        user_instructions_sql_database: str
+        user_instructions_data_visualization: str
+        routing_preprocessor_decision: str
         sql_query_code: str
         sql_database_function: str
         data_sql: dict
@@ -359,39 +393,94 @@ def make_sql_data_analyst(
         plot_required: bool
         data_visualization_function: str
         plotly_graph: dict
+        plotly_error: str
         max_retries: int
         retry_count: int
         
-    def route_to_visualization(state) -> Command[Literal["data_visualization_agent", "__end__"]]: 
+    def preprocess_routing(state: PrimaryState):
+        print("---SQL DATA ANALYST---")
+        print("*************************")
+        print("---PREPROCESS ROUTER---")
+        question = state.get("user_instructions")
         
-        response = llm.invoke(f"Respond in 1 word ('plot' or 'table'). Is the user requesting a plot? If unknown, select 'table'. \n\n User Instructions:\n{state.get('user_instructions')}")
+        # Chart Routing and SQL Prep
+        response = routing_preprocessor.invoke({"user_instructions": question})
         
-        if response.content == 'plot':
-            plot_required = True
-            goto="data_visualization_agent"
-        else:
-            plot_required = False
-            goto="__end__"
+        return {
+            "user_instructions_sql_database": response.get('user_instructions_sql_database'),
+            "user_instructions_data_visualization": response.get('user_instructions_data_visualization'),
+            "routing_preprocessor_decision": response.get('routing_preprocessor_decision'),
+        }
+    
+    def router_chart_or_table(state: PrimaryState):
+        print("---ROUTER: CHART OR TABLE---")
+        return "chart" if state.get('routing_preprocessor_decision') == "chart" else "table"
+    
+    
+    def invoke_sql_database_agent(state: PrimaryState):
         
-        return Command(
-            update={
-                'data_raw': state.get("data_sql"),
-                'plot_required': plot_required,
-            },
-            goto=goto
-        )
+        response = sql_database_agent.invoke({
+            "user_instructions": state.get("user_instructions_sql_database"),
+            "max_retries": state.get("max_retries"),
+            "retry_count": state.get("retry_count"),
+        })
 
-    workflow = StateGraph(PrimaryState)
+        return {
+            "messages": response.get("messages"),
+            "data_sql": response.get("data_sql"),
+            "sql_query_code": response.get("sql_query_code"),
+            "sql_database_function": response.get("sql_database_function"),
+            
+        }
+        
+    def invoke_data_visualization_agent(state: PrimaryState):
+        
+        response = data_visualization_agent.invoke({
+            "user_instructions": state.get("user_instructions_data_visualization"),
+            "data_raw": state.get("data_sql"),
+            "max_retries": state.get("max_retries"),
+            "retry_count": state.get("retry_count"),
+        })
+        
+        return {
+            "messages": response.get("messages"),
+            "data_visualization_function": response.get("data_visualization_function"),
+            "plotly_graph": response.get("plotly_graph"),
+            "plotly_error": response.get("data_visualization_error"),
+        }
 
-    workflow.add_node("sql_database_agent", sql_database_agent)
-    workflow.add_node("route_to_visualization", route_to_visualization)
-    workflow.add_node("data_visualization_agent", data_visualization_agent)
+    def route_printer(state: PrimaryState):
+        print("---ROUTE PRINTER---")
+        print(f"    Route: {state.get('routing_preprocessor_decision')}")
+        print("---END---")
+        return {}
+
+    workflow = StateGraph(PrimaryState)
+    
+    workflow.add_node("routing_preprocessor", preprocess_routing)
+    workflow.add_node("sql_database_agent", invoke_sql_database_agent)
+    workflow.add_node("data_visualization_agent", invoke_data_visualization_agent)
+    workflow.add_node("route_printer", route_printer)
 
-    workflow.add_edge(START, "sql_database_agent")
-    workflow.add_edge("sql_database_agent", "route_to_visualization")
-    workflow.add_edge("data_visualization_agent", END)
+    workflow.add_edge(START, "routing_preprocessor")
+    workflow.add_edge("routing_preprocessor", "sql_database_agent")
+    
+    workflow.add_conditional_edges(
+        "sql_database_agent", 
+        router_chart_or_table,
+        {
+            "chart": "data_visualization_agent",
+            "table": "route_printer"
+        }
+    )
+    
+    workflow.add_edge("data_visualization_agent", "route_printer")
+    workflow.add_edge("route_printer", END)
 
-    app = workflow.compile(checkpointer=checkpointer)
+    app = workflow.compile(
+        checkpointer=checkpointer, 
+        name=AGENT_NAME
+    )
 
     return app
 

ai_data_science_team/templates/agent_templates.py

@@ -40,6 +40,21 @@ class BaseAgent(CompiledStateGraph):
         self._params = params
         self._compiled_graph = self._make_compiled_graph()
         self.response = None
+        self.name = self._compiled_graph.name
+        self.checkpointer = self._compiled_graph.checkpointer
+        self.store = self._compiled_graph.store
+        self.output_channels = self._compiled_graph.output_channels
+        self.nodes = self._compiled_graph.nodes
+        self.stream_mode = self._compiled_graph.stream_mode
+        self.builder = self._compiled_graph.builder
+        self.channels = self._compiled_graph.channels
+        self.input_channels = self._compiled_graph.input_channels
+        self.input_schema = self._compiled_graph.input_schema
+        self.output_schema = self._compiled_graph.output_schema
+        self.debug = self._compiled_graph.debug
+        self.interrupt_after_nodes = self._compiled_graph.interrupt_after_nodes
+        self.interrupt_before_nodes = self._compiled_graph.interrupt_before_nodes
+        self.config = self._compiled_graph.config
 
     def _make_compiled_graph(self):
         """
@@ -197,6 +212,24 @@ class BaseAgent(CompiledStateGraph):
         """
         return self.get_output_jsonschema()['properties']
     
+    def get_state(self, config, *, subgraphs = False):
+        """
+        Returns the state of the agent.
+        """
+        return self._compiled_graph.get_state(config, subgraphs=subgraphs)
+    
+    def get_state_history(self, config, *, filter = None, before = None, limit = None):
+        """
+        Returns the state history of the agent.
+        """
+        return self._compiled_graph.get_state_history(config, filter=filter, before=before, limit=limit)
+    
+    def update_state(self, config, values, as_node = None):
+        """
+        Updates the state of the agent.
+        """
+        return self._compiled_graph.update_state(config, values, as_node)
+    
     def get_response(self):
         """
         Returns the response generated by the agent.
@@ -237,6 +270,7 @@ def create_coding_agent_graph(
     checkpointer: Optional[Callable] = None,
     bypass_recommended_steps: bool = False,
     bypass_explain_code: bool = False,
+    agent_name: str = "coding_agent"
 ):
     """
     Creates a generic agent graph using the provided node functions and node names.
@@ -281,6 +315,8 @@ def create_coding_agent_graph(
         Whether to skip the recommended steps node.
     bypass_explain_code : bool, optional
         Whether to skip the final explain code node.
+    name : str, optional
+        The name of the agent graph.
 
     Returns
     -------
@@ -366,10 +402,10 @@ def create_coding_agent_graph(
         workflow.add_edge(explain_code_node_name, END)
     
     # Finally, compile
-    if human_in_the_loop:
-        app = workflow.compile(checkpointer=checkpointer)
-    else:
-        app = workflow.compile()
+    app = workflow.compile(
+        checkpointer=checkpointer,
+        name=agent_name,
+    )
     
     return app
 
@@ -574,7 +610,7 @@ def node_func_execute_agent_from_sql_connection(
     
     # Retrieve SQLAlchemy connection and code snippet from the state
     is_engine = isinstance(connection, sql.engine.base.Engine)
-    conn = connection.connect() if is_engine else connection
+    connection = connection.connect() if is_engine else connection
     agent_code = state.get(code_snippet_key)
     
     # Ensure the connection object is provided

ai_data_science_team/tools/dataframe.py

@@ -74,7 +74,12 @@ def get_dataframe_summary(
     return summaries
 
 
-def _summarize_dataframe(df: pd.DataFrame, dataset_name: str, n_sample=30, skip_stats=False) -> str:
+def _summarize_dataframe(
+    df: pd.DataFrame, 
+    dataset_name: str, 
+    n_sample=30, 
+    skip_stats=False
+) -> str:
     """Generate a summary string for a single DataFrame."""
     # 1. Convert dictionary-type cells to strings
     #    This prevents unhashable dict errors during df.nunique().

ai_data_science_team/tools/eda.py

@@ -2,11 +2,44 @@
 from typing import Annotated, Dict, Tuple, Union
 
 import os
+import tempfile
 
 from langchain.tools import tool
 
 from langgraph.prebuilt import InjectedState  
 
+from ai_data_science_team.tools.dataframe import get_dataframe_summary
+
+
+@tool(response_format='content')
+def explain_data(
+    data_raw: Annotated[dict, InjectedState("data_raw")],
+    n_sample: int = 30,
+    skip_stats: bool = False,
+):
+    """
+    Tool: explain_data
+    Description:
+        Provides an extensive, narrative summary of a DataFrame including its shape, column types,
+        missing value percentages, unique counts, sample rows, and (if not skipped) descriptive stats/info.
+
+    Parameters:
+        data_raw (dict): Raw data.
+        n_sample (int, default=30): Number of rows to display.
+        skip_stats (bool, default=False): If True, omit descriptive stats/info.
+
+    LLM Guidance:
+        Use when a detailed, human-readable explanation is needed—i.e., a full overview is preferred over a concise numerical summary.
+
+    Returns:
+        str: Detailed DataFrame summary.
+    """
+    print("    * Tool: explain_data")
+    import pandas as pd
+    
+    result = get_dataframe_summary(pd.DataFrame(data_raw), n_sample=n_sample, skip_stats=skip_stats)
+    
+    return result
 
 @tool(response_format='content_and_artifact')
 def describe_dataset(
@@ -15,21 +48,33 @@ def describe_dataset(
     """
     Tool: describe_dataset
     Description:
-        Describe the dataset by computing summary
-        statistics using the DataFrame's describe() method.
-        
+        Compute and return summary statistics for the dataset using pandas' describe() method.
+        The tool provides both a textual summary and a structured artifact (a dictionary) for further processing.
+
+    Parameters:
+    -----------
+    data_raw : dict
+        The raw data in dictionary format.
+
+    LLM Selection Guidance:
+    ------------------------
+    Use this tool when:
+      - The request emphasizes numerical descriptive statistics (e.g., count, mean, std, min, quartiles, max).
+      - The user needs a concise statistical snapshot rather than a detailed narrative.
+      - Both a brief text explanation and a structured data artifact (for downstream tasks) are required.
+
     Returns:
     -------
     Tuple[str, Dict]:
-        content: A textual summary of the DataFrame's descriptive statistics.
-        artifact: A dictionary (from DataFrame.describe()) for further inspection.
+        - content: A textual summary indicating that summary statistics have been computed.
+        - artifact: A dictionary (derived from DataFrame.describe()) containing detailed statistical measures.
     """
     print("    * Tool: describe_dataset")
     import pandas as pd
     df = pd.DataFrame(data_raw)
     description_df = df.describe(include='all')
     content = "Summary statistics computed using pandas describe()."
-    artifact = description_df.to_dict()
+    artifact = {'describe_df': description_df.to_dict()}
     return content, artifact
 
 
@@ -226,8 +271,8 @@ def generate_sweetviz_report(
     data_raw: Annotated[dict, InjectedState("data_raw")],
     target: str = None,
     report_name: str = "sweetviz_report.html",
-    report_directory: str = os.path.join(os.getcwd(), "reports"),
-    open_browser: bool = True,
+    report_directory: str = None,  # <-- Default to None
+    open_browser: bool = False,
 ) -> Tuple[str, Dict]:
     """
     Tool: generate_sweetviz_report
@@ -243,9 +288,10 @@ def generate_sweetviz_report(
     report_name : str, optional
         The file name to save the Sweetviz HTML report. Default is "sweetviz_report.html".
     report_directory : str, optional
-        The directory where the report should be saved. Defaults to a 'reports' directory in the current working directory.
+        The directory where the report should be saved. 
+        If None, a temporary directory is created and used.
     open_browser : bool, optional
-        Whether to open the report in a web browser. Default is True.
+        Whether to open the report in a web browser. Default is False.
     
     Returns:
     --------
@@ -254,28 +300,37 @@ def generate_sweetviz_report(
         artifact: A dictionary with the report file path and optionally the report's HTML content.
     """
     print("    * Tool: generate_sweetviz_report")
+
+    # Import sweetviz
     try:
         import sweetviz as sv
     except ImportError:
         raise ImportError("Please install the 'sweetviz' package to use this tool. Run: pip install sweetviz")
+    
     import pandas as pd
+    
     # Convert injected raw data to a DataFrame.
     df = pd.DataFrame(data_raw)
     
+    # If no directory is specified, use a temporary directory.
+    if not report_directory:
+        report_directory = tempfile.mkdtemp()
+        print(f"    * Using temporary directory: {report_directory}")
+    else:
+        # Ensure user-specified directory exists.
+        if not os.path.exists(report_directory):
+            os.makedirs(report_directory)
+    
     # Create the Sweetviz report.
     report = sv.analyze(df, target_feat=target)
     
-    # Ensure the directory exists; default is os.getcwd()/reports
-    if not os.path.exists(report_directory):
-        os.makedirs(report_directory)
-    
     # Determine the full path for the report.
     full_report_path = os.path.join(report_directory, report_name)
     
     # Save the report to the specified HTML file.
     report.show_html(
         filepath=full_report_path,
-        open_browser=True,
+        open_browser=open_browser,
     )
     
     # Optionally, read the HTML content (if desired to pass along in the artifact).
@@ -285,9 +340,13 @@ def generate_sweetviz_report(
     except Exception:
         html_content = None
     
-    content = f"Sweetviz EDA report generated and saved as '{os.path.abspath(full_report_path)}'."
+    content = (
+        f"Sweetviz EDA report generated and saved as '{os.path.abspath(full_report_path)}'. "
+        f"{'This was saved in a temporary directory.' if 'tmp' in report_directory else ''}"
+    )
     artifact = {
         "report_file": os.path.abspath(full_report_path),
         "report_html": html_content,
     }
     return content, artifact
+

ai_data_science_team/utils/messages.py

@@ -0,0 +1,27 @@
+
+
+
+def get_tool_call_names(messages):
+    """
+    Method to extract the tool call names from a list of LangChain messages.
+    
+    Parameters:
+    ----------
+    messages : list
+        A list of LangChain messages.
+        
+    Returns:
+    -------
+    tool_calls : list
+        A list of tool call names.
+    
+    """
+    tool_calls = []
+    for message in messages:
+        try: 
+            if "tool_call_id" in list(dict(message).keys()):
+                tool_calls.append(message.name)
+        except:
+            pass
+    return tool_calls
+

{ai_data_science_team-0.0.0.9012.dist-info → ai_data_science_team-0.0.0.9014.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: ai-data-science-team
-Version: 0.0.0.9012
+Version: 0.0.0.9014
 Summary: Build and run an AI-powered data science team.
 Home-page: https://github.com/business-science/ai-data-science-team
 Author: Matt Dancho
@@ -18,7 +18,7 @@ Requires-Dist: langchain
 Requires-Dist: langchain_community
 Requires-Dist: langchain_openai
 Requires-Dist: langchain_experimental
-Requires-Dist: langgraph>=0.2.57
+Requires-Dist: langgraph>=0.2.74
 Requires-Dist: openai
 Requires-Dist: pandas
 Requires-Dist: sqlalchemy
@@ -152,7 +152,11 @@ This is a top secret project I'm working on. It's a multi-agent data science app
 
 #### 🔥 Agentic Applications
 
-1. **SQL Database Agent App:** Connects any SQL Database, generates SQL queries from natural language, and returns data as a downloadable table. [See Application](/apps/sql-database-agent-app/)
+1. **NEW Exploratory Data Copilot**: An AI-powered data science app that performs automated exploratory data analysis (EDA) with EDA Reporting, Missing Data Analysis, Correlation Analysis, and more. [See Application](/apps/exploratory-copilot-app/)
+
+![Exploratory Data Copilot](/img/apps/ai_exploratory_copilot.jpg)
+
+2. **SQL Database Agent App:** Connects any SQL Database, generates SQL queries from natural language, and returns data as a downloadable table. [See Application](/apps/sql-database-agent-app/)
 
 ### Agents Available Now
 

{ai_data_science_team-0.0.0.9012.dist-info → ai_data_science_team-0.0.0.9014.dist-info}/RECORD

@@ -1,31 +1,32 @@
-ai_data_science_team/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_data_science_team/_version.py,sha256=BybGt-zGNDZsdJxDMV3xmjghiRF8jmwG3ov_dt_rM7E,26
+ai_data_science_team/__init__.py,sha256=LmogkhGnxvvVe1ukJM6I6lXy4B7SuCr5eXZpwjyDMKQ,444
+ai_data_science_team/_version.py,sha256=D4dUl-fYnimOU_VSzvrmJm30_IoaF_9m9dTLp8HE6rQ,26
 ai_data_science_team/orchestration.py,sha256=xiIFOsrLwPdkSmtme7wNCCGv8XopnMTNElNzlZokL-4,303
 ai_data_science_team/agents/__init__.py,sha256=Gnotza9SKr_0IxuaX8k1nsZK48wXkkeZcGcrR1EqNks,668
-ai_data_science_team/agents/data_cleaning_agent.py,sha256=V5tJMwGJK0JwrF_H-7r3S0E8UkAY6ci4BGxqjhZiGBI,27352
-ai_data_science_team/agents/data_loader_tools_agent.py,sha256=fnkOvmrXzvTTt1mnAyTlsF_7ZGrkp3P97YU_LgeffMg,8445
-ai_data_science_team/agents/data_visualization_agent.py,sha256=tJy9Ehnh9mvAu6H--TXI8esSHmK1RW_L1RDAdn7Xek4,28821
-ai_data_science_team/agents/data_wrangling_agent.py,sha256=LxzphH-TmrFG0GjejGOjulhPq4SsWFo5Y9tk4WEuN4M,32347
-ai_data_science_team/agents/feature_engineering_agent.py,sha256=KmPBkj7WUBz6LFUlDDfQHMi7ujXwsH5P9LWRS-F4tdM,31026
-ai_data_science_team/agents/sql_database_agent.py,sha256=1K2o3NiuKgGKdbMz_Tq9IeQ8xhXjpfGOxx9lArZh1yE,31173
+ai_data_science_team/agents/data_cleaning_agent.py,sha256=aZLhnN2EBlY_hmAg_r73dwi1w5utSFNEgEs8aWl8Cho,27991
+ai_data_science_team/agents/data_loader_tools_agent.py,sha256=TFKzYqV6cvU-sMbfL-hg8-NgF_Hz3nysGFldvb5K3fM,9327
+ai_data_science_team/agents/data_visualization_agent.py,sha256=eUSTzTOm5aLJ6Cqnk-hRuXeVbYyy0RIzN8_0LLy0P9o,29387
+ai_data_science_team/agents/data_wrangling_agent.py,sha256=6tiDO1i-5s2Ju6_MsLoJMflUuRSf_1oTSsSKcLlgzEc,33376
+ai_data_science_team/agents/feature_engineering_agent.py,sha256=xZGDFnmM6wx4bi3e4c_dNOZzGcxBmX8k0iveL7dlA-k,31608
+ai_data_science_team/agents/sql_database_agent.py,sha256=fln8unefn5Jd2exeyGs-9PljyLXAK60HI81tJACYeCY,31726
 ai_data_science_team/ds_agents/__init__.py,sha256=dnuagUTebTDHhGXbCt-hZIilzXMSUwyHaEI7sOxhvoE,95
-ai_data_science_team/ds_agents/eda_tools_agent.py,sha256=y65lsBXhQNOGwWealEho6uFxGSTW7FNfvTUZnW8_XNY,7609
+ai_data_science_team/ds_agents/eda_tools_agent.py,sha256=x0kTwDo0BNbYzgA0YamMWdqRjx0upZgeXp9nF6C6_8E,8364
 ai_data_science_team/ds_agents/modeling_tools_agent.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_data_science_team/ml_agents/__init__.py,sha256=qq3UlDCRV_z4FHQ1jj3YR6zPbA6kuCvYCisj_bHYfO4,190
-ai_data_science_team/ml_agents/h2o_ml_agent.py,sha256=DamR72agrTKfdcdhablmP2mpbj0CqtMonP-QU8p7o9w,33394
+ai_data_science_team/ml_agents/h2o_ml_agent.py,sha256=S0uayngaVwVUyA4zy05QYlq5NXrNHb723NeF2rns0Y0,33934
 ai_data_science_team/ml_agents/h2o_ml_tools_agent.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_data_science_team/ml_agents/mlflow_tools_agent.py,sha256=zbT0KIsmQp_sEyxzXRguhqx5913Q2yPYyKGU6TUWEM8,11067
-ai_data_science_team/multiagents/__init__.py,sha256=aI4GztEwmkexZKT5XHcH3cAjO-xYUhncb3yfPJQDqTA,99
-ai_data_science_team/multiagents/sql_data_analyst.py,sha256=kmmED3gLf5STWWY6ZVJYd7_Pt8NMl6SHyBocuQzRDGk,14193
+ai_data_science_team/ml_agents/mlflow_tools_agent.py,sha256=QImaZnS8hPdrU7GI6pZ0dUDO-LXx40MSA3XyMDppIh0,12003
+ai_data_science_team/multiagents/__init__.py,sha256=5tpmZBQ_UT5SKDCS_NivZhN19HEStKIcstiqSXPXDl0,208
+ai_data_science_team/multiagents/pandas_data_analyst.py,sha256=O662v-75tLqHHrVNjncsPeR2FB4MWSWruJRIF-YO-fg,13581
+ai_data_science_team/multiagents/sql_data_analyst.py,sha256=Fpue6WcX9x18kEH3kfEi8kkFoG9HhQ1AZiWw6Y6FXOo,18502
 ai_data_science_team/multiagents/supervised_data_analyst.py,sha256=uduCYpicga-UCf9nPQktQggW96-HDlqvioYmEdWejtI,158
 ai_data_science_team/parsers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_data_science_team/parsers/parsers.py,sha256=hIsMZXRHz9hqs8R1ebymKA7D6NxOf5UVMpDAr_gGhE8,2027
 ai_data_science_team/templates/__init__.py,sha256=_IcyFUu_mM8dFtttz95h0csJZ-XWDP3cEFuf22-R5RM,330
-ai_data_science_team/templates/agent_templates.py,sha256=Lezp0ugtIP3m5WUOmjLwghNnjjyQVQecysONeIHWwi0,29133
+ai_data_science_team/templates/agent_templates.py,sha256=QHRNZVmIfeClEef2Fr2Wb9J2GG91REJOKUUEY71Dszs,30767
 ai_data_science_team/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_data_science_team/tools/data_loader.py,sha256=ITs_6UAJ0m9h68R9_LruiaJSElv9l7SxTQYryI7YZPY,14702
-ai_data_science_team/tools/dataframe.py,sha256=qSflGDByqqCXv4TjuvOFvGPZmegzeOesb0Y4i4Y0gdQ,4551
-ai_data_science_team/tools/eda.py,sha256=UGD6PC12RsB_UmStvR4TmSqv0noxjM4DkzY-kHjI0-E,10591
+ai_data_science_team/tools/dataframe.py,sha256=cckplDWu9SsA_PRo89pYsyVCmBE0PoDIwMv6tuLunT4,4572
+ai_data_science_team/tools/eda.py,sha256=KoryXso_5zOPDq7jwcUAMEXV-AIzpWb62zzbUHVtgtM,12687
 ai_data_science_team/tools/h2o.py,sha256=gSK0f2FULfAfipFTTjDMUS6DjHwFFvvl4jxshr6QpS0,38997
 ai_data_science_team/tools/mlflow.py,sha256=8NTkSOvbTk01GOmwFaMkLBRse80w9Kk7Ypi6Fv4kTII,29475
 ai_data_science_team/tools/sql.py,sha256=vvz_CiOg6GqXo2_mlF4kq5IS6if79dpaizAgLR9sRyg,4784
@@ -33,10 +34,11 @@ ai_data_science_team/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 ai_data_science_team/utils/html.py,sha256=1MBcjNyATi3FPOyVdqf6-_QYCJmDVQWmVPIInUr50dk,628
 ai_data_science_team/utils/logging.py,sha256=7wFOv6GGhXR_RPbh-8p0GyrS608XOnZtiaGK2IbDl_s,2081
 ai_data_science_team/utils/matplotlib.py,sha256=d6DZfCXvZ5Kocxtsp92etIymKW2cRBcUG9GmCOMtgJo,1145
+ai_data_science_team/utils/messages.py,sha256=feWIPGsv8ly9jpNnS97SoPsn1feaY1Km0VCbHTbRpI8,549
 ai_data_science_team/utils/plotly.py,sha256=nST-NG0oizKVHhH6HsjHUpTUumq9bCccBdxjuaJWnVQ,504
 ai_data_science_team/utils/regex.py,sha256=lwarbLqTA2VfNQSyqKCl-PBlH_0WH3zXZvYGBYGUiu4,5144
-ai_data_science_team-0.0.0.9012.dist-info/LICENSE,sha256=Xif0IRLdd2HGLATxV2EVp91aSY6KOuacRr_6BorKGzA,1084
-ai_data_science_team-0.0.0.9012.dist-info/METADATA,sha256=geRCFLG3YO9uprp_CGKiqCTSThg06L2U6WxVqYKzyM8,12704
-ai_data_science_team-0.0.0.9012.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ai_data_science_team-0.0.0.9012.dist-info/top_level.txt,sha256=CnoMgOphCoAdGTLueWdCVByVyjwOubaGiTB1lchdy4M,21
-ai_data_science_team-0.0.0.9012.dist-info/RECORD,,
+ai_data_science_team-0.0.0.9014.dist-info/LICENSE,sha256=Xif0IRLdd2HGLATxV2EVp91aSY6KOuacRr_6BorKGzA,1084
+ai_data_science_team-0.0.0.9014.dist-info/METADATA,sha256=a35LbXyxf_XiP82m_P5HLFwPrmuzXkNWbyfzGge7dHA,13021
+ai_data_science_team-0.0.0.9014.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ai_data_science_team-0.0.0.9014.dist-info/top_level.txt,sha256=CnoMgOphCoAdGTLueWdCVByVyjwOubaGiTB1lchdy4M,21
+ai_data_science_team-0.0.0.9014.dist-info/RECORD,,