Fix ollama again
This commit is contained in:
parent
380d500e32
commit
05975de3af
1 changed files with 3 additions and 3 deletions
|
@ -456,11 +456,11 @@ class Ollama(commands.Cog):
|
||||||
buffer = io.StringIO()
|
buffer = io.StringIO()
|
||||||
if not view.cancel.is_set():
|
if not view.cancel.is_set():
|
||||||
async for line in self.ollama_stream(response.content):
|
async for line in self.ollama_stream(response.content):
|
||||||
buffer.write(line["assistant"])
|
buffer.write(line["message"]["content"])
|
||||||
embed.description += line["assistant"]
|
embed.description += line["message"]["content"]
|
||||||
embed.timestamp = discord.utils.utcnow()
|
embed.timestamp = discord.utils.utcnow()
|
||||||
if len(embed.description) >= 4096:
|
if len(embed.description) >= 4096:
|
||||||
embed.description = embed.description = "..." + line["assistant"]
|
embed.description = embed.description = "..." + line["message"]["content"]
|
||||||
|
|
||||||
if view.cancel.is_set():
|
if view.cancel.is_set():
|
||||||
break
|
break
|
||||||
|
|
Loading…
Reference in a new issue