from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool import datetime from datetime import date import requests import pytz import yaml from tools.final_answer import FinalAnswerTool from tools.visit_webpage import VisitWebpageTool from Gradio_UI import GradioUI import os from huggingface_hub import login login(token = os.getenv('nangelov_hf_token')) FOOTBALL_DATA_TOKEN = os.getenv('football_data_token') @tool def get_Competition_Code(competition_name: str)-> str: """A tool that returns the competition code for a given competition name. Args: competition_name: A string representing the full name of the competition, e.g., "Premier League" or "Bundesliga". """ competitions = { "WC": "FIFA World Cup", "CL": "UEFA Champions League", "BL1": "Bundesliga", "DED": "Eredivisie", "BSA": "Campeonato Brasileiro Série A", "PD": "Primera Division", "FL1": "Ligue 1", "ELC": "Championship", "PPL": "Primeira Liga", "EC": "European Championship", "SA": "Serie A", "PL": "Premier League" } print(competition_name) # Iterate through the dictionary to find the matching competition name. for code, name in competitions.items(): if name.lower() == competition_name.lower(): print(code) return code # Return None if no match is found. return None @tool def search_Competition_Matches_Results_By_Date(competition_code: str, matchday: int)-> dict: """A tool that fetch football games results for a given competition from a specified date. Args: competition_code: A string representing the code of the competition, e.g., "Premier League" or "Bundesliga". matchday: An integer representing the matchday for which to retrieve match results format: Integer /[1-4]+[0-9]*/. """ # Create the URL using an f-string url = f'https://api.football-data.org/v4/competitions/{competition_code}/matches?matchday={matchday}' print(url) # Set up the headers with your API token. headers = { 'X-Auth-Token': FOOTBALL_DATA_TOKEN } # Make the request response = requests.get(url, headers=headers) # Check if the request was successful. if response.status_code == 200: return response.json() # Parse and return the JSON response. else: # Return error details as a dict return {"error": response.status_code, "message": response.text} @tool def get_current_time_in_timezone(timezone: str) -> str: """A tool that fetches the current local time in a specified timezone. Args: timezone: A string representing a valid timezone (e.g., 'America/New_York'). """ try: # Create timezone object tz = pytz.timezone(timezone) # Get current time in that timezone local_time = datetime.datetime.now(tz).strftime("%Y-%m-%d %H:%M:%S") return f"The current local time in {timezone} is: {local_time}" except Exception as e: return f"Error fetching time for timezone '{timezone}': {str(e)}" final_answer = FinalAnswerTool() visit_webpage = VisitWebpageTool() # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder: # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' model = HfApiModel( max_tokens=2096, temperature=0.5, model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded custom_role_conversions=None, ) # Import tool from Hub image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True) with open("prompts.yaml", 'r') as stream: prompt_templates = yaml.safe_load(stream) agent = CodeAgent( model=model, tools=[final_answer, visit_webpage, search_Competition_Matches_Results_By_Date, get_current_time_in_timezone, get_Competition_Code], ## add your tools here (don't remove final answer) max_steps=6, verbosity_level=1, grammar=None, planning_interval=None, name=None, description=None, prompt_templates=prompt_templates ) GradioUI(agent).launch()