Update routes.py

routes.py

@@ -12,32 +12,23 @@ from prompt_templates import PromptTemplates
 router = APIRouter()
 trainer = get_trainer()
 
-# Global bot identifier.
-GLOBAL_BOT_ID = None
-
 @router.post("/initialize_bot", response_model=InitializeBotResponse)
 def initialize_bot():
     """
-    Initializes a global bot and returns its bot_id.
-    If the bot is already initialized, the existing bot_id is returned.
+    Initializes a bot and returns its bot_id.
     """
-    global GLOBAL_BOT_ID
     try:
-        if GLOBAL_BOT_ID is None:
-            GLOBAL_BOT_ID = trainer.initialize_bot_id()
-        return InitializeBotResponse(bot_id=GLOBAL_BOT_ID)
+        bot_id = trainer.initialize_bot_id()
+        return InitializeBotResponse(bot_id=bot_id)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
 
 @router.post("/upload_document")
-async def upload_document(file: UploadFile = File(...)):
+async def upload_document(bot_id: str = Form(...), file: UploadFile = File(...)):
     """
-    Saves the uploaded file temporarily and adds it to the global bot's knowledge base.
+    Saves the uploaded file temporarily and adds it to the specified bot's knowledge base.
     """
-    global GLOBAL_BOT_ID
-    if GLOBAL_BOT_ID is None:
-        raise HTTPException(status_code=400, detail="Global bot not initialized.")
     try:
         # Save the file to a temporary location.
         with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(file.filename)[1]) as tmp:
@@ -45,8 +36,8 @@ async def upload_document(file: UploadFile = File(...)):
             tmp.write(contents)
             tmp_path = tmp.name
 
-        # Add the document using the temporary file path to the global bot.
-        trainer.add_document_from_path(tmp_path, GLOBAL_BOT_ID)
+        # Add the document using the temporary file path to the specified bot.
+        trainer.add_document_from_path(tmp_path, bot_id)
 
         # Remove the temporary file.
         os.remove(tmp_path)
@@ -56,14 +47,11 @@ async def upload_document(file: UploadFile = File(...)):
 
 
 @router.post("/create_bot")
-def create_bot(prompt_type: str = Form(...)):
+def create_bot(bot_id: str = Form(...), prompt_type: str = Form(...)):
     """
-    Finalizes the creation (build) of the global bot using the specified prompt type.
+    Finalizes the creation (build) of the bot using the specified prompt type.
     This endpoint assigns the appropriate custom prompt template before finalizing bot creation.
     """
-    global GLOBAL_BOT_ID
-    if GLOBAL_BOT_ID is None:
-        raise HTTPException(status_code=400, detail="Global bot not initialized.")
     try:
         # Determine the appropriate prompt template based on the provided prompt_type.
         if prompt_type == "university":
@@ -90,23 +78,20 @@ def create_bot(prompt_type: str = Form(...)):
             # Default prompt template if prompt_type is not recognized.
             prompt_template = PromptTemplates.get_quiz_solving_prompt()
         
-        # Set the custom prompt template and create the global bot.
-        trainer.create_bot(GLOBAL_BOT_ID,prompt_template)
-        return {"message": f"Bot {GLOBAL_BOT_ID} created successfully."}
+        # Create the bot using the specified bot_id and prompt template.
+        trainer.create_bot(bot_id, prompt_template)
+        return {"message": f"Bot {bot_id} created successfully."}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
 
-@router.post("/new_chat", response_model=NewChatResponse)
-def new_chat():
+@router.post("/new_chat/{bot_id}", response_model=NewChatResponse)
+def new_chat(bot_id: str):
     """
-    Creates a new chat session for the global bot.
+    Creates a new chat session for the specified bot.
     """
-    global GLOBAL_BOT_ID
-    if GLOBAL_BOT_ID is None:
-        raise HTTPException(status_code=400, detail="Global bot not initialized.")
     try:
-        chat_id = trainer.new_chat(GLOBAL_BOT_ID)
+        chat_id = trainer.new_chat(bot_id)
         return NewChatResponse(chat_id=chat_id)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
@@ -116,30 +101,24 @@ def new_chat():
 def send_query(query_request: QueryRequest):
     """
     Processes a query and returns the bot's response along with any web sources.
-    Uses the global bot. Note: The prompt type is not set here.
+    The QueryRequest model must include the bot_id, chat_id, and query.
     """
-    global GLOBAL_BOT_ID
-    if GLOBAL_BOT_ID is None:
-        raise HTTPException(status_code=400, detail="Global bot not initialized.")
     try:
         response, web_sources = trainer.get_response(
-            query_request.query, GLOBAL_BOT_ID, query_request.chat_id
+            query_request.query, query_request.bot_id, query_request.chat_id
         )
         return QueryResponse(response=response, web_sources=web_sources)
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
 
-@router.get("/list_chats")
-def list_chats():
+@router.get("/list_chats/{bot_id}")
+def list_chats(bot_id: str):
     """
-    Returns a list of previous chat sessions for the global bot.
+    Returns a list of previous chat sessions for the specified bot.
     """
-    global GLOBAL_BOT_ID
-    if GLOBAL_BOT_ID is None:
-        raise HTTPException(status_code=400, detail="Global bot not initialized.")
     try:
-        chats = trainer.list_chats(GLOBAL_BOT_ID)
+        chats = trainer.list_chats(bot_id)
         return chats
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))