Skip to content

Commit f196a4e

Browse files
committed
Add database connection close statement in asset retrieval error handling
1 parent c20d0b8 commit f196a4e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

llmstack/assets/apis.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import logging
22

3+
from django import db
34
from rest_framework import viewsets
45
from rest_framework.permissions import AllowAny, IsAuthenticated
56
from rest_framework.response import Response as DRFResponse
@@ -75,4 +76,5 @@ def get_asset_data(self, objref, request_user, request_session=None, include_dat
7576
return response
7677
except Exception as e:
7778
logger.error(f"Error retrieving asset: {e}")
79+
db.connections.close_all()
7880
return None

0 commit comments

Comments
 (0)