laantungir/routstr-core
1
0
Fork 0
mirror of https://github.com/Routstr/routstr-core.git synced 2026-08-09 11:04:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

use correct header response

Browse Source
This commit is contained in:
9qeklajc
2025-07-26 01:22:36 +02:00
parent bdfb662dc7
commit 1161539fe3
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
router/payment/x_cashu.py
+3 -3
View File
@@ -154,6 +154,7 @@ async def handle_streaming_response(
except json.JSONDecodeError:
continue
response_headers = dict(response.headers)
# If we found usage data, calculate cost and refund
if usage_data and model:
response_data = {"usage": usage_data, "model": model}
@@ -163,11 +164,10 @@ async def handle_streaming_response(
refund_amount = amount - cost_data.total_msats
if refund_amount > 0:
refund_token = await send_refund(refund_amount, unit)
response.headers["X-Cashu"] = refund_token
response_headers["X-Cashu"] = refund_token
except Exception as e:
print(f"Error calculating cost for streaming response: {e}")
response_headers = dict(response.headers)
if "transfer-encoding" in response_headers:
del response_headers["transfer-encoding"]
if "content-encoding" in response_headers:
@@ -219,7 +219,7 @@ async def handle_non_streaming_response(
print("refund: ", refund_amount)
if refund_amount > 0:
refund_token = await send_refund(refund_amount, unit)
response.headers["X-Cashu"] = refund_token
response_headers["X-Cashu"] = refund_token
print(f"Refunded {refund_amount} msats")
return Response(