Repository files navigation

Tree-search visualization during code generation. We visualize a reasoning algorithm which learns from feedback, automatically correcting itself by analyzing error tracebacks to refine its solutions. In this case, we benchmark Python programming problems from the HumanEval dataset.

Branches is an AI tool for graph-based prototyping of advanced algorithms for LLM reasoning and planning -- like Tree of Thoughts and Reflexion. Branches is adapted fromFlux.

Designed for researchers and developers, it allows users to directly interact with AI reasoning processes, streamlining the exploration of complex coding challenges and strategic problem-solving.

Branches automatically expands decision trees to solve programming problems from theHumanEval dataset, visualizing reasoning chains and facilitating self-correction through error tracebacks. This is found on themain branch and is currently hosted.

Branches includes a specialized evaluation mechanism for theGame of 24 puzzle, leveraging a scoring system to enhance breadth-first search (BFS) by prioritizing promising paths. This is found on thegame-of-24 branch.

• 🌳Automated Tree Expansion: Leveraging Tree of Thoughts for dynamic expansion in problem-solving.
• 🧠Pre-loaded Prompts: Curated for search-based reasoning to solve specific problems.
• 💻Code Interpretation: Instant execution and error analysis for self-correcting AI-generated code.
• 🔍Scoring Mechanism: Advanced BFS for the Game of 24 with node evaluation for search optimization.
• 📊Interactive Visualization: Graphical representation of tree searches for easy analysis and education. Largely adapted fromFlux.

To get started with Branches, you can either visitcode-gen-tree.vercel.app for the hosted version or run it locally by following the instructions below.

npm i -g vercelvercel

Our commitment to enhancing Branches continues, with exciting new developments on the way:

• More reasoning and planning algorithms beyond the defaults (#10)
• Node Value Editing and Regenerate Subtree Functionality (#5)
• UI Color Fixes and Customization Features (#6)
• Address Model/UI Timeout Issues (#7)
• Enhance Game of 24 Logic, Model Cost Tracking, and Prompt Engineering (#8)

Your contributions make Branches better. Whether it’s bug reports, new features, or feedback, we welcome it all! Report bugs or request features by creating an issuehere.

Branches is open-source and continues to uphold theMIT license.