nex/college-bot-v2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Activity Actions

Fix ollama again

Browse source
This commit is contained in:
Nexus 2024-01-12 15:43:16 +00:00
parent 380d500e32
commit 05975de3af
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/cogs/ollama.py
View file

@ -456,11 +456,11 @@ class Ollama(commands.Cog):
buffer = io.StringIO()
if not view.cancel.is_set():
async for line in self.ollama_stream(response.content):
buffer.write(line["assistant"])
embed.description += line["assistant"]
buffer.write(line["message"]["content"])
embed.description += line["message"]["content"]
embed.timestamp = discord.utils.utcnow()
if len(embed.description) >= 4096:
embed.description = embed.description = "..." + line["assistant"]
embed.description = embed.description = "..." + line["message"]["content"]
if view.cancel.is_set():
break