kanshan/ollama-python
1
0
Fork 0
mirror of https://github.com/ollama/ollama-python.git synced 2026-01-13 21:57:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add AbstractAsyncContextManager to BaseClient

Browse Source
This commit is contained in:
ParthSareen 2025-02-18 15:48:39 -08:00
parent fdb18c9c92
commit 8ac47707d8
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ollama/_client.py
View File

@ -76,7 +76,7 @@ from ollama._types import (
T = TypeVar('T')
class BaseClient(contextlib.AbstractContextManager):
class BaseClient(contextlib.AbstractContextManager, contextlib.AbstractAsyncContextManager):
def __init__(
self,
client,
@ -120,9 +120,6 @@ class BaseClient(contextlib.AbstractContextManager):
def __exit__(self, exc_type, exc_val, exc_tb):
self.close()
async def __aenter__(self) -> Any:
return self
async def __aexit__(self, exc_type, exc_val, exc_tb):
await self.close()