File size: 4,713 Bytes
9b5b26a
 
 
 
c19d193
132becb
6aae614
8fe992b
9b5b26a
132becb
 
 
9b5b26a
08bf6cf
132becb
9b5b26a
ad6bbbf
9b5b26a
ad6bbbf
 
9b5b26a
132becb
ad6bbbf
 
 
 
 
 
 
 
 
 
 
592d72a
9514ad8
e1db3ec
132becb
e1db3ec
da72fc1
e1db3ec
da72fc1
e1db3ec
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
da72fc1
e1db3ec
da72fc1
 
e1db3ec
132becb
e1db3ec
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
592d72a
9b5b26a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8c01ffb
 
6aae614
ae7a494
 
 
 
e121372
bf6d34c
 
29ec968
fe328e0
13d500a
8c01ffb
 
9b5b26a
 
8c01ffb
861422e
 
9b5b26a
8c01ffb
8fe992b
c0ed62a
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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
import datetime
import requests
import pytz
import yaml
import os
from tools.final_answer import FinalAnswerTool

from Gradio_UI import GradioUI
alchemy_api_key = os.environ['ALCHEMY_API_KEY'] 
weather_api = os.environ['WEATHER_API']


@tool
def get_gas_fee_estimate(start: str, end: str) -> str: #it's import to specify the return type
    #Keep this format for the description / args / args description but feel free to modify the tool
    """A tool that estimates the gas fee for a transaction that occurs on the polygon blockchain.
    Args:
        start: A string representing a valid wallet address (e.g., 0xfe3b557e8fb62b89f4916b721be55ceb828dbd73)
        end: A second string representing a valid wallet address
    """
    url = f"https://polygon-mainnet.g.alchemy.com/v2/:{alchemy_api_key}"
    
    payload = { "0": {
        "from": {start},
        "to": {end},
        "value": "0x1"
    } }
    headers = {"Content-Type": "application/json"}
    try:
        response = requests.post(url, json=payload, headers=headers)
        return f"The estimated gas fee is {response['result']}"
    except error:
        return f"Error trying to process given wallet addresses {start} and/or {end}"
        
@tool
def get_asset_price(asset: str, base_curr: str) -> str:

    """A tool that gets the current price of a given digital asset.
    Args:
        asset: A string representing a valid digital asset name (e.g., ethereum).
        base_curr: A string that represents a valid base currency (e.g., usd).
    """

    url = "https://api.coingecko.com/api/v3/simple/price"
    params = {
        "ids": asset,
        "vs_currencies": base_curr
    }

    response = requests.get(url, params=params)
    if response.status_code == 200:
        data = response.json()
        price = data[asset][base_curr]
        return f"Current {asset} price: ${price}"
    else:
        return f"Error: {response.status_code}, {response.text}"

@tool
def get_weather_weatherapi(city: str, state: str) -> str:
    """A tool that gets the current weather.
    Args:
        city: A string that represents a valid city name (e.g., San Diego).
        state: A String that represents a valid state name (e.g., California).
    """
    api_key = weather_api 
    location = f"{city},{state}"
    url = f"http://api.weatherapi.com/v1/current.json"

    params = {
        "key": api_key,
        "q": location
    }

    try:
        response = requests.get(url, params=params)
        data = response.json()

        if response.status_code != 200:
            return f"API Error: {data.get('error', {}).get('message', 'Unknown error')}"

        temp = data["current"]["temp_f"]
        condition = data["current"]["condition"]["text"]
        name = data["location"]["name"]
        region = data["location"]["region"]

        return f"The current weather in {name}, {region} is {temp}°F with {condition}."

    except Exception as e:
        return f"Error: {e}"


@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()

# 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=[get_current_time_in_timezone,get_weather_weatherapi, get_asset_price, DuckDuckGoSearchTool(), final_answer], ## 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()