AhmedSohair commited on
Commit
8529eaf
·
verified ·
1 Parent(s): a5f69af

Fix null check on failed trace generation in preview mode

Browse files
Files changed (1) hide show
  1. generate_holistic_traces.py +5 -2
generate_holistic_traces.py CHANGED
@@ -349,8 +349,11 @@ def main():
349
  print(f" Profile: {json.dumps(ex['profile'], indent=4)}", flush=True)
350
 
351
  trace = generate_single_trace(client, deployment, ex["comments"], ex["profile"])
352
- print(f" Trace ({len(trace.split())} words):", flush=True)
353
- print(f" {trace}", flush=True)
 
 
 
354
 
355
  print(f"\n=== Preview complete. Run without --preview to generate all traces. ===", flush=True)
356
  return
 
349
  print(f" Profile: {json.dumps(ex['profile'], indent=4)}", flush=True)
350
 
351
  trace = generate_single_trace(client, deployment, ex["comments"], ex["profile"])
352
+ if trace:
353
+ print(f" Trace ({len(trace.split())} words):", flush=True)
354
+ print(f" {trace}", flush=True)
355
+ else:
356
+ print(f" ⚠ FAILED to generate trace", flush=True)
357
 
358
  print(f"\n=== Preview complete. Run without --preview to generate all traces. ===", flush=True)
359
  return