From cbd581cd8e61f793474fd409acb252e3f0707cee Mon Sep 17 00:00:00 2001
From: Shroominic <db2002dominic@gmail.com>
Date: Tue, 29 Apr 2025 13:08:50 +0800
Subject: [PATCH] pay-out

---
 router/proxy.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/router/proxy.py b/router/proxy.py
index 30ac020e..c70d3faf 100644
--- a/router/proxy.py
+++ b/router/proxy.py
@@ -1,3 +1,4 @@
+import asyncio
 import os
 import json
 from fastapi import APIRouter, Request, BackgroundTasks, Depends
@@ -5,6 +6,8 @@ from fastapi.responses import Response, StreamingResponse
 import httpx
 import re
 
+from router.cashu import pay_out
+
 from .auth import validate_bearer_key, pay_for_request, adjust_payment_for_tokens
 from .db import AsyncSession, get_session
 
@@ -143,6 +146,7 @@ async def proxy(
                         key, response_json, session
                     )
                     response_json["cost"] = cost_data
+                    asyncio.create_task(pay_out(session))
                     return Response(
                         content=json.dumps(response_json).encode(),
                         status_code=response.status_code,
@@ -162,6 +166,7 @@ async def proxy(
         background_tasks.add_task(response.aclose)
         background_tasks.add_task(client.aclose)
 
+        asyncio.create_task(pay_out(session))
         return StreamingResponse(
             response.aiter_bytes(),
             status_code=response.status_code,