fix bugs

app.py

@@ -36,7 +36,7 @@ def reliable_url(url:str)-> bool:
     return False
 
 @tool
-def url_to_claim(url:str)-> dict:
+def url_to_news(url:str)-> dict:
     """A tool that receives a news url and returns the main claim of the webpage content and some aditional context
     Args:
         url: the address of the webpage we want to summarize into main claim and context
@@ -45,22 +45,27 @@ def url_to_claim(url:str)-> dict:
         response = requests.get(url, timeout=10)
         response.raise_for_status()
     except requests.exceptions.RequestException as e:
-        return f"An error occurred: {e}"
+        return {"title": "Error", "context": f"An error occurred: {e}"}
 
     soup = BeautifulSoup(response.content, "html.parser")
-    title = soup.title.text
+    if soup.title:
+        title = soup.title.text
+    else:
+        title = ""
+        
     context = soup.get_text()[:5000]
 
-    return {"claim": title, "context": context}
+    return {"title": title, "context": context}
 
 @tool
-def support_decision(claim:str, news="":dict)-> str:
+def support_decision(model, claim:str, news:dict=None)-> str:
     """A tool that decides if a given news article supports, contradicts or have an unclear opinion about a given claim
     Args:
         claim: the claim we want to decide whether it is supported or contradicted
         news: the news that may support or contradict the claim
     """
-    model = HfApiModel()
+    if news is None:
+        news = {"title": "", "context": ""}
 
     prompt = f"""
     Analyse the sentiment of the provided news article in contrast with the given claim and respond with only one of the following words 
@@ -68,12 +73,13 @@ def support_decision(claim:str, news="":dict)-> str:
 
     Claim: "{claim}"
     
-    News: "{news}"
+    News: "{news["title"]}
+    {news["context"]}"
     
     Response:
     """
 
-     try:
+    try:
         response = model.run(prompt)
         response = response.strip().lower()
         
@@ -110,16 +116,18 @@ def check(claim:str)-> str:
     Args:
         claim: the claim we want to check for support
     """
+    analyzer = HfApiModel()
+    
     support_reliable = []
     contradict_reliable = []
     support_others = []
     contradict_others = []
     
-    news_articles = web_search(claim)
+    news_articles = search(claim)
     for article in news_articles:
         reliable_source = reliable_url(article)
-        news = url_to_claim(article)
-        decision = support_decision(claim, news)
+        news = url_to_news(article)
+        decision = support_decision(analyzer, claim, news)
         if reliable_source:
             if decision == "supports":
                 support_reliable.append(article)
@@ -169,7 +177,7 @@ with open("prompts.yaml", 'r') as stream:
 agent = ToolCallingAgent(
     model=model,
     tools=[final_answer,
-           url_to_claim,
+           url_to_news,
            support_decision,
            search,
            check],