Skip to content

feature: end-to-end agent workflow #120

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
vaibs-d merged 53 commits into from
May 24, 2025
Merged

feature: end-to-end agent workflow #120

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
53 commits
Select commit Hold shift + click to select a range
2bf027f
fix: broken mlflow unit test
marcellodebernardi May 19, 2025
264a43c
refactor: move mls agent to plexe/agents
marcellodebernardi May 19, 2025
1c13c39
refactor: remove 'constraints' as we don't use this
marcellodebernardi May 20, 2025
1ce12af
refactor: remove plexe exceptions as we don't use them
marcellodebernardi May 20, 2025
d4a1bc9
fix: remove references to constraints
marcellodebernardi May 20, 2025
38dc89a
refactor: move cot callback to internal
marcellodebernardi May 20, 2025
91c380a
refactor: move agent tools to plexe/tools package
marcellodebernardi May 20, 2025
f6bbf14
docs: update docs to reflect refactoring
marcellodebernardi May 20, 2025
0900ba2
feat: add checkpointing and resumption
marcellodebernardi May 20, 2025
880d850
feat: mlflow logging with parent runs for resumption
marcellodebernardi May 21, 2025
076c361
fix: checkpointing disabled by default
marcellodebernardi May 21, 2025
4845466
fix: ensure mlflow exp is active
marcellodebernardi May 21, 2025
40076cf
refactor: move mlops_engineer to plexe/agents
marcellodebernardi May 21, 2025
10ae3b7
refactor: put all agents under orchestrator agent
marcellodebernardi May 21, 2025
aa9045f
fix: remove schema and eda from pre-build steps
marcellodebernardi May 21, 2025
1e91a58
docs: update multi-agent-system.md
marcellodebernardi May 21, 2025
c72e133
fix: remove unnecessary todos from models.py
marcellodebernardi May 21, 2025
e2ab836
fix: input schema not extracted after run
marcellodebernardi May 21, 2025
909a699
fix: dangling instruction in schema_resolver_prompt_templates.yaml
marcellodebernardi May 21, 2025
6646b4a
fix: eda agent missing from orchestrator
marcellodebernardi May 21, 2025
adb767e
fix: mlflow callback labels every build failed
marcellodebernardi May 21, 2025
4f5ff6d
fix: dataset analyser has no prompt template
marcellodebernardi May 21, 2025
b93c04f
fix: remove custom run names from mlflow child runs
marcellodebernardi May 21, 2025
5d8e3fc
fix: mlflow logging failing because of safe_get
marcellodebernardi May 21, 2025
6a26a78
chore: remove unnecessary comments
marcellodebernardi May 21, 2025
b61f98b
fix: prompt templates to improve agent behaviour
marcellodebernardi May 22, 2025
36fbb6a
refactor: move object registry to plexe/core
marcellodebernardi May 22, 2025
4e60626
feat: add resumption to model build
marcellodebernardi May 22, 2025
5c3d6b4
feat: add feature engineering agent
marcellodebernardi May 22, 2025
1db7733
feat: introduce immutability for object registry
marcellodebernardi May 22, 2025
87d7a0b
fix: tell eda agent to stop creating new datasets
marcellodebernardi May 22, 2025
c6bdfa6
fix: allow required imports for feature engineer
marcellodebernardi May 22, 2025
9c77763
fix: clarify dataset analyser task
marcellodebernardi May 22, 2025
f1ab294
fix: clarify feature engineer task
marcellodebernardi May 22, 2025
55b3a64
fix: incorrect registry import in dataset splitter
marcellodebernardi May 22, 2025
a9afc69
fix: miscellaneous fixes to dataset splitting and schema resolution
marcellodebernardi May 22, 2025
7e95184
fix: miscellaneous fixes to feature engineer
marcellodebernardi May 22, 2025
c8d571d
feat: save dataset splitting code to bundle
marcellodebernardi May 22, 2025
dc1b727
fix: eda report retrieving not always working
marcellodebernardi May 23, 2025
a1e48c8
fix: ensure manager doesn't try to do work itself
marcellodebernardi May 23, 2025
6cd94f3
fix: dataset splitting code registered incorrectly
marcellodebernardi May 23, 2025
569552a
fix: switch feature engineer to use ml engineer model
marcellodebernardi May 23, 2025
9188b8a
fix: add statsmodels to data analyser and feature engineer
marcellodebernardi May 23, 2025
d898c98
chore: update packages to upgrade smolagents
marcellodebernardi May 23, 2025
01d750e
chore: bump to 0.22.0
marcellodebernardi May 23, 2025
92432dc
feat: add model testing agent
marcellodebernardi May 23, 2025
fead55a
feat: add model testing agent
marcellodebernardi May 23, 2025
fc48452
refactor: switch to yaml for model archive
marcellodebernardi May 24, 2025
b979075
fix: clean up example scripts
marcellodebernardi May 24, 2025
dce15a8
Merge remote-tracking branch 'origin/feature/iterative-model-building…
marcellodebernardi May 24, 2025
55d351f
fix: disable resumption for now
marcellodebernardi May 24, 2025
eada7f0
fix: remove unnecessary unit tests
marcellodebernardi May 24, 2025
2a66001
fix: remove unnecessary unit tests
marcellodebernardi May 24, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion CLAUDE.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ The core architecture is as follows: agents go in `plexe/agents/*`, tools in `pl
- `plexe/models.py`: Core `Model` class with build/predict functionality
- `plexe/agents/schema_resolver.py`: Agent inferring input/output schemas
- `plexe/internal/agents.py`: Multi-agent system implementation (`PlexeAgent` class)
- `plexe/internal/models/tools/`: Tools for code generation, execution, validation
- `plexe/tools/`: Tools for code generation, execution, validation
- `plexe/config.py`: Configuration management and prompt templates
- `plexe/internal/common/registries/objects.py`: Shared object registry for agents
- `plexe/datasets.py`: Dataset handling and synthetic data generation
Expand Down
Loading