Update tests.py

tests.py

@@ -379,7 +379,7 @@ def read_excel_file(filename) -> dict:
     return excel_data_dict
 @mcp.tool()
 def scrape_websites(url_list:list,query:str) -> list:
-    """Scrapes specific website content.query is the question you want to ask about the content of the website.e.g-query:Give .pptx links in the website,Summarise the content in very  great detail,etc..Note:Max urls in url_list is 3."""
+    """Scrapes specific website content.query is the question you want to ask about the content of the website.e.g-query:Give .pptx links in the website,Summarise the content in very  great detail,etc.Maximum 4 urls can be passed at a time."""
 
     conn = http.client.HTTPSConnection("scrapeninja.p.rapidapi.com")
 
@@ -390,6 +390,7 @@ def scrape_websites(url_list:list,query:str) -> list:
     'Content-Type': "application/json"
     }
     Output=""
+    links=""
     content=""
     for urls in url_list:
         payload = {"url" :urls} 
@@ -400,6 +401,7 @@ def scrape_websites(url_list:list,query:str) -> list:
         content=content+str(data.decode("utf-8"))
         
     #Only thing llama 4 is good for.
+
     response = clienty.chat.completions.create(
         model="meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
         messages=[
@@ -408,8 +410,16 @@ def scrape_websites(url_list:list,query:str) -> list:
     )
     for chunk in response:
         Output = Output +str(chunk.choices[0].delta.content)
-        
-    return {"website_content":Output}
+    #--------------
+    response2 = clienty.chat.completions.create(
+        model="meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
+        messages=[
+        {"role": "user", "content": f"Give all relevant links in this content.The links may be relevant image links , file links , video links , website links , etc .You must give Minimum 10 links and maximum 20 links.[CONTENT]:{content}"}
+    ],stream=True
+    )
+    for chunk in response2:
+        links = links +str(chunk.choices[0].delta.content)
+    return {"website_content":Output,"relevant_links":links}