__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
|
break
|
||||||
yield line
|
yield line
|
||||||
|
|
||||||
def __await__(self):
|
async def flatten(self) -> "OllamaDownloadHandler":
|
||||||
|
"""Returns the current instance, but fully consumed."""
|
||||||
async for _ in self:
|
async for _ in self:
|
||||||
pass
|
pass
|
||||||
|
return self
|
||||||
|
|
||||||
|
|
||||||
class OllamaChatHandler:
|
class OllamaChatHandler:
|
||||||
|
|
Loading…
Reference in a new issue