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

update the async client

Browse Source
This commit is contained in:
nguu0123 2024-12-12 21:12:46 +02:00
parent e44b04b039
commit d129151a13
No known key found for this signature in database
GPG Key ID: 6AE7349D14811E55
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ollama/_client.py
View File

@ -1080,7 +1080,7 @@ class AsyncClient(BaseClient):
async def delete(self, model: str) -> StatusResponse:
r = await self._request_raw(
'DELETE',
'POST',
'/api/delete',
json=DeleteRequest(
model=model,

2
tests/test_client.py
View File

@ -1179,7 +1179,7 @@ async def test_async_client_create_blob_exists(httpserver: HTTPServer):
@pytest.mark.asyncio
async def test_async_client_delete(httpserver: HTTPServer):
httpserver.expect_ordered_request(PrefixPattern('/api/delete'), method='DELETE').respond_with_response(Response(status=200))
httpserver.expect_ordered_request(PrefixPattern('/api/delete'), method='POST').respond_with_response(Response(status=200))
client = AsyncClient(httpserver.url_for('/api/delete'))
response = await client.delete('dummy')
assert response['status'] == 'success'