Fix null check on failed trace generation in preview mode
Browse files
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 |
-
|
| 353 |
-
|
|
|
|
|
|
|
|
|
|
| 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
|