Consistently type the submit callback

Author: cpsievert

shiny/templates/chat/llm-enterprise/aws-bedrock-anthropic/app.py

@@ -35,6 +35,6 @@
 
 # Define a callback to run when the user submits a message
 @chat.on_user_submit
-async def handle_user_input(user_input):
+async def handle_user_input(user_input: str):
     response = chat_model.stream(user_input)
     await chat.append_message_stream(response)

shiny/templates/chat/llm-enterprise/azure-openai/app.py

@@ -36,6 +36,6 @@
 
 # Define a callback to run when the user submits a message
 @chat.on_user_submit
-async def handle_user_input(user_input):
+async def handle_user_input(user_input: str):
     response = chat_model.stream(user_input)
     await chat.append_message_stream(response)

shiny/templates/chat/llms/anthropic/app.py

@@ -36,6 +36,6 @@
 
 # Generate a response when the user submits a message
 @chat.on_user_submit
-async def handle_user_input(user_input):
+async def handle_user_input(user_input: str):
     response = chat_model.stream(user_input)
     await chat.append_message_stream(response)

shiny/templates/chat/llms/google/app.py

@@ -32,6 +32,6 @@
 
 # Generate a response when the user submits a message
 @chat.on_user_submit
-async def handle_user_input(user_input):
+async def handle_user_input(user_input: str):
     response = chat_model.stream(user_input)
     await chat.append_message_stream(response)

shiny/templates/chat/llms/langchain/app.py

@@ -37,6 +37,6 @@
 
 # Define a callback to run when the user submits a message
 @chat.on_user_submit
-async def handle_user_input(user_input):
+async def handle_user_input(user_input: str):
     response = chat_model.stream(user_input)
     await chat.append_message_stream(response)

shiny/templates/chat/llms/ollama/app.py

@@ -28,6 +28,6 @@
 
 # Generate a response when the user submits a message
 @chat.on_user_submit
-async def handle_user_input(user_input):
+async def handle_user_input(user_input: str):
     response = chat_model.stream(user_input)
     await chat.append_message_stream(response)

shiny/templates/chat/llms/openai/app.py

@@ -36,6 +36,6 @@
 
 # Generate a response when the user submits a message
 @chat.on_user_submit
-async def handle_user_input(user_input):
+async def handle_user_input(user_input: str):
     response = chat_model.stream(user_input)
     await chat.append_message_stream(response)

shiny/ui/_chat.py

@@ -105,7 +105,7 @@ class Chat:
 
     # Define a callback to run when the user submits a message
     @chat.on_user_submit
-    async def handle_user_input(user_input):
+    async def handle_user_input(user_input: str):
         # Create a response message stream
         response = await my_model.generate_response(user_input, stream=True)
         # Append the response into the chat