Buffy commited on
Commit
f9312e5
·
1 Parent(s): 79c96e0

diag: dump raw frames

Browse files
Files changed (1) hide show
  1. node_probe.py +2 -4
node_probe.py CHANGED
@@ -135,10 +135,8 @@ def register_routes(fa_app):
135
  await ws.send(json.dumps({"clientContent": {"turns": [{"role": "user", "parts": [{"text": q}]}], "turnComplete": True}}))
136
  setup = False
137
  continue
138
- t = f.get("serverContent", {}).get("outputTranscription", {}).get("text")
139
- if t:
140
- frames.append({"t": round(time.monotonic() - t0, 2), "len": len(t)})
141
- if f.get("serverContent", {}).get("generationComplete"):
142
  break
143
 
144
  try:
 
135
  await ws.send(json.dumps({"clientContent": {"turns": [{"role": "user", "parts": [{"text": q}]}], "turnComplete": True}}))
136
  setup = False
137
  continue
138
+ frames.append({"t": round(time.monotonic() - t0, 2), "keys": sorted(f.keys()), "raw": json.dumps(f)[:400]})
139
+ if len(frames) >= 6:
 
 
140
  break
141
 
142
  try: