File size: 4,263 Bytes
9b5b26a
 
bdf1a29
9b5b26a
 
c19d193
6aae614
4381f4e
8fe992b
9b5b26a
 
0964cc9
e42b0dc
0964cc9
af0d5eb
0964cc9
375298f
3018b1a
c948137
bdf1a29
58c2ada
bdf1a29
 
 
 
 
 
 
 
 
 
 
 
 
 
 
733fef2
 
bdf1a29
 
135231c
d21f9e0
bdf1a29
 
 
 
 
375298f
256f87d
9b5b26a
375298f
 
9b5b26a
375298f
bdf1a29
6347b3b
d21f9e0
375298f
bdf1a29
 
 
 
 
 
 
 
 
 
4e5660b
bdf1a29
 
 
9b5b26a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8c01ffb
 
6aae614
4381f4e
ae7a494
 
 
 
e121372
bf6d34c
 
29ec968
fe328e0
13d500a
8c01ffb
 
9b5b26a
 
8c01ffb
861422e
 
9b5b26a
8c01ffb
8fe992b
375298f
8c01ffb
 
 
 
 
 
861422e
8fe992b
 
9b5b26a
8c01ffb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
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()