Skip to content
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.

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

Train multi-node tasks including cycle multi-hop rag and trainable agent #306

Open
wants to merge 35 commits into
base: main
Choose a base branch
from

Conversation

liyin2015
Copy link
Member

@liyin2015 liyin2015 commented Dec 17, 2024

, trainable, added component graph with cycl…e, output graph as simplified version of the dynamic computation graph
Screenshot 2024-12-23 at 10 36 28 PM
Screenshot 2024-12-23 at 10 38 06 PM

Auto-optimize an agent with planner.

The graph is from the text grad debug mode. And when click on a node, it will show all details on the gradients.

Big updates to the codebase:

  1. component is used as a trainable component with forward and call. and gradcomponent is for units with ability to backward (use together with add_successor_map_fn). -> many previous component is no longer a component, this include Prompt, Parser, OutputParser. The visualization of these parts need to be done manually to whichever use it and does not come out of box. This is not important for now.
  2. React is fully updated to handle training.
  3. Toolmanager is a trainable component
  4. FuncTool is a trainable component.
  5. Generator. Output_processors is no longer a component but a Parser type. we no longer need of fun_to_component, but func_to_parser. This requires us to change the apis. And write a migration plan

Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@liyin2015 liyin2015 changed the title build cycle multi-hop rag… build cycle multi-hop rag and trainable agent Dec 24, 2024
@liyin2015 liyin2015 changed the title build cycle multi-hop rag and trainable agent Train multi-node tasks including cycle multi-hop rag and trainable agent Dec 26, 2024
… graidents which makes the input args way too big, and add parameter and output parameter allowed_components
…dient pass through grad component correction, the code follows the 3 cases of gradcomponent design now, still need to further optimize the agent
…ckward engine and optimizer, has made it 2% higher accuracy in step 3 than before
…nt clearly and make many previous component only a class
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant