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.

About

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 from Flux.

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.

Code Generation (HumanEval)

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

Game of 24

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

Features

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

Usage

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

Deploy to Vercel

npm i -g vercel
vercel

Stay Tuned For

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)

Contributing

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 issue here.

License

Branches is open-source and continues to uphold the MIT license.