Skip to content

Commit d36d478

Browse files
vowelparrotUndertone0809
authored andcommitted
Add Run Collector Callback (langchain-ai#6133)
Add a callback handler that can collect nested run objects. Useful for evaluation.
1 parent 0ea2cb1 commit d36d478

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
"""A tracer that collects all nested runs in a list."""
2+
from typing import Any, List
3+
4+
from langchain.callbacks.tracers.base import BaseTracer
5+
from langchain.callbacks.tracers.schemas import Run
6+
7+
8+
class RunStackCallbackHandler(BaseTracer):
9+
"""A tracer that collects all nested runs in a list.
10+
11+
Useful for inspection and for evaluation."""
12+
13+
name = "run-collector_callback_handler"
14+
15+
def __init__(self, **kwargs: Any) -> None:
16+
super().__init__(**kwargs)
17+
self.traced_runs: List[Run] = []
18+
19+
def _persist_run(self, run: Run) -> None:
20+
self.traced_runs.append(run)

0 commit comments

Comments
 (0)