ai-data-science-team 0.0.0.9007__py3-none-any.whl → 0.0.0.9009__py3-none-any.whl

ai_data_science_team/tools/regex.py

@@ -76,4 +76,89 @@ def format_agent_name(agent_name: str) -> str:
     
     formatted_name = agent_name.strip().replace("_", " ").upper()
     
-    return f"---{formatted_name}----"
+    return f"---{formatted_name}----"
+
+def format_recommended_steps(raw_text: str, heading: str = "# Recommended Steps:") -> str:
+    # Split text by newline and strip leading/trailing whitespace
+    lines = raw_text.strip().split('\n')
+    
+    # Remove empty lines from the start
+    while lines and not lines[0].strip():
+        lines.pop(0)
+
+    seen_heading = False
+    new_lines = []
+
+    for line in lines:
+        # If this line *is exactly* the heading, check if we've seen it already
+        if line.strip() == heading:
+            if seen_heading:
+                # Skip duplicates
+                continue
+            else:
+                seen_heading = True
+        new_lines.append(line)
+
+    # If heading was never seen, prepend it
+    if not seen_heading:
+        new_lines.insert(0, heading)
+
+    return "\n".join(new_lines)
+
+def get_generic_summary(report_dict: dict, code_lang = "python") -> str:
+    """
+    Takes a dictionary of unknown structure (e.g., from json.loads(...)) 
+    and returns a textual summary. It assumes:
+      1) 'report_title' (if present) should be displayed first.
+      2) If a key includes 'code' or 'function', 
+         the value is treated as a code block.
+      3) Otherwise, key-value pairs are displayed as text.
+
+    Parameters
+    ----------
+    report_dict : dict
+        The dictionary holding the agent output or user report.
+
+    Returns
+    -------
+    str
+        A formatted summary string.
+    """
+    # 1) Grab the report title (or default)
+    title = report_dict.get("report_title", "Untitled Report")
+
+    lines = []
+    lines.append(f"# {title}")
+
+    # 2) Iterate over all other keys
+    for key, value in report_dict.items():
+        # Skip the title key, since we already displayed it
+        if key == "report_title":
+            continue
+
+        # 3) Check if it's code or function
+        # (You can tweak this logic if you have different rules)
+        key_lower = key.lower()
+        if "code" in key_lower or "function" in key_lower:
+            # Treat as code
+            lines.append(f"\n## {format_agent_name(key).upper()}")
+            lines.append(f"```{code_lang}\n" + str(value) + "\n```")
+        else:
+            # 4) Otherwise, just display the key-value as text
+            lines.append(f"\n## {format_agent_name(key).upper()}")
+            lines.append(str(value))
+
+    return "\n".join(lines)
+
+def remove_consecutive_duplicates(messages):
+    unique_messages = []
+    prev_message = None
+
+    for msg in messages:
+        if msg.content != prev_message:
+            unique_messages.append(msg)
+        prev_message = msg.content  # Update previous message to current
+
+    return unique_messages
+
+

ai_data_science_team/utils/plotly.py

@@ -0,0 +1,24 @@
+
+
+import json
+import plotly.io as pio
+
+def plotly_from_dict(plotly_graph_dict: dict):
+    """
+    Convert a Plotly graph dictionary to a Plotly graph object.
+    
+    Parameters:
+    -----------
+    plotly_graph_dict: dict
+        A Plotly graph dictionary.
+        
+    Returns:
+    --------
+    plotly_graph: plotly.graph_objs.graph_objs.Figure
+        A Plotly graph object.
+    """
+    
+    if plotly_from_dict is None:
+        return None
+    
+    return pio.from_json(json.dumps(plotly_graph_dict))

ai_data_science_team-0.0.0.9009.dist-info/METADATA

@@ -0,0 +1,245 @@
+Metadata-Version: 2.2
+Name: ai-data-science-team
+Version: 0.0.0.9009
+Summary: Build and run an AI-powered data science team.
+Home-page: https://github.com/business-science/ai-data-science-team
+Author: Matt Dancho
+Author-email: mdancho@business-science.io
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openpyxl
+Requires-Dist: langchain
+Requires-Dist: langchain_community
+Requires-Dist: langchain_openai
+Requires-Dist: langchain_experimental
+Requires-Dist: langgraph>=0.2.57
+Requires-Dist: openai
+Requires-Dist: pandas
+Requires-Dist: sqlalchemy
+Requires-Dist: numpy
+Requires-Dist: plotly
+Requires-Dist: streamlit
+Requires-Dist: scikit-learn
+Requires-Dist: xgboost
+Provides-Extra: machine-learning-agent
+Requires-Dist: h2o; extra == "machine-learning-agent"
+Provides-Extra: all
+Requires-Dist: h2o; extra == "all"
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+<div align="center">
+  <a href="https://github.com/business-science/ai-data-science-team">
+    <picture>
+      <img src="/img/ai_data_science_team_logo.jpg" alt="AI Data Science Team" width="400">
+  </picture>
+  </a>
+</div>
+<div align="center">
+  <em>An AI-powered data science team of agents to help you perform common data science tasks 10X faster</em>
+</div>
+<div align="center">
+  <a href="https://pypi.python.org/pypi/ai-data-science-team"><img src="https://img.shields.io/pypi/v/ai-data-science-team.svg?style=for-the-badge" alt="PyPI"></a>
+  <a href="https://github.com/business-science/ai-data-science-team"><img src="https://img.shields.io/pypi/pyversions/ai-data-science-team.svg?style=for-the-badge" alt="versions"></a>
+  <a href="https://github.com/business-science/ai-data-science-team/blob/main/LICENSE"><img src="https://img.shields.io/github/license/business-science/ai-data-science-team.svg?style=for-the-badge" alt="license"></a>
+</div>
+
+
+# Your AI Data Science Team (🪖 An Army Of Agents)
+
+**An AI-powered data science team of agents to help you perform common data science tasks 10X faster**.
+
+[**Please ⭐ us on GitHub (it takes 2 seconds and means a lot).**](https://github.com/business-science/ai-data-science-team)
+
+*Beta - This Python library is under active development. There may be breaking changes that occur until release of 0.1.0.* 
+
+---
+
+The AI Data Science Team of Copilots includes Agents that specialize data cleaning, preparation, feature engineering, modeling (machine learning), and interpretation of various business problems like:
+
+- Churn Modeling
+- Employee Attrition
+- Lead Scoring
+- Insurance Risk
+- Credit Card Risk
+- And more
+
+## Table of Contents
+
+- [Your AI Data Science Team (🪖 An Army Of Agents)](#your-ai-data-science-team--an-army-of-agents)
+  - [Table of Contents](#table-of-contents)
+  - [Companies That Want A Custom AI Data Science Team (And AI Apps)](#companies-that-want-a-custom-ai-data-science-team-and-ai-apps)
+  - [Generative AI for Data Scientists Workshop](#generative-ai-for-data-scientists-workshop)
+  - [Data Science Agents](#data-science-agents)
+    - [NEW: Multi-Agents](#new-multi-agents)
+    - [Coming Soon: Data Science Apps](#coming-soon-data-science-apps)
+    - [Agents Available Now](#agents-available-now)
+      - [Data Science Agents](#data-science-agents-1)
+      - [Multi-Agents](#multi-agents)
+    - [Agents Coming Soon](#agents-coming-soon)
+  - [Disclaimer](#disclaimer)
+  - [Installation](#installation)
+  - [Usage](#usage)
+    - [Example 1: Feature Engineering with the Feature Engineering Agent](#example-1-feature-engineering-with-the-feature-engineering-agent)
+    - [Example 2: Cleaning Data with the Data Cleaning Agent](#example-2-cleaning-data-with-the-data-cleaning-agent)
+  - [Contributing](#contributing)
+  - [License](#license)
+- [Want To Become A Full-Stack Generative AI Data Scientist?](#want-to-become-a-full-stack-generative-ai-data-scientist)
+
+## Companies That Want A Custom AI Data Science Team (And AI Apps)
+
+Want to have your own _customized_ enterprise-grade AI Data Science Team and *domain-specific* AI-powered Apps? 
+
+**Send inquiries here:** [https://www.business-science.io/contact.html](https://www.business-science.io/contact.html)
+
+## Generative AI for Data Scientists Workshop
+
+If you're an aspiring data scientist who wants to learn how to build AI Agents and AI Apps for your company that performs Data Science, Business Intelligence, Churn Modeling, Time Series Forecasting, and more, then I'd love to help you. 
+
+[**Register for my next Generative AI for Data Scientists workshop here.**](https://learn.business-science.io/ai-register)
+
+## Data Science Agents
+
+This project is a work in progress. New data science agents will be released soon.
+
+![Data Science Team](/img/ai_data_science_team.jpg)
+
+### NEW: Multi-Agents
+
+This is the internals of the SQL Data Analyst Agent that connects to SQL databases to pull data into the data science environment. It creates pipelines to automate data extraction, performs Joins, Aggregations, and other SQL Query operations. And it includes a Data Visualization Agent that creates visualizations to help you understand your data.:
+
+![Business Intelligence SQL Agent](/img/multi_agent_sql_data_visualization.jpg)
+
+### Coming Soon: Data Science Apps
+
+This is a top secret project I'm working on. It's a multi-agent data science app that performs time series forecasting.
+
+![Multi-Agent Data Science App](/img/ai_powered_apps.jpg)
+
+### Agents Available Now
+
+#### Data Science Agents
+
+1. **Data Wrangling Agent:** Merges, Joins, Preps and Wrangles data into a format that is ready for data analysis.
+2. **Data Visualization Agent:** Creates visualizations to help you understand your data. Returns JSON serializable plotly visualizations.
+3. **Data Cleaning Agent:** Performs Data Preparation steps including handling missing values, outliers, and data type conversions.
+4. **Feature Engineering Agent:** Converts the prepared data into ML-ready data. Adds features to increase predictive accuracy of ML models.
+5. **SQL Database Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations.
+
+#### Multi-Agents
+
+1. **SQL Data Analyst Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations. Includes a Data Visualization Agent that creates visualizations to help you understand your data.
+
+### Agents Coming Soon
+
+1. **Data Analyst:** Analyzes data structure, creates exploratory visualizations, and performs correlation analysis to identify relationships.
+2. **Machine Learning Agent:** Builds and logs the machine learning models.
+3. **Interpretability Agent:** Performs Interpretable ML to explain why the model returned predictions including which features were the most important to the model.
+4. **Supervisor:** Forms task list. Moderates sub-agents. Returns completed assignment. 
+
+## Disclaimer
+
+**This project is for educational purposes only.**
+
+- It is not intended to replace your company's data science team
+- No warranties or guarantees provided
+- Creator assumes no liability for financial loss
+- Consult an experienced Generative AI Data Scientist for building your own custom AI Data Science Team
+- If you want a custom enterprise-grade AI Data Science Team, [send inquiries here](https://www.business-science.io/contact.html). 
+
+By using this software, you agree to use it solely for learning purposes.
+
+## Installation
+
+``` bash
+pip install git+https://github.com/business-science/ai-data-science-team.git --upgrade
+```
+
+## Usage
+
+[See all examples here.](/examples)
+
+### Example 1: Feature Engineering with the Feature Engineering Agent
+
+[See the full example here.](/examples/feature_engineering_agent.ipynb)
+
+``` python
+feature_engineering_agent = FeatureEngineeringAgent(model = llm)
+
+feature_engineering_agent.invoke_agent(
+    data_raw = df,
+    user_instructions = "Make sure to scale and center numeric features",
+    target_variable = "Churn",
+    max_retries = 3,
+)
+```
+
+``` bash
+---FEATURE ENGINEERING AGENT----
+    * CREATE FEATURE ENGINEER CODE
+    * EXECUTING AGENT CODE
+    * EXPLAIN AGENT CODE
+```
+
+``` python
+feature_engineering_agent.get_data_engineered()
+```
+
+### Example 2: Cleaning Data with the Data Cleaning Agent
+
+[See the full example here.](/examples/data_cleaning_agent.ipynb) 
+
+``` python
+data_cleaning_agent = DataCleaningAgent(model = llm)
+
+response = data_cleaning_agent.invoke_agent(
+    data_raw = df,
+    user_instructions = "Don't remove outliers when cleaning the data.",
+    max_retries = 3,
+)
+```
+
+``` bash
+---DATA CLEANING AGENT----
+    * CREATE DATA CLEANER CODE
+    * EXECUTING AGENT CODE
+    * EXPLAIN AGENT CODE
+```
+
+``` python
+data_cleaning_agent.get_data_cleaned()
+```
+
+## Contributing
+
+1. Fork the repository
+2. Create a feature branch
+3. Commit your changes
+4. Push to the branch
+5. Create a Pull Request
+
+## License
+
+This project is licensed under the MIT License. See LICENSE file for details. 
+
+# Want To Become A Full-Stack Generative AI Data Scientist?
+
+![Generative AI Data Scientist](/img/become_a_generative_ai_data_scientist.jpg)
+
+I teach Generative AI Data Science to help you build AI-powered data science apps. [**Register for my next Generative AI for Data Scientists workshop here.**](https://learn.business-science.io/ai-register)
+
+

ai_data_science_team-0.0.0.9009.dist-info/RECORD

@@ -0,0 +1,28 @@
+ai_data_science_team/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_data_science_team/_version.py,sha256=IEp7uHd_8RlLYvLFCpp_wJCutdvJI7cJ73IN0GzK3ts,26
+ai_data_science_team/orchestration.py,sha256=xiIFOsrLwPdkSmtme7wNCCGv8XopnMTNElNzlZokL-4,303
+ai_data_science_team/agents/__init__.py,sha256=KSwxfciazWyaDG-xM93SadiIyT6X4d3uJLTdvHvVKq0,553
+ai_data_science_team/agents/data_cleaning_agent.py,sha256=OWJ3tEA5cy2fo92bTmKS8CDA48ZRRqmWg2kH7cacjDM,27337
+ai_data_science_team/agents/data_visualization_agent.py,sha256=pm7yln3GI91mOAjwDveenWwYXtJqh990oFvsoFhX3aA,28864
+ai_data_science_team/agents/data_wrangling_agent.py,sha256=UUoejYBmVFdM4At_CKQjYUyFHkaloowdd6yAElfeV9Q,32332
+ai_data_science_team/agents/feature_engineering_agent.py,sha256=bngc0COOYa8AolJwQrNuO1aDRgwBCp6LCN9_otIscWk,31011
+ai_data_science_team/agents/sql_database_agent.py,sha256=M_7IBOu7ISZZEtDAC9KGQIE7FPaXSyQ5IdD8vu91_DM,31164
+ai_data_science_team/ml_agents/__init__.py,sha256=fA5uX6dSVMAf2ApmBJXEArbnKNmsmuE0nbBsCeNAksk,86
+ai_data_science_team/ml_agents/h2o_ml_agent.py,sha256=h6kz8ZPw7ApCdfrKBqggHfBnwBJ3kDSzLMwxMxz_2tM,55181
+ai_data_science_team/multiagents/__init__.py,sha256=aI4GztEwmkexZKT5XHcH3cAjO-xYUhncb3yfPJQDqTA,99
+ai_data_science_team/multiagents/sql_data_analyst.py,sha256=2gETU9O5t9R5Ut1kEW1T3H-6Sh8xDzDfQmFV3i5lMKs,14233
+ai_data_science_team/multiagents/supervised_data_analyst.py,sha256=uduCYpicga-UCf9nPQktQggW96-HDlqvioYmEdWejtI,158
+ai_data_science_team/templates/__init__.py,sha256=_IcyFUu_mM8dFtttz95h0csJZ-XWDP3cEFuf22-R5RM,330
+ai_data_science_team/templates/agent_templates.py,sha256=pphuitXUVv21ljr_H-aof6Xq78KvDY0adF3K6lXGEz4,29107
+ai_data_science_team/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_data_science_team/tools/logging.py,sha256=7wFOv6GGhXR_RPbh-8p0GyrS608XOnZtiaGK2IbDl_s,2081
+ai_data_science_team/tools/metadata.py,sha256=3lPxLEUr3I9AF6wIKx5en-GV6JVkpUHDSLQxKj1N5Gs,9313
+ai_data_science_team/tools/parsers.py,sha256=BAi-fJT7BBt9nRS3w5n9LDTsu7JAJsH8CAI9-Qf7jCs,2086
+ai_data_science_team/tools/regex.py,sha256=lwarbLqTA2VfNQSyqKCl-PBlH_0WH3zXZvYGBYGUiu4,5144
+ai_data_science_team/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+ai_data_science_team/utils/plotly.py,sha256=nST-NG0oizKVHhH6HsjHUpTUumq9bCccBdxjuaJWnVQ,504
+ai_data_science_team-0.0.0.9009.dist-info/LICENSE,sha256=Xif0IRLdd2HGLATxV2EVp91aSY6KOuacRr_6BorKGzA,1084
+ai_data_science_team-0.0.0.9009.dist-info/METADATA,sha256=iMIyksmkPmuj9TI_oIa1lXGJYKVspxhWvflBJ1zlW0o,9875
+ai_data_science_team-0.0.0.9009.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+ai_data_science_team-0.0.0.9009.dist-info/top_level.txt,sha256=CnoMgOphCoAdGTLueWdCVByVyjwOubaGiTB1lchdy4M,21
+ai_data_science_team-0.0.0.9009.dist-info/RECORD,,

ai_data_science_team-0.0.0.9007.dist-info/METADATA

@@ -1,183 +0,0 @@
-Metadata-Version: 2.2
-Name: ai-data-science-team
-Version: 0.0.0.9007
-Summary: Build and run an AI-powered data science team.
-Home-page: https://github.com/business-science/ai-data-science-team
-Author: Matt Dancho
-Author-email: mdancho@business-science.io
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: openpyxl
-Requires-Dist: langchain
-Requires-Dist: langchain_community
-Requires-Dist: langchain_openai
-Requires-Dist: langchain_experimental
-Requires-Dist: langgraph>=0.2.57
-Requires-Dist: openai
-Requires-Dist: pandas
-Requires-Dist: numpy
-Requires-Dist: plotly
-Requires-Dist: streamlit
-Requires-Dist: scikit-learn
-Requires-Dist: xgboost
-Dynamic: author
-Dynamic: author-email
-Dynamic: description
-Dynamic: description-content-type
-Dynamic: home-page
-Dynamic: requires-dist
-Dynamic: requires-python
-Dynamic: summary
-
-# Your AI Data Science Team (An Army Of Copilots)
-
-**An AI-powered data science team of copilots that uses agents to help you perform common data science tasks 10X faster**.
-
-**Star ⭐ This GitHub (Takes 2 seconds and means a lot).**
-
-*Beta - This Python library is under active development. There may be breaking changes that occur until release of 0.1.0.* 
-
----
-
-The AI Data Science Team of Copilots includes Agents that specialize data cleaning, preparation, feature engineering, modeling (machine learning), and interpretation of various business problems like:
-
-- Churn Modeling
-- Employee Attrition
-- Lead Scoring
-- Insurance Risk
-- Credit Card Risk
-- And more
-
-## Table of Contents
-
-- [Your AI Data Science Team (An Army Of Copilots)](#your-ai-data-science-team-an-army-of-copilots)
-  - [Table of Contents](#table-of-contents)
-  - [Companies That Want An AI Data Science Team Copilot](#companies-that-want-an-ai-data-science-team-copilot)
-  - [Free Generative AI For Data Scientists Workshop](#free-generative-ai-for-data-scientists-workshop)
-  - [Data Science Agents](#data-science-agents)
-    - [Coming Soon: Multi-Agents](#coming-soon-multi-agents)
-    - [Agents Available Now](#agents-available-now)
-    - [Agents Coming Soon](#agents-coming-soon)
-  - [Disclaimer](#disclaimer)
-  - [Installation](#installation)
-  - [Usage](#usage)
-    - [Example 1: Feature Engineering with the Feature Engineering Agent](#example-1-feature-engineering-with-the-feature-engineering-agent)
-    - [Example 2: Cleaning Data with the Data Cleaning Agent](#example-2-cleaning-data-with-the-data-cleaning-agent)
-  - [Contributing](#contributing)
-  - [License](#license)
-
-## Companies That Want An AI Data Science Team Copilot
-
-If you are interested in having your own custom enteprise-grade AI Data Science Team Copilot, send inquiries here: [https://www.business-science.io/contact.html](https://www.business-science.io/contact.html)
-
-## Free Generative AI For Data Scientists Workshop
-
-If you want to learn how to build AI Agents for your company that performs Data Science, Business Intelligence, Churn Modeling, Time Series Forecasting, and more, [register for my next Generative AI for Data Scientists workshop here.](https://learn.business-science.io/ai-register)
-
-## Data Science Agents
-
-This project is a work in progress. New data science agents will be released soon.
-
-![Data Science Team](/img/ai_data_science_team.jpg)
-
-### Coming Soon: Multi-Agents
-
-This is the internals of the Business Intelligence SQL Agent I'm working on:
-
-![Business Intelligence SQL Agent](/img/multi_agent_sql_data_visualization.jpg)
-
-### Agents Available Now
-
-1. **Data Wrangling Agent:** Merges, Joins, Preps and Wrangles data into a format that is ready for data analysis.
-2. **Data Visualization Agent:** Creates visualizations to help you understand your data. Returns JSON serializable plotly visualizations.
-3. **Data Cleaning Agent:** Performs Data Preparation steps including handling missing values, outliers, and data type conversions.
-4. **Feature Engineering Agent:** Converts the prepared data into ML-ready data. Adds features to increase predictive accuracy of ML models.
-5. **SQL Database Agent:** Connects to SQL databases to pull data into the data science environment. Creates pipelines to automate data extraction. Performs Joins, Aggregations, and other SQL Query operations.
-
-### Agents Coming Soon
-
-1. **Data Analyst:** Analyzes data structure, creates exploratory visualizations, and performs correlation analysis to identify relationships.
-2. **Machine Learning Agent:** Builds and logs the machine learning models.
-3. **Interpretability Agent:** Performs Interpretable ML to explain why the model returned predictions including which features were the most important to the model.
-4. **Supervisor:** Forms task list. Moderates sub-agents. Returns completed assignment. 
-
-## Disclaimer
-
-**This project is for educational purposes only.**
-
-- It is not intended to replace your company's data science team
-- No warranties or guarantees provided
-- Creator assumes no liability for financial loss
-- Consult an experienced Generative AI Data Scientist for building your own custom AI Data Science Team
-- If you want a custom enterprise-grade AI Data Science Team, [send inquiries here](https://www.business-science.io/contact.html). 
-
-By using this software, you agree to use it solely for learning purposes.
-
-## Installation
-
-``` bash
-pip install git+https://github.com/business-science/ai-data-science-team.git --upgrade
-```
-
-## Usage
-
-[See all examples here.](/examples)
-
-### Example 1: Feature Engineering with the Feature Engineering Agent
-
-[See the full example here.](/examples/feature_engineering_agent.ipynb)
-
-``` python
-feature_engineering_agent = make_feature_engineering_agent(model = llm)
-
-response = feature_engineering_agent.invoke({
-    "user_instructions": "Make sure to scale and center numeric features",
-    "target_variable": "Churn",
-    "data_raw": df.to_dict(),
-    "max_retries":3, 
-    "retry_count":0
-})
-```
-
-``` bash
----FEATURE ENGINEERING AGENT----
-    * CREATE FEATURE ENGINEER CODE
-    * EXECUTING AGENT CODE
-    * EXPLAIN AGENT CODE
-```
-
-### Example 2: Cleaning Data with the Data Cleaning Agent
-
-[See the full example here.](/examples/data_cleaning_agent.ipynb) 
-
-``` python
-data_cleaning_agent = make_data_cleaning_agent(model = llm)
-
-response = data_cleaning_agent.invoke({
-    "user_instructions": "Don't remove outliers when cleaning the data.",
-    "data_raw": df.to_dict(),
-    "max_retries":3, 
-    "retry_count":0
-})
-```
-
-``` bash
----DATA CLEANING AGENT----
-    * CREATE DATA CLEANER CODE
-    * EXECUTING AGENT CODE
-    * EXPLAIN AGENT CODE
-```
-
-## Contributing
-
-1. Fork the repository
-2. Create a feature branch
-3. Commit your changes
-4. Push to the branch
-5. Create a Pull Request
-
-## License
-
-This project is licensed under the MIT License. See LICENSE file for details. 
-

ai_data_science_team-0.0.0.9007.dist-info/RECORD

@@ -1,21 +0,0 @@
-ai_data_science_team/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_data_science_team/_version.py,sha256=VJYpfOaKsXjGzPOsT6kYyVW6T9bFBqxt6Ph3qF8t-A8,26
-ai_data_science_team/orchestration.py,sha256=xiIFOsrLwPdkSmtme7wNCCGv8XopnMTNElNzlZokL-4,303
-ai_data_science_team/agents/__init__.py,sha256=rcF18rBsOuPJqJKvoffh6lwr4Nwm24MErM2u4H4Th9s,467
-ai_data_science_team/agents/data_cleaning_agent.py,sha256=gixYY4wGehKK_ROgU7CVOzijghmVQGD4hyK9uKhc8Hw,20890
-ai_data_science_team/agents/data_visualization_agent.py,sha256=wePFZbdB4kBah8m_iy6f4IDyjl6L6zBWzIgigJEXdk8,12933
-ai_data_science_team/agents/data_wrangling_agent.py,sha256=5w1kytoWLE4p3hj0YHVuXcgCd304eNQac-Zrrgmnr2s,16735
-ai_data_science_team/agents/feature_engineering_agent.py,sha256=UaaU3VkPhjOV0NbrYXedRb6eHOcOWWiGYhB_srrYWvg,17571
-ai_data_science_team/agents/sql_database_agent.py,sha256=mRbEAPHP6NlwQac2_VL9RuyIfCCtrmXTrzu5RLzOoeU,16031
-ai_data_science_team/templates/__init__.py,sha256=bNrKGmWXQG7GRczln_zVfUQLzxzp7hSwlLyNtLxleu4,278
-ai_data_science_team/templates/agent_templates.py,sha256=xohVgEfxPcVukPLpPfV7mZ0cpFgp-oJVLZRWCv2V-WU,19948
-ai_data_science_team/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_data_science_team/tools/logging.py,sha256=7wFOv6GGhXR_RPbh-8p0GyrS608XOnZtiaGK2IbDl_s,2081
-ai_data_science_team/tools/metadata.py,sha256=tbnca_tDp67oBA6qD29AKVooJG10VqGr4vwzj4rPUas,8348
-ai_data_science_team/tools/parsers.py,sha256=BAi-fJT7BBt9nRS3w5n9LDTsu7JAJsH8CAI9-Qf7jCs,2086
-ai_data_science_team/tools/regex.py,sha256=vkfdvi9pDe582p-fh_7cB07Wb0dOR2CsiVq-wUO3mas,2491
-ai_data_science_team-0.0.0.9007.dist-info/LICENSE,sha256=Xif0IRLdd2HGLATxV2EVp91aSY6KOuacRr_6BorKGzA,1084
-ai_data_science_team-0.0.0.9007.dist-info/METADATA,sha256=KcMFR2V9_wbepdKsrlFdfc7UB7t-Hf7i75x67LPXw3Q,6783
-ai_data_science_team-0.0.0.9007.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-ai_data_science_team-0.0.0.9007.dist-info/top_level.txt,sha256=CnoMgOphCoAdGTLueWdCVByVyjwOubaGiTB1lchdy4M,21
-ai_data_science_team-0.0.0.9007.dist-info/RECORD,,