__await__ -> async flatten()
This commit is contained in:
parent
086dffd88c
commit
5743be54c0
1 changed files with 3 additions and 1 deletions
|
@ -110,9 +110,11 @@ class OllamaDownloadHandler:
|
|||
break
|
||||
yield line
|
||||
|
||||
def __await__(self):
|
||||
async def flatten(self) -> "OllamaDownloadHandler":
|
||||
"""Returns the current instance, but fully consumed."""
|
||||
async for _ in self:
|
||||
pass
|
||||
return self
|
||||
|
||||
|
||||
class OllamaChatHandler:
|
||||
|
|
Loading…
Reference in a new issue