Improving Agents is a Data Mining Problem — Vivek Trivedy, LangChain

AI Engineer · 20 min · 194 sentences · from YouTube's caption track

Each timecode opens YouTube at the start of that sentence. Line anchors (#s42) are the cue ids in the WebVTT, and every line carries its start and end seconds. All transcripts has every talk, and the whole corpus as one file.

00:01[music]
00:12Hey everyone.
00:13Uh I'm Vivek and I lead applied research at LangChain and I'm going to talk about something that I think is sexy, which is data mining, but it's not as sexy as LLM, so we're going to try to like make it sexy together.
00:27And the problem that we're going to talk about today is how do we continuously improve agents, but how do we do that via data?
00:35So, to start, I'm going to tell a little story that I think maybe a lot of us have felt before, like I ran my agent, it did a bunch of things,
00:46it made some mistakes.
00:49Now, I ask someone like, what do I actually do about that?
00:51Like, I have all this data, made some mistakes, what now?
00:56Basically, what we're going to do today is we're going to motivate a recipe for what we should do to continuously improve agents over time, and then I'm going to talk from some lived experience and like some stuff that we help customers do to run this over large-scale trace data.
01:11So, the first step in building a successful agent is shipping it.
01:16So, if you put it out into the real world, then it can operate in environments and then you can get feedback from what it's doing.
01:23The second step is collect a ton of traces.
01:27So, agents operate in the environment every single time they operate, they do tool calls, they have output messages, they call APIs, they use CLIs.
01:35All of that generates data and we want to store all of that so we can like do stuff with it.
01:42The next thing is the data mining in this talk, which is once we have tons of trace data, maybe gigabytes, maybe terabytes, depending on like how many agents you're shipping,
01:52we're going to do data mining over that.
01:54And I promise I will tell you exactly what data mining we're going do, uh but we're going to do some over it.
02:00And then the fun part, which is I collected that data, I read it, I curated it, and now we actually need to run the experiments in a data-driven way to see, "Hey, is this new prompt, or is this new tool, or is this new orchestration,
02:14or is this new loop, is it actually improving things based on the previous traces that I've seen?"
02:21And this is maybe a bit of a hot take, but continual learning is super hot right now.
02:27I'm talking about it, this whole room is going to hear about it for the next like five, six hours.
02:30Um but there's a very tight coupling between what observability is and what continual learning is.
02:37And the main reason for that is that agents that operate in environments, they produce trace data, and what continual learning for agents and continual learning for humans basically is is I do a bunch of stuff in the world,
02:50I think about what I did, and then I need to update my definition, like my knowledge, stuff I write down, in order to respond to the feedback from the environment.
03:00And if you're continual learning company, you need traces, and if you have traces, then you can try to do continual learning over your agents.
03:09I had to put in a meme because if you look at your data, then you can be like Will Hunting, if anyone's seen the movie, where like everything is super, super easy,
03:18and you can like improve over time, and I promised Halima I would put this in there, so putting it in there.
03:25Cool.
03:26So, why am I talking a bunch about traces anyway?
03:30So, I'm sure a ton of us were software engineers before, we're software engineers now, and on the left we have a code block, and we can sort of like read the code, and in my head, I can almost reason over what this code does.
03:43I can see the functions, I can see like how they call each other.
03:46I can roughly understand the logic in Python.
03:49Um that doesn't exactly exist in agent world because agents have prompts, they have tools, they have skills, they have hooks, they have middlewares, some agents call other agents and I orchestrate them in swarms.
04:01It's really, really hard for humans to reason about how certain prompts that they change are actually going to affect agent behavior at scale.
04:09And this also varies between the different domains that you're doing it on.
04:13So, a prompt change and you're using for the medical domain is going to be like completely different than a prompt change that you want to do for the law domain.
04:19And in general, over the last four years since the ChatGPT moment, we've started trading determinism for autonomy.
04:28And in that shift, sort of what we need to do is create tools and create systems to still understand agents when they're autonomously operating in environments.
04:40So, I talked about traces.
04:42Um, why like why should you read them?
04:45And at LangChain, what do we actually do when we're reading traces?
04:49So, we centralize a bunch of our data, so we put everything in a tracing project and this is usually either like per agent or like centralized across all of our agents.
04:58And then what we do is we send agents to read traces from other agents, right?
05:05And then we look for a bunch of different things.
05:07And we might ask for, hey, like find a bunch of like good and bad interactions where like users got upset or like users are like really happy.
05:15Um, another question I might ask is uh this is a technical question like agents now run for millions of tokens.
05:22Does the agent get really dumb after the first compaction?
05:25After the second compaction?
05:27Does it never get dumb?
05:28Like how do we actually answer these questions?
05:29We need to do it by actually looking at the traces.
05:32And then the the other thing is like if I look at the traces, then I can try to prove some counterfactuals, which is hey, like I ran GPT 5.5 for this and I heard like GLM is really good.
05:44What happens if I run GLM 5.2 for this task, and how do I compare them?
05:49Metrics, awesome.
05:50The The trace level captures the actual like behavior that users see.
05:54So, that's also like very helpful for seeing behavior like fine-grained scales.
05:59And the way that we sort of think about the data that's being generated by agents is that the data that we see today is going to be the smallest that humans have ever seen in their entire lives because we're in this massive exponential
06:14shift to our agents are doing more and more work in the economy.
06:18And what that means is like the amount of data that humans have produced in our entire lifetime will soon be eclipsed by agents running on like year scales, and then 6-month scales, and 3-month scales, and then maybe every day, right?
06:32Um and to understand a ton of that data, roughly what we need to do is contend with a couple problems.
06:38There's more, but these are the two that I'm going to focus on.
06:41So, one, uh reading traces at scale is super expensive, uh especially if you have millions of traces, and if you have millions of tokens per trace, right?
06:52Um think of it as like an input token cost.
06:54You can like literally multiply the input token cost uh times the number of traces times like how big each trace is on average, right?
07:02Um the the other thing is [clears throat] if I have a super long interaction with a coding agent like Cloud Code or Codex or like deep agents, um I can't even read that trace with another agent because that
07:15that context like doesn't fit in memory, right?
07:17So, it's like we we need to develop systems so I can sort of treat that context as like an external object, and then I can sort of query into it, right?
07:23So, we we need to build agents to efficiently mine data from other agents, and it's it's no longer as simple as just like feeding the data into context, and there's like tricks that we'll sort of talk about uh to to do that well.
07:36Great.
07:37So, one of the things that I think is really, really cool in the last 6 months is that open models have basically hit an inflection point in intelligence that we at LangChain don't reach for the frontier models for every single use case.
07:51We're quite conscious about what is the minimum level of intelligence that I need to do any given task.
07:58And like practically speaking, honestly, yes, we start with Opus, we start with 55 because we just want to know if the task is even possible.
08:06But then once we reach that sort of like waterline, then we like look back at those traces and we see, "Hey, can we use an open model to do the same thing?"
08:15So, this is a bunch of work that we did with Harvey and then their lab legal benchmark.
08:20Basically, what we're looking at is can I match the trace judging capability of Opus with an open cheaper model?
08:28And the answer is roughly yes at like an order or like two orders of magnitude cheaper.
08:35And like the way we do that is we try a bunch of models, we do a bunch of like harness engineering, and the harness engineering is informed by a bunch of the traces that we read.
08:43So, it's like, "Hey, like Opus reasons about things in this way.
08:46Maybe that's because of the prompt.
08:48Maybe Opus is just smarter, which it is than a bunch of the open models, but that might mean I need to give it a little bit more guidance so it can reach the sort of same intelligence level at like a much
08:56much lower cost."
08:58And the the other thing that we sort of look at is like harness engineering is amazing.
09:04You get instant feedback and you can sort of like run on your evals, but eventually what we find is you hit a threshold of intelligence where it's like, "If I keep tweaking this prompt, I'm not going to get too much more out of it."
09:17And once we reach that point, we sort of look at, "Okay, can I actually like fine-tune the model on my domain-specific task?"
09:24And can I like make it better on those tasks?
09:27And what we find is if we take like base models and we tune them on like very specific vertical tasks, which is what a lot of our customers do,
09:35they don't really care about the entire variance of tasks.
09:37like they care about what their customers care about.
09:39So, if we focus on that narrow set of tasks, then we can fine-tune base models to sort of like reach and then also go beyond frontier performance.
09:47And I think one sort of like small thing I'll mention as a lot of people are getting into fine-tuning is that another sort of like economic decision is that you can move from token costs
09:59to hardware costs.
10:01And this is like can be a really big change, right?
10:04Cuz like you're very used to, "Hey, like a million tokens cost this much."
10:08Not as much like this cluster sort of costs this much.
10:10But for like very high inference workloads, we find it to be way cheaper just to like run a cluster and I get like unlimited inference on that cluster.
10:18I don't have to worry about tokens, but I can just do the calculation of like, "Hey, um this will end up being cheaper and then I can spin it down when I don't need it."
10:27Cool.
10:28And I said all of this um so we obviously like built a product to do that.
10:32Uh I won't shill it too much, but it's LangSwith engine.
10:36Uh basically, this product is trying to automate this loop for you, which is if you have any volume of trace data and you're looking for something that trace data or you want to generate e-vals from that trace data
10:48or you want to like generate feedback for like humans to read from that trace data, it will go read all of it.
10:53It'll like find issues.
10:54It'll agentically search over it.
10:56They can like prepare data sets for you to do something after.
11:00And a bit of a leader, um what that something basically is is the outputs of this trace mining exercise.
11:08So, there's like three things that I mentioned here uh which we see a bunch and we kind of put into the product.
11:14So, one is distillation and fine-tuning, which is let's say I'm running GLM 5.2. It's doing great, but I think that I can run this task like way cheaper with like a 9B or 13B model.
11:27Then what I'll do is like I'll take the good traces and the good examples from the GLM 5.2 runs, I'll prepare them in a data set, and then I'll try to fine-tune a small model on that data set to like mimic behavior,
11:37essentially, right?
11:38And this is like distillation, SFT.
11:41The The other one is generating evals in environments.
11:45So, maybe another slightly hot take, I think you can basically define agent behavior by showing the evals that you ran on it, right?
11:53Like, if someone showed me all the things that they're trying to test their agent on, I think I would have a rough idea about how that agent is going to behave because it literally like hill climbs those evals,
12:03and you you alter the behavior of the agent to make the evals pass, right?
12:08Like, the purpose of evals is roughly to try to make them pass, right?
12:12So, I update my agent so that they essentially pass.
12:15And then the the other thing is um like humans are still in the loop.
12:19Like, I need to know that customers are happy.
12:22I also want to know what my agents are doing.
12:24I just don't have the bandwidth to read a bunch of traces.
12:27So, preparing content for humans is still like really, really valuable today, especially in like high-trust domains like legal and medical.
12:37Like, some human needs to review this, um but they can't read it all, so we try to make it easy for them to process all that data.
12:46Great.
12:46This is um maybe a bit of a throwback.
12:48Like, how many people here know what like scikit-learn is?
12:51Uh maybe put your uh psych This crowd is just awesome.
12:54Um cool.
12:55So, uh when I was like first doing my PhD, uh my PhD was like kind of trying to do this, but like add new algorithms to scikit-learn.
13:05And like what scikit-learn basically is uh an abstract level, it's a bunch of helpers to fit learning systems to data.
13:15Right?
13:15And like classical machine learning, I had like a data set, I tried to fit it to it, but I think the same principles that we use in modern in I got I call it classical machine learning, it's like 6 years ago.
13:26Um that we do in classical machine learning uh definitely still apply to this agent-first world.
13:31Um, the way that they apply is what I like to call model harness task fit.
13:38So, we still have this sort of like fit function that I'm going to try to like take my data, take a harness, take a model, and I'm going to try to fit it all together to make sure that all of my tasks pass, right?
13:51The algorithms look slightly different, uh but the overall process of machine learning doesn't really look that different, and we'll talk about maybe roughly what our job becomes in this data-first,
14:05agent-first, fit-first world.
14:07So, a couple of our main jobs now are find good fit functions.
14:12So, these are like auto research.
14:14This is tons of great work that's being done in RL on different methods like OPD, OPSD, trySFT.
14:22And also find good data, right?
14:24So, if you put those two things together, then that is basically the applied or just overall research question that every team has to make their agents better.
14:33I like some some examples that we've seen that are like very popular that we're pretty bullish on are just generally auto research.
14:40So, if you have some sort of score that you can make number go up, uh agents are pretty good at making that number go up.
14:47They might cheat a little bit and you need to like check them on some stuff.
14:50Um, but this sort of like general feedback loop of do something, read the results, read the traces, and then do an update ends up being pretty useful.
14:59And then I talked about like model fine-tuning a bunch as well.
15:02Um, so we we just like went and did this.
15:05Uh, this was I think even before the term auto research came out, but a lot of people were doing it, which is hey, like terminal benches like really hard.
15:12Uh, what would happen if an agent just like read its traces, uh proposed experiments, and then tried to do fixes?
15:19Um, I think one like really key thing here is uh giving agents dense feedback signals.
15:26So, like terminal bench, the output is just a number, right?
15:29Like did you pass or did you not pass?
15:31Uh that's like kind of helpful, but if I give you like a super random task, like you just did a bunch of stuff, and then I just said like you failed or you passed,
15:38uh if you failed, like you wouldn't really have a good signal to figure out what you should do next, right?
15:43So, densifying feedback is uh really good way to improve agents, and like traces are the substrate that hold that feedback.
15:51And then agents are very good at like reading those uh those traces and like figuring out like what to do next.
15:56Um and then this sort of question always comes up, which is when should I like harness eng, when should I fine-tune, uh should I do more harness eng after it?
16:08I'm like pretty bullish on the idea of if you need to do something for improving your agent, the best thing that you can do is collect feedback as quickly as possible,
16:20like either from humans labeling or just letting the agents run.
16:23So, like harness engineering gives you feedback in maybe 2 minutes.
16:27Um once you sort of saturate the harness engineering ceiling, right, then you can maybe try to do like fine-tuning after that, but we find a lot of teams are happy with harness engineering
16:37and uh it solves their customer use case, so like we always sort of sort of recommend it.
16:41And then we have this like sort of sandwich, which is like try harness engineering, try to do fine-tuning to sort of like break through that ceiling, and then do more harness engineering again if you need to.
16:51And then I'll sort of end on the the idea generally of continual learning is that there's an agent taking actions in the environment, and then it needs to use that information,
17:04sorry guys, needs to use that information to update information about itself, right?
17:11So, it's like I did a bunch of these tasks, and like I need to update my prompts to make sure I do them more efficiently.
17:17Or users are users keep asking to search for these types of things, I should maybe tell like tell my creator that like they're they're doing this sort of stuff, right?
17:28It's like taking action in the environment kind of like humans do and updating ourselves.
17:32What that looks like today, slightly unclear, but we think that you're going to have to do it across all three axes, which is one, collect a bunch of training data, which is like observational
17:43data from agents taking actions.
17:46The The other one is like harness updates generally, like you know, the the Codex harness and the Cloud Code harness and like our harness and everyone's harness, like they look a certain way because like models are are trained in them and they look a certain way because of the tasks that they do in the real world and
18:00we think like evolving those over time is going to be super important in in order to make them work.
18:06And the the last thing is like memory.
18:10So, uh we humans are like really good at like remembering stuff over time, but we are not append-only logs of information.
18:17And if agents are going to be working with us over like year, 5-year, decade, lifetime time scales, we cannot just append everything to like a really big file and then search over it.
18:28There's a ton of stuff that needs to happen with like updating those files over time and then just making memory like really efficient.
18:34But, we think a lot of that actually comes from this idea of scaling sleep time compute and and dreaming generally.
18:40So, it's like read all of the traces over the entire agent life cycle and then like do
18:45[music]
18:45things to update agent state.
18:49Awesome.
18:49So, like quick quick takeaways, uh mining traces gives you signals to hill climb on.
18:55I would say like if you have an agent, just turn on tracing and point an agent at it and that's like the easiest thing that you can do to see like
19:03to basically understand what your agents are doing.
19:06We're very excited about open models.
19:08We want to help you fine-tune open models.
19:11We provide them as a service as well.
19:12So, if you're interested in that, would would love to chat how you can use open models to make everything smarter and cheaper.
19:18Um, continual learning is about operating environments and then integrating that data back into agent state.
19:24And then finally, I think this is so cool that like we have systems that's going to pre- that are going to produce more data than we ever have before.
19:31Uh, we need to all come up with like interesting research directions to learn how to like manage that at scale and like make all of our agents better.
19:39And with that, thank you.
19:41Thank you all for coming.
19:42[applause]