diff --git a/matrix_pygmalion_bot/ai/runpod.py b/matrix_pygmalion_bot/ai/runpod.py index d7e5179..b34931a 100644 --- a/matrix_pygmalion_bot/ai/runpod.py +++ b/matrix_pygmalion_bot/ai/runpod.py @@ -115,7 +115,7 @@ async def generate_sync( output.sort(key=len, reverse=True) text = output[0] else: - text = r_json["output"] + text = output if api_mode == "runpod": reply = text.removeprefix(prompt) else: