ai-data-science-team 0.0.0.9007__tar.gz → 0.0.0.9008__tar.gz

ai_data_science_team-0.0.0.9008/MANIFEST.in

@@ -0,0 +1 @@
+include requirements.txt

{ai_data_science_team-0.0.0.9007 → ai_data_science_team-0.0.0.9008}/PKG-INFO

@@ -1,10 +1,15 @@
 Metadata-Version: 2.2
 Name: ai-data-science-team
-Version: 0.0.0.9007
+Version: 0.0.0.9008
 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
@@ -23,6 +28,7 @@ Requires-Dist: scikit-learn
 Requires-Dist: xgboost
 Dynamic: author
 Dynamic: author-email
+Dynamic: classifier
 Dynamic: description
 Dynamic: description-content-type
 Dynamic: home-page
@@ -30,11 +36,28 @@ Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
 
-# Your AI Data Science Team (An Army Of Copilots)
+<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" 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" 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?v" alt="license"></a>
+</div>
 
-**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).**
+# 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.* 
 
@@ -51,12 +74,13 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
 
 ## Table of Contents
 
-- [Your AI Data Science Team (An Army Of Copilots)](#your-ai-data-science-team-an-army-of-copilots)
+- [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 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)
+  - [Companies That Want A Custom AI Data Science Team (And AI Apps)](#companies-that-want-a-custom-ai-data-science-team-and-ai-apps)
+  - [Free How To Build AI Agents for Data Scientists Workshop](#free-how-to-build-ai-agents-for-data-scientists-workshop)
   - [Data Science Agents](#data-science-agents)
     - [Coming Soon: Multi-Agents](#coming-soon-multi-agents)
+    - [...And after that, the Multi-Agent Data Science Apps](#and-after-that-the-multi-agent-data-science-apps)
     - [Agents Available Now](#agents-available-now)
     - [Agents Coming Soon](#agents-coming-soon)
   - [Disclaimer](#disclaimer)
@@ -66,14 +90,19 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
     - [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)
 
-## Companies That Want An AI Data Science Team Copilot
+Want to have your own _customized_ enterprise-grade AI Data Science Team and domain-specifici AI-powered Apps? 
 
-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)
+**Send inquiries here:** [https://www.business-science.io/contact.html](https://www.business-science.io/contact.html)
 
-## Free Generative AI For Data Scientists Workshop
+## Free How To Build AI Agents 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)
+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
 
@@ -87,6 +116,12 @@ 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)
 
+### ...And after that, the Multi-Agent 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
 
 1. **Data Wrangling Agent:** Merges, Joins, Preps and Wrangles data into a format that is ready for data analysis.
@@ -129,15 +164,14 @@ pip install git+https://github.com/business-science/ai-data-science-team.git --u
 [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
-})
+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
@@ -147,19 +181,22 @@ response = feature_engineering_agent.invoke({
     * 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 = 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
-})
+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
@@ -169,6 +206,10 @@ response = data_cleaning_agent.invoke({
     * EXPLAIN AGENT CODE
 ```
 
+``` python
+data_cleaning_agent.get_data_cleaned()
+```
+
 ## Contributing
 
 1. Fork the repository
@@ -181,3 +222,10 @@ response = data_cleaning_agent.invoke({
 
 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.9007 → ai_data_science_team-0.0.0.9008}/README.md

@@ -1,8 +1,25 @@
-# Your AI Data Science Team (An Army Of Copilots)
+<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" 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" 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?v" alt="license"></a>
+</div>
 
-**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).**
+# 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.* 
 
@@ -19,12 +36,13 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
 
 ## Table of Contents
 
-- [Your AI Data Science Team (An Army Of Copilots)](#your-ai-data-science-team-an-army-of-copilots)
+- [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 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)
+  - [Companies That Want A Custom AI Data Science Team (And AI Apps)](#companies-that-want-a-custom-ai-data-science-team-and-ai-apps)
+  - [Free How To Build AI Agents for Data Scientists Workshop](#free-how-to-build-ai-agents-for-data-scientists-workshop)
   - [Data Science Agents](#data-science-agents)
     - [Coming Soon: Multi-Agents](#coming-soon-multi-agents)
+    - [...And after that, the Multi-Agent Data Science Apps](#and-after-that-the-multi-agent-data-science-apps)
     - [Agents Available Now](#agents-available-now)
     - [Agents Coming Soon](#agents-coming-soon)
   - [Disclaimer](#disclaimer)
@@ -34,14 +52,19 @@ The AI Data Science Team of Copilots includes Agents that specialize data cleani
     - [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)
 
-## Companies That Want An AI Data Science Team Copilot
+Want to have your own _customized_ enterprise-grade AI Data Science Team and domain-specifici AI-powered Apps? 
 
-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)
+**Send inquiries here:** [https://www.business-science.io/contact.html](https://www.business-science.io/contact.html)
 
-## Free Generative AI For Data Scientists Workshop
+## Free How To Build AI Agents 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)
+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
 
@@ -55,6 +78,12 @@ 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)
 
+### ...And after that, the Multi-Agent 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
 
 1. **Data Wrangling Agent:** Merges, Joins, Preps and Wrangles data into a format that is ready for data analysis.
@@ -97,15 +126,14 @@ pip install git+https://github.com/business-science/ai-data-science-team.git --u
 [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
-})
+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
@@ -115,19 +143,22 @@ response = feature_engineering_agent.invoke({
     * 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 = 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
-})
+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
@@ -137,6 +168,10 @@ response = data_cleaning_agent.invoke({
     * EXPLAIN AGENT CODE
 ```
 
+``` python
+data_cleaning_agent.get_data_cleaned()
+```
+
 ## Contributing
 
 1. Fork the repository
@@ -149,3 +184,10 @@ response = data_cleaning_agent.invoke({
 
 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.9008/ai_data_science_team/_version.py

@@ -0,0 +1 @@
+__version__ = "0.0.0.9008"

{ai_data_science_team-0.0.0.9007 → ai_data_science_team-0.0.0.9008}/ai_data_science_team/agents/__init__.py

@@ -1,6 +1,6 @@
 from ai_data_science_team.agents.data_cleaning_agent import make_data_cleaning_agent, DataCleaningAgent
-from ai_data_science_team.agents.feature_engineering_agent import make_feature_engineering_agent
-from ai_data_science_team.agents.data_wrangling_agent import make_data_wrangling_agent
-from ai_data_science_team.agents.sql_database_agent import make_sql_database_agent
-from ai_data_science_team.agents.data_visualization_agent import make_data_visualization_agent
+from ai_data_science_team.agents.feature_engineering_agent import make_feature_engineering_agent, FeatureEngineeringAgent
+from ai_data_science_team.agents.data_wrangling_agent import make_data_wrangling_agent, DataWranglingAgent
+from ai_data_science_team.agents.sql_database_agent import make_sql_database_agent, SQLDatabaseAgent
+from ai_data_science_team.agents.data_visualization_agent import make_data_visualization_agent, DataVisualizationAgent