Refactored the primeira liga tool.

app.py

@@ -6,6 +6,7 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from tools.visit_webpage import VisitWebpageTool
 from tools.web_search import DuckDuckGoSearchTool
+from tools.primeira_liga import PrimeiraLigaTeamStatsTool
 from typing import Any, Dict, Optional
 
 from Gradio_UI import GradioUI
@@ -44,101 +45,7 @@ def curiosity(input: str) -> str:
     """
     return "Wow, amazing!"
 
-@tool
-def get_primeira_liga_team_stats(team_name: str) -> str:
-    """
-    Retrieve standing and last 3 match scores for a given Primeira Liga team using TheSportsDB API.
-    
-    Args:
-        team_name (str): Name of the team to search for
-    
-    Returns:
-        Dict containing team statistics or None if team not found
-    """
-
-    def _determine_match_outcome(home_team: str, away_team: str, 
-                              home_score: int, away_score: int, 
-                              target_team: str) -> str:
-        """
-        Determine match outcome (W/D/L) from the perspective of the target team.
-        """
-        
-        if home_team == target_team:
-            if home_score > away_score:
-                return 'W'
-            elif home_score < away_score:
-                return 'L'
-            else:
-                return 'D'
-        else:
-            if away_score > home_score:
-                return 'W'
-            elif away_score < home_score:
-                return 'L'
-            else:
-                return 'D'
-
-    # API Base URL
-    BASE_URL = "https://www.thesportsdb.com/api/v1/json/3"
-    
-    # Hardcoded Primeira Liga League ID
-    PRIMEIRA_LIGA_ID = 4344
-    
-    try:
-        # Step 1: Search for the team to get its ID
-        team_search_url = f"{BASE_URL}/searchteams.php?t={team_name.replace(' ', '%20')}"
-        team_search_response = requests.get(team_search_url)
-        team_search_data = team_search_response.json()
-        
-        if not team_search_data.get('teams'):
-            return "Error: Team not found"
-        
-        # Get the first matching team's ID
-        team_id = team_search_data['teams'][0]['idTeam']
-        
-        # Step 2: Get league table
-        table_url = f"{BASE_URL}/lookuptable.php?l={PRIMEIRA_LIGA_ID}"
-        table_response = requests.get(table_url)
-        table_data = table_response.json()
-        
-        # Find team's standing
-        standing = next(
-            (int(entry['intRank']) for entry in table_data.get('table', []) 
-             if entry['idTeam'] == team_id), 
-            None
-        )
-        
-        # Step 3: Get last 5 events for the team
-        last_events_url = f"{BASE_URL}/eventslast.php?id={team_id}"
-        last_events_response = requests.get(last_events_url)
-        last_events_data = last_events_response.json()
-        
-        # Process last 3 matches
-        last_3_matches = []
-        for event in last_events_data.get('results', [])[:3]:
-            match = {
-                'opponent': event['strAwayTeam'] if event['strHomeTeam'] == team_name else event['strHomeTeam'],
-                'result': f"{event['intHomeScore']} - {event['intAwayScore']}",
-                'outcome': _determine_match_outcome(
-                    event['strHomeTeam'], event['strAwayTeam'], 
-                    event['intHomeScore'], event['intAwayScore'], 
-                    team_name
-                )
-            }
-            last_3_matches.append(match)
-        
-        return str({
-            "team": team_name,
-            "standing": standing,
-            "last_3_matches": last_3_matches
-        })
-    
-    except Exception as e:
-        print(f"Error retrieving team stats: {e}")
-        return str("Error retrieving team stats")
-
-
-
+get_primeira_liga_team_stats = PrimeiraLigaTeamStatsTool()
 final_answer = FinalAnswerTool()
 visit_webpage = VisitWebpageTool()
 web_search = DuckDuckGoSearchTool()

tools/primeira_liga.py

@@ -0,0 +1,94 @@
+from smolagents.tools import Tool
+import requests
+
+class PrimeiraLigaTeamStatsTool(Tool):
+    name = "get_primeira_liga_team_stats"
+    description = "Retrieve standing and last 3 match scores for a given Primeira Liga team."
+    inputs = {'team_name': {'type': 'string', 'description': 'Name of the team to search for'}}
+    output_type = "string"
+
+    def forward(self, team_name: str) -> str:
+        def _determine_match_outcome(home_team: str, away_team: str, 
+                                home_score: int, away_score: int, 
+                                target_team: str) -> str:
+            """
+            Determine match outcome (W/D/L) from the perspective of the target team.
+            """
+            
+            if home_team == target_team:
+                if home_score > away_score:
+                    return 'W'
+                elif home_score < away_score:
+                    return 'L'
+                else:
+                    return 'D'
+            else:
+                if away_score > home_score:
+                    return 'W'
+                elif away_score < home_score:
+                    return 'L'
+                else:
+                    return 'D'
+
+        # API Base URL
+        BASE_URL = "https://www.thesportsdb.com/api/v1/json/3"
+        
+        # Hardcoded Primeira Liga League ID
+        PRIMEIRA_LIGA_ID = 4344
+        
+        try:
+            # Step 1: Search for the team to get its ID
+            team_search_url = f"{BASE_URL}/searchteams.php?t={team_name.replace(' ', '%20')}"
+            team_search_response = requests.get(team_search_url)
+            team_search_data = team_search_response.json()
+            
+            if not team_search_data.get('teams'):
+                return "Error: Team not found"
+            
+            # Get the first matching team's ID
+            team_id = team_search_data['teams'][0]['idTeam']
+            
+            # Step 2: Get league table
+            table_url = f"{BASE_URL}/lookuptable.php?l={PRIMEIRA_LIGA_ID}"
+            table_response = requests.get(table_url)
+            table_data = table_response.json()
+            
+            # Find team's standing
+            standing = next(
+                (int(entry['intRank']) for entry in table_data.get('table', []) 
+                if entry['idTeam'] == team_id), 
+                None
+            )
+            
+            # Step 3: Get last 5 events for the team
+            last_events_url = f"{BASE_URL}/eventslast.php?id={team_id}"
+            last_events_response = requests.get(last_events_url)
+            last_events_data = last_events_response.json()
+            
+            # Process last 3 matches
+            last_3_matches = []
+            for event in last_events_data.get('results', [])[:3]:
+                match = {
+                    'opponent': event['strAwayTeam'] if event['strHomeTeam'] == team_name else event['strHomeTeam'],
+                    'result': f"{event['intHomeScore']} - {event['intAwayScore']}",
+                    'outcome': _determine_match_outcome(
+                        event['strHomeTeam'], event['strAwayTeam'], 
+                        event['intHomeScore'], event['intAwayScore'], 
+                        team_name
+                    )
+                }
+                last_3_matches.append(match)
+            
+            return str({
+                "team": team_name,
+                "standing": standing,
+                "last_3_matches": last_3_matches
+            })
+        
+        except Exception as e:
+            print(f"Error retrieving team stats: {e}")
+            return str("Error retrieving team stats")
+
+    def __init__(self, *args, **kwargs):
+        self.is_initialized = False
+