Fix /ollama pull

This commit is contained in:
Nexus 2024-06-11 01:21:34 +01:00
parent af11baeeaa
commit 3c61504cb3
Signed by: nex
GPG key ID: 0FA334385D0B689F

View file

@ -434,6 +434,7 @@ class Chat(commands.Cog):
last_completed = 0
last_completed_ts = time.time()
async with ollama_client(str(server.base_url)) as client:
async for line in await client.pull(model, stream=True):
if view.event.is_set():
embed.add_field(name="Error!", value="Download cancelled.")