Skip to content

[MRG]add memory to each coponent#292

Merged
huangyz0918 merged 4 commits into
MLSysOps:mainfrom
embersax:componentmemory
May 16, 2025
Merged

[MRG]add memory to each coponent#292
huangyz0918 merged 4 commits into
MLSysOps:mainfrom
embersax:componentmemory

Conversation

@embersax
Copy link
Copy Markdown
Contributor

Closes #

This PR implements a component memory system that tracks execution of all major components (Advisor, Planner, Coder, Debugger, etc.) in MLE-Agent.

Changes:

  • Added ComponentMemory class for storing component traces
  • Added trace_component decorator for automatically tracking component executions
  • Added memory tracking to all agent components
  • Added CLI command to view component traces

After run the whole process of memory start, you can verify the trace has been stored by:

mle traces --component advisor

Before submitting this PR, please make sure you have:

  • confirmed all checks still pass OR confirm CI build passes.
  • verified that any code or assets from external sources are properly credited in comments and/or in
    the credit file.

@dosubot dosubot Bot added size:XL This PR changes 500-999 lines, ignoring generated files. enhancement New feature or request labels Apr 12, 2025
@embersax embersax closed this Apr 20, 2025
@embersax embersax reopened this Apr 20, 2025
@huangyz0918
Copy link
Copy Markdown
Member

Thanks for submitting this PR! Actually we have lanceDB, so it is no need to introduce SQLite, we can reuse the lanceDB ( the memory.py) to create a new table for saving the traces.

…se simple text embediing for component memory
@embersax
Copy link
Copy Markdown
Contributor Author

Thanks for submitting this PR! Actually we have lanceDB, so it is no need to introduce SQLite, we can reuse the lanceDB ( the memory.py) to create a new table for saving the traces.

Hi I have changed SQLlite to LanceDB, and for component memory I use sentence-transformers.

@huangyz0918
Copy link
Copy Markdown
Member

If you think the PR is ready, you can change the title by adding [MRG]

@embersax embersax changed the title add memory to each coponent [MRG]add memory to each coponent May 10, 2025
@embersax
Copy link
Copy Markdown
Contributor Author

Hi, thanks for the comment, just add the tag.

Copy link
Copy Markdown
Member

@huangyz0918 huangyz0918 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM! Thank you for this contribution 👍

@dosubot dosubot Bot added the lgtm This PR has been approved by a maintainer label May 16, 2025
@huangyz0918 huangyz0918 merged commit ee7836d into MLSysOps:main May 16, 2025
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request lgtm This PR has been approved by a maintainer size:XL This PR changes 500-999 lines, ignoring generated files.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants