Update routes.py

routes.py

@@ -101,6 +101,8 @@ def send_query(query_request: QueryRequest):
     """
     Processes a query and returns the bot's response along with any web sources.
     """
+    # trainer = get_trainer()
+    # trainer.load_bot(query_request.bot_id)
     prompt_type = query_request.prompt_type
     try:
         # Assign the prompt template based on the prompt type.
@@ -127,7 +129,7 @@ def send_query(query_request: QueryRequest):
         else:
             # Default prompt template if prompt_type is not recognized.
             prompt_template = PromptTemplates.get_quiz_solving_prompt()
-        trainer.set_custom_prompt_template(bot_id, prompt_template)
+        trainer.set_custom_prompt_template(query_request.bot_id, prompt_template)
         response, web_sources = trainer.get_response(
             query_request.query, query_request.bot_id, query_request.chat_id
         )