TaskWeaver — Code-First Data Analytics Agent

TaskWeaver

TaskWeaver (by Microsoft Research) is a code-first agent framework specifically designed for data analytics tasks. Instead of string templating, it generates and executes Python code at each step, making it uniquely powerful for structured data manipulation, statistical analysis, and ML pipeline automation.

Key Features

• Code-first planning: every sub-task emits executable Python
• Stateful session with persistent memory across turns
• Plugin system: wrap any Python library as an agent tool
• Rich data type support: DataFrames, NumPy arrays, images, time series
• Human-in-the-loop option for reviewing generated code before execution
• Built-in plugin library: SQL, sklearn, pandas, matplotlib, …

Quick Start

git clone https://github.com/microsoft/TaskWeaver
pip install -e .
# edit taskweaver/config.yaml (set LLM endpoint)
python taskweaver/app/app.py

from taskweaver.app.app import TaskWeaverApp

app = TaskWeaverApp(app_dir=".", config={"llm.model": "gpt-4o"})
session = app.get_session()
response = session.chat("Load sales.csv and show me monthly revenue trends")
print(response.post_list[-1].message)

Install via ai-supply

npx ai-supply add taskweaver-code-first-data-agent

Curated mirror of the open-source TaskWeaver (MIT). Get it from the source.