Devang Savita Ram Mohan

At Scale, our Enterprise research team focuses on testing novel approaches to real-life scenarios, grounded in real customer data, so that we can further capability advancements that are actually applicable and useful in scaled, enterprise contexts. 
We take our findings from experiments like the one we share in today&rsquo;s blog and codify them into our MLE tool suites so they can immediately put the results into practice and meaningfully improve delivery for our enterprise clients. In the case of the following experiment, we developed a PlanByPythonExecutor tool that is already being used in production.&nbsp;
<h2 dir="ltr">Intro</h2>
Imagine you&rsquo;re an edtech, looking to train your model on a set of question and answer pairs from previous students across multiple domains like Math, English, and History. The issue? You&rsquo;re parsing Q&amp;A data from all kinds of legacy files: websites, textbooks, and more, leading to unstructured HTML, XML, and free text jargon. Your model can&rsquo;t parse the Q&amp;A pairs. It needs a cleaned version of this dataset.&nbsp;
Converting content and datasets into human-legible or vice versa, to LLM-legible data, is an open and interesting topic with a lot of potential for discovery. In fact, Andrej Karpathy discussed the challenges of the "LLMification" of human-readable education content&nbsp;<a href="https://x.com/karpathy/status/1961128638725923119" target="_blank" rel="noopener">on X recently.</a>
<img src="https://img.plasmic.app/img-optimizer/v1/img?src=b483bae2f9d7f182dca89fb2136f779d.png&amp;f=webp&amp;q=75" alt="" width="1200" height="675">A sample snippet of your messy, unparsed Q&amp;A data set
So what do you do? You&rsquo;re most likely shouting: use an LLM to clean up the file! There are two options to doing this:
<ol>
<li dir="ltr" aria-level="1">
Ask an LLM to read each question-answer pair and rewrite them in a clean way
</li>
<li dir="ltr" aria-level="1">
Ask an LLM to write a script to parse all the question-answer pairs at once using a standardized library like beautifulsoup
</li>
</ol>
On option 1, we found that LLMs are prone to getting distracted by large amounts of context, missing key details, failing to persist tables, misinterpreting tags, and other issues.&nbsp;&nbsp;
In this blog, we mostly explore option 2 which introduces its own set of challenges such as needing to learn how to handle edge cases, writing the proper formatting code, allowing the user to equip the LLM with helpful utilities and more. We&rsquo;ll go into detail about a method we developed to use LLMs to call multiple tools to solve this problem.&nbsp;
The key insight is we didn't need to rely on the LLM to actually do the cleaning of the data, we just needed it to come up with a plan on how to clean the data. The plan can then be executed, and the feedback circulated back to the LLM in the form of formatted examples, failures etc so that the LLM can refine the plan.
We found the standard approach to toolchaining insufficient. Simply giving the LLM access to multiple tools and asking it to clean the data led to extremely poor results, if it could do it at all. Instead, when we gave the LLM access to a python sandbox pre-loaded with these tools and asked it to develop the aforementioned plan, the output was significantly improved. For the remainder of this blog, we dive into why this happened, how we set up our experiment, and what the findings mean for you.&nbsp;
<h2 dir="ltr">So what&rsquo;s the problem with the standard approach to toolchaining?</h2>
You might be thinking &ldquo;Surely this data formatting task is quite simple for an LLM?&rdquo;. We actually found that it was anything but!&nbsp;
A reasonable first attempt to solve this task would be to provide the LLM with a set of tools (eg: a bunch of formatters, utilities to read the data, etc.) and let it figure out how it wants to use them. However, we found this approach breaks down for anything more interesting than a toy dataset. The reason for this is the following: in this scenario, the LLM is responsible for the orchestration of the tools. As a result, all tool responses are routed back to the LLM, which then has to provide the output of this tool call as input to the next tool call.&nbsp;
While flexible, this method often becomes costly and error-prone. As outputs move from tool to tool, the LLM can unintentionally rewrite inputs, such as injecting random strings, altering formats, or losing critical information. In turn, this creates downstream errors, bloats the context window, and is something the LLM simply does not need to do.
<img src="https://img.plasmic.app/img-optimizer/v1/img?src=3e767be1ebf3f50c4d9b182acf10dd78.png&amp;f=webp&amp;q=75" alt="" width="1200" height="675">
Consider the following example: let's say the output of a tool call is a complex, nested, 20000-character HTML string. Asking the LLM to reproduce it exactly as input for the formatter is fraught with danger - even a single missing character/tag would break the formatter downstream! Interestingly, in their new gpt-oss release, OpenAI tacitly acknowledges this challenge of tool-call chaining. To train this suite of models, they use a <a href="https://cookbook.openai.com/articles/openai-harmony#message-format">Harmony Response format</a> that provides a routing mechanism which will allow tool call outputs to be redirected to receivers other than the agent that called the tool, although they don&rsquo;t show any examples of doing this yet. This futureproofs this message format to scenarios where one needs to compose tool calls (tool_call_A output -&gt; tool_call_B input) or have a separate agent understand the tool call output (tool_call_A output triggered by Agent X -&gt; message for Agent Y)
&nbsp;
<h2 dir="ltr">To address this problem, we asked the LLM to formulate a plan and enable toolchaining</h2>
Motivated by how we as humans approach large, daunting tasks, we instead asked the LLM to formulate a plan. A Plan Executor Tool then carries out this plan and provides only relevant, curated context as feedback. In this way, the LLM is still able to refine its plan based on how well it performed, but is no longer required to orchestrate the toolchaining, and nor is it throttled by overwhelming context.&nbsp;
We found that LLMs are exceptionally strong at generating these plans and can reason through tool-based workflows, but their performance depends heavily on how tools are exposed and how plans are structured. We tested three distinct approaches to generating these plans: Plan as JSON with toolchaining, Plan as Python with tool injection, and Plan as Python without tool injection, aka a stateful code executor. We tested against a baseline of an MLE writing custom parsing functions to clean the provided data.
By enabling the LLM to generate a Python-based plan and leverage pre-built tools from our subject-matter experts and ML engineers, we achieved results nearly as accurate as human-only workflows at a fraction of the cost and time. Next, we dive into the details for the approach to the plans.&nbsp;
<h2 dir="ltr">We tested three distinct options for plans&nbsp;</h2>
Having established the need for a plan, the question then becomes what format this plan should take. We explored three options: Plan as Python with no tool injection, Plan as JSON, and Plan as Python with tool injection. We tested these three options against a baseline, handwritten set of custom parsing scripts.
<img src="https://img.plasmic.app/img-optimizer/v1/img?src=1b0a312565f134539fa85cf60e56a1e6.png&amp;f=webp&amp;q=75" alt="" width="1200" height="675">
<h3 dir="ltr">Baseline: No LLMs, Handwritten, Custom Python Parsing Functions</h3>
To test the plan&rsquo;s effectiveness, we first set a baseline with no LLM intervention. An engineer wrote custom parsing functions to clean the data. Unsurprisingly, this process proved highly effective but took a full week to develop and test. The functions were also bespoke to this dataset, making them tightly coupled to the provided data.
<h3 dir="ltr">Plan as JSON with Tool Injection</h3>
The Plan as JSON consists of a name, a description and a list of steps that the executor must carry out. This might look like:
<pre class="language-markup"><code>{
"name": "LLMPlan",
"description": "This plan is designed for a data formatting task",
"steps": [
{
"name": "load_csv",
"action": "load_csv",
"params": {"path": "input_data.csv"}
}
...
{
"name": "run_python_code",
"action": "run_python",
"params": {"code": "&lt;LLM generated code&gt;"}
}
...
{
"name": "save_formatted_csv",
"action": "save_csv",
"params": {"path": "output_data.csv", "df": "$stepk.result"}
}
]
}
</code></pre>
The action in each step could be a tool that the LLM has access to, or python code that the LLM writes on its own. The crux is that, via this json, the LLM is effectively able to compose a sequence of tool calls by referencing results of previous steps without ever receiving the output explicitly.
We found that the LLM was able to accurately reason about the control flow (e.g., iterating over data, conditionally selecting formatters) and revise plans after failed executions. Overall the method proved workable when the LLM had access to a basic Python executor capable of invoking formatter tools. When analysing the traces of the model though, we noticed that most of the heavy lifting was happening within the python tool. This prompted the question: are we throttling the LLM with enforcing such a rigid JSON plan structure? What if, instead, we allowed it to write an arbitrary plan as python code?
<h3 dir="ltr">Plan as Python with Tool Injection</h3>
Next, we allowed the LLM to break free from its Plan as JSON shackles and write arbitrary python code. Where this differed from a vanilla code-interpreter tool was that we injected all the tools that the LLM would otherwise have at its disposal into its runtime. So instead of the classical tool-calling paradigm, we allow the LLM to focus on writing good code, while providing it with the inductive bias of certain utilities it can leverage (e.g. HTML formatters written by one of our MLEs). This seemed an especially natural segue given how models are getting more specialised and adept at writing code.
This approach still provided us the ability to not task the LLM to do things it isn't needed for, but at the same time give it flexibility to structure the plan in whatever shape it wants! With minimal human input, the LLM was able to leverage the provided utilities to format the data to a level comparable to that implemented by our engineers, often revising scripts and fixing bugs as it understood the data better.&nbsp;
<h3 dir="ltr">Plan as Python (aka Stateful Code Interpreter)</h3>
In this experiment, we give the LLM access to a single tool: a stateful code interpreter and ask it to carry out the task. We wanted to see if the utilities were indeed useful or if the LLM does best with a clean slate.
While the LLM was able to implement reasonable formatters, the quality of output in most cases was mostly inferior to that of the other methods that leveraged the utilities. Interestingly, however, the LLM ended up implementing a completely new formatter for a question type we had not previously identified and significantly improved the quality of the question-answer parsing as a result.&nbsp;
&nbsp;
<h2 dir="ltr">Here&rsquo;s what we found</h2>
In this experiment, we used LLM-as-a-judge to evaluate two things:
<ol>
<li dir="ltr" aria-level="1">
Completeness: Whether the LLM was able to extract the full question-answer pair (e.g.all conditions qualifiers should be present in the question)
</li>
<li dir="ltr" aria-level="1">
Formatting: Were all parts of the question + answer formatted correctly, accounting for different format types like tables etc
</li>
</ol>
&nbsp;
Overall, the baseline approach with handwritten parsing functions produced the best results. This is unsurprising given the logic was specifically developed for and tested on this particular dataset. However, Plan as Python with Tool Injection and Plan as JSON were both able to effectively leverage these handwritten tools to yield results that were almost as good. While the two plans did not differ meaningfully in performance compared to one another, Plan as Python provides more flexibility.
&nbsp;
The Plan as Python (aka Code Interpreter Tool) which didn&rsquo;t have access to our handwritten parsing functions overall performed slightly worse in our LLM-as-a-judge evaluations. Interestingly however, the LLM ended up implementing a completely new formatter for a question type we had not previously identified and significantly improved the quality of the question-answer parsing as a result. Thus, we show that overall the Code Interpreter Tool was inferior in our internal LLM-as-a-judge evaluations but for certain question types, it significantly outperformed all other approaches.
Here&rsquo;s a look at the results below:
<img src="https://img.plasmic.app/img-optimizer/v1/img?src=2a80d6733932e2e34807ac38cb75993c.png&amp;f=webp&amp;q=75" alt="" width="1200" height="675">
<img src="https://img.plasmic.app/img-optimizer/v1/img?src=043ea3ca6db6efaedf612b736e02e3b9.png&amp;f=webp&amp;q=75" alt="" width="1200" height="675">
To recap, we found that injecting MCP servers and tools into a python session that the LLM can write code in is very powerful. In general, we have shown the ability to get LLMs help us do this data formatting task and in a limited capacity, LLMs have found things that were better than a human.
&nbsp;
<h2 dir="ltr">What does this mean for you?</h2>
If you&rsquo;re tackling messy data formatting at large volumes and want to use LLMs, here&rsquo;s the takeaway: planning-based with toolchaining approaches work really well. They let an LLM break down a complex job into smaller steps and use the right tools along the way without asking the LLM to do things it isn&rsquo;t suited for or in fact required to do.&nbsp;
&nbsp;
<ul>
<li dir="ltr" aria-level="1">
Plan as Python is flexible and works well for open-ended tasks. You can give it your own helper tools or logic, and it will figure out how to use them.
</li>
<li dir="ltr" aria-level="1">
Plan as JSON is more rigid but easier to read and debug. It can still tap into the same execution engine as Plan as Python when you need more flexibility.
</li>
</ul>
&nbsp;
Going forward, we&rsquo;re looking at adding a feedback loop: a simple Python executor could write small helper functions for tricky question types, and we&rsquo;d feed those back into Plan as Python with Tool Injection. That way, we can get close to the accuracy of handwritten parsing while keeping the workflow general and reusable.
&nbsp;
<h2 dir="ltr">What&rsquo;s next</h2>
Thanks to this work, we now have access to our Plan as Python tool to bring to our various engagements allowing LLM Agents to freely use premade enterprise SME or Scale MLE defined tools. As a result, we&rsquo;ve experienced a significant reduction in the cost of cleaning up messy data (something we know we&rsquo;ll always need when working with enterprises 😀)
We&rsquo;ve integrated these planning and tool-injection capabilities into our internal MLE toolkit via a robust Stateful Python Executor Tool. This will allow others across Scale to prototype similar workflows on live customer data with minimal setup. Future directions include extending this framework to more complex agent loops, supporting self-healing pipelines (e.g., retrying failed steps), and fine-tuning our models on the traces by the LLM while generating these plans to further refine their ability to carry out tool-enabled loops.
Every day it seems like there&rsquo;s some new &ldquo;breakthrough&rdquo; with AI agents and while it&rsquo;s true that the pace of new agentic capabilities is astonishing, as an enterprise, what matters is whether those breakthroughs can actually work in an enterprise context. For our Enterprise research team, that&rsquo;s our differentiator. We start with real customer use cases and test advanced approaches against real customer data. Those learnings are then codified back into our platform, so every research finding becomes part of our delivery pipeline for customers, like these results of our data formatting toolchaining experiment.&nbsp;
Cleaning data is just one small step in the journey toward adopting AI in production. At Scale, we specialize in partnering with leading enterprises to tackle hard, high-impact problems like these. If you&rsquo;re struggling with a similar issue or just find this research interesting, reach out to our team at <a href="http://scale.com/enterprise/agentic-solutions">scale.com/demo. </a>You can also learn more at <a href="http://scale.com/enterprise/agentic-solutions">scale.com/enterprise/agentic-solutions</a>.

We gave the LLM access to a python sandbox and asked it to plan as a method for toolchaining

Devang Savita Ram Mohan

Toolchaining: The Problem No One is Talking About