Is it possible to save traces locally?

[msurguy]

Describe the feature

Would like to save traces locally for debugging, especially when using custom models. Is it possible to save traces to file(s) locally?

[rm-openai]

Yup. Something like this should work:

1. Create a trace processor that exports to a file

class MyTraceProcessor(TracingProcessor):
    def __init__(self, output_dir: str):
        self._output_dir = output_dir
        self._output_file = open(os.path.join(output_dir, "traces.jsonl"), "w")

    def on_trace_start(self, trace: Trace) -> None:
        pass

    def on_trace_end(self, trace: Trace) -> None:
        self._output_file.write(json.dumps(trace.export()))
        pass

    def on_span_start(self, span: Span[Any]) -> None:
        pass

    def on_span_end(self, span: Span[Any]) -> None:
        self._output_file.write(json.dumps(span.export()))

    def shutdown(self, timeout: float | None = None):
        self._output_file.close()

    def force_flush(self):
        pass

and register it as an additional processor:

from agents import add_trace_processor

add_trace_processor(MyTraceProcessor("output"))

[pontus-devoteam]

Made a SDK in GO, you could check it out. Allows you to run everything locally.
https://github.com/pontus-devoteam/agent-sdk-go

[vincenzodomina]

Added a pull request #192 for the missing TracingProcessor export. That was necessary for me to make the above example work for me.

Also a little improvement to not throw an error for the missing output dir in case you did not add one like me:

def __init__(self, output_dir: str):
        self._output_dir = output_dir
        # added this to create dir and file if missing
        os.makedirs(output_dir, exist_ok=True)
        self._output_file = open(os.path.join(output_dir, "traces.jsonl"), "w")