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

Fix view

Browse source
This commit is contained in:
Nexus 2024-03-22 09:15:45 +00:00
parent 3e0152ee24
commit 54e6076a9d
Signed by: nex
GPG key ID: 0FA334385D0B689F
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/cogs/ollama.py
View file

@ -241,7 +241,7 @@ class PromptSelector(discord.ui.View):
self.system_prompt = modal.value
@discord.ui.button(label="Set System Prompt", style=discord.ButtonStyle.primary, custom_id="usr")
async def set_system_prompt(self, btn: discord.ui.Button, interaction: Interaction):
async def set_user_prompt(self, btn: discord.ui.Button, interaction: Interaction):
modal = OllamaGetPrompt(self.ctx)
await interaction.response.send_modal(modal)
await modal.wait()
@ -344,6 +344,7 @@ class Ollama(commands.Cog):
)
]
):
system_query = None
if context is not None:
if not self.history.get_thread(context):
await ctx.respond("Invalid context key.")
@ -359,6 +360,7 @@ class Ollama(commands.Cog):
await ctx.respond("Select edit your prompts, as desired. Click done when you want to continue.", view=v)
await v.wait()
query = v.user_prompt or query
system_query = v.system_prompt
model = model.casefold()
try: