feat: make model choice optional for /ask command (#57)

Author: rodonguyen

src/discord_bot/bot.py

@@ -26,17 +26,17 @@ async def on_message_create(event: interactions.api.events.MessageCreate) -> Non
 
 
 @interactions.slash_command(name="ask", description="Ask an LLM to answer anything")
+@interactions.slash_option(
+  name="prompt", description="Enter your prompt", required=True, opt_type=interactions.OptionType.STRING, min_length=10
+)
 @interactions.slash_option(
   name="model",
   description="Choose an LLM model",
-  required=True,
+  required=False,
   opt_type=interactions.OptionType.STRING,
-  autocomplete=True,
+  choices=[interactions.SlashCommandChoice(name=model, value=model) for model in MODEL_CHOICES],
 )
-@interactions.slash_option(
-  name="prompt", description="Enter your prompt", required=True, opt_type=interactions.OptionType.STRING, min_length=10
-)
-async def ask(ctx: interactions.SlashContext, model: str = "", prompt: str = "") -> None:
+async def ask(ctx: interactions.SlashContext, prompt: str = "", model: str = DEFAULT_MODEL) -> None:
   if model not in MODEL_CHOICES:
     await ctx.send(f"Invalid model `{model}`. Please choose from `{MODEL_CHOICES}`.")
     return
@@ -48,25 +48,6 @@ async def ask(ctx: interactions.SlashContext, model: str = "", prompt: str = "")
     await ctx.send(r)
 
 
-@ask.autocomplete("model")
-async def autocomplete(ctx: interactions.AutocompleteContext) -> None:
-  string_option_input = ctx.input_text
-  # you can use ctx.kwargs.get("name") to get the current state of other options - note they can be empty too
-  # make sure you respond within three seconds
-
-  filtered_choices = [choice for choice in MODEL_CHOICES if string_option_input in choice]
-
-  await ctx.send(
-    choices=[
-      {
-        "name": choice,
-        "value": choice,
-      }
-      for choice in filtered_choices
-    ]
-  )
-
-
 @interactions.slash_command(name="review-resume", description="Ask an LLM to review a resume")
 @interactions.slash_option(
   name="url",

tests/test_llm.py

@@ -20,14 +20,6 @@ async def test_answer_question__LLM_should_response() -> None:
   assert not response[0].startswith("Error")
 
 
-@pytest.mark.asyncio
-async def test_answer_question__invalid_model() -> None:
-  invalid_model = "not-a-gpt"
-  response = await llm.answer_question(invalid_model, simple_prompt, AI_SERVER_URL)
-
-  assert response[0].startswith("Error")
-
-
 @pytest.mark.asyncio
 async def test_answer_concurrent_question__should_be_at_the_same_time() -> None:
   n_models = 2