UPDATE: trainChatbot

app.py

@@ -510,13 +510,18 @@ async def listChatbotSources(vectorstore: str):
     return result
 
 
+class TrainChatbot(BaseModel):
+    vectorstore: str
+    urls: list[str]
+
 
 @app.post("/trainChatbot")
-async def trainChatbot(vectorstore: str):
+async def trainChatbot(trainChatbotConfig: TrainChatbot):
+    vectorstore, UrlSources = trainChatbotConfig.vectorstore, trainChatbotConfig.urls
     texts = []
     sources = []
-    fileTypes = [supabase.table("ConversAI_ChatbotDataSources").select("sourceEndpoint").eq("sourceContentURL", x).execute().data[0]["sourceEndpoint"] for x in sources]
-    for source, fileType in zip(sources, fileTypes):
+    fileTypes = [supabase.table("ConversAI_ChatbotDataSources").select("sourceEndpoint").eq("sourceContentURL", x).execute().data[0]["sourceEndpoint"] for x in UrlSources]
+    for source, fileType in zip(UrlSources, fileTypes):
         if ((fileType == "/loadPDF") | (fileType == "/loadImagePDF")):
             r = requests.get(source)
             file = eval(r.content.decode("utf-8"))