Commit bf0294a

benny-dou <60535774+benny-dou@users.noreply.github.com>
2025-05-07 11:28:25
fix(gemini): delete uploaded file once finished
1 parent 02788ea
Changed files (1)
src/asr/gemini_asr.py
@@ -104,6 +104,8 @@ Notes:
 
         # all chunks are processed
         await modify_progress(message=status, text=blockquote(beautify_llm_response(runtime_texts)), force_update=True)
+        if uploaded_audio.name:  # delete file once finished
+            await app.aio.files.delete(name=uploaded_audio.name)
     except Exception as e:
         logger.error(e)
     return {"texts": transcriptions, "sent_messages": sent_messages}
@@ -158,6 +160,8 @@ async def gemini_nonstream_asr(path: str | Path, voice_format: str) -> str:
                     response_schema=list[Transcription],
                 ),
             )
+            if uploaded_audio.name:  # delete file once finished
+                client.files.delete(name=uploaded_audio.name)
             if parsed := glom(response.model_dump(), "parsed"):
                 return generate_transcription(parsed)
         except Exception as e: