scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

add exit and aenter

Browse source
This commit is contained in:
Ishaan Jaff 2024-04-06 19:27:26 -07:00
parent a38d3b17c5
commit 9be250c0f0
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
litellm/llms/custom_httpx/http_handler.py
View file

@ -22,6 +22,13 @@ class AsyncHTTPHandler:
# Close the client when you're done with it
await self.client.aclose()
async def __aenter__(self):
return self.client
async def __aexit__(self):
# close the client when exiting
await self.client.aclose()
async def get(
self, url: str, params: Optional[dict] = None, headers: Optional[dict] = None
):